|
|
| import subprocess, json |
| from PIL import Image, ExifTags |
|
|
| def get_video_rotation(path): |
|
|
| try: |
| info = json.loads( |
| subprocess.check_output([ |
| "ffprobe", |
| "-v", "error", |
| "-select_streams", "v:0", |
| "-show_entries", "stream_side_data=rotation", |
| "-of", "json", |
| str(path), |
| ],stderr=subprocess.DEVNULL) |
| ) |
|
|
| return next( |
| (sd["rotation"] |
| for sd in info.get("streams", [{}])[0].get("side_data_list", []) |
| if "rotation" in sd), |
| 0, |
| ) |
| except (subprocess.CalledProcessError, json.JSONDecodeError, KeyError) as e: |
| try: |
| info = json.loads( |
| subprocess.check_output([ |
| "ffprobe", |
| "-v", "error", |
| "-select_streams", "v:0", |
| "-show_entries", "stream_tags=rotate", |
| "-of", "json", |
| str(path), |
| ],stderr=subprocess.DEVNULL) |
| ) |
| streams = info.get("streams", []) |
| if streams: |
| tags = streams[0].get("tags", {}) |
| rotation = tags.get("rotate") |
| if rotation: |
| return int(rotation) |
| except Exception: |
| pass |
| return 0 |
|
|
| def get_image_rotation(path: str) -> int: |
| """ |
| Get the rotation angle of an image based on EXIF Orientation. |
| Returns 0, 90, 180, or 270 degrees. |
| """ |
| try: |
| img = Image.open(path) |
| exif = img._getexif() |
| if not exif: |
| return 0 |
| |
| |
| orientation_key = next( |
| (k for k, v in ExifTags.TAGS.items() if v == "Orientation"), None |
| ) |
| if not orientation_key: |
| return 0 |
| orientation = exif.get(orientation_key, 1) |
|
|
| |
| return {3: 180, 6: 270, 8: 90}.get(orientation, 0) |
| except Exception: |
| return 0 |
