tcsenpai/pensieve
1
0
Fork 0
mirror of https://github.com/tcsenpai/pensieve.git synced 2025-06-06 03:05:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: bad way to overwrite exif for webp

Browse Source
This commit is contained in:
arkohut 2024-09-29 16:15:53 +08:00
parent a27145e0e0
commit d30307c824
1 changed files with 17 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
memos/utils.py
View File

@ -2,27 +2,23 @@ from PIL import Image
import piexif
from PIL.PngImagePlugin import PngInfo
import json
from pathlib import Path
def write_image_metadata(image_path, metadata):
img = Image.open(image_path)
image_path_str = str(image_path)
if image_path_str.lower().endswith((".jpg", ".jpeg", ".tiff")):
exif_dict = piexif.load(image_path)
updated_description = json.dumps(metadata).encode("utf-8")
exif_dict["0th"][piexif.ImageIFD.ImageDescription] = updated_description
if image_path_str.lower().endswith((".jpg", ".jpeg", ".tiff", ".webp")):
exif_dict = {"0th": {}, "Exif": {}, "GPS": {}, "1st": {}, "thumbnail": None}
exif_dict["0th"][piexif.ImageIFD.ImageDescription] = json.dumps(
metadata
).encode("utf-8")
exif_bytes = piexif.dump(exif_dict)
img.save(image_path, exif=exif_bytes)
elif image_path_str.lower().endswith(".png"):
metadata_info = PngInfo()
metadata_info.add_text("Description", json.dumps(metadata))
img.save(image_path, "PNG", pnginfo=metadata_info)
elif image_path_str.lower().endswith(".webp"):
# Convert metadata to bytes
metadata_bytes = json.dumps(metadata).encode('utf-8')
# Use exif parameter to save metadata
img.save(image_path, "WebP", exif=metadata_bytes, quality=85)
else:
print(f"Skipping unsupported file format: {image_path_str}")
@ -31,32 +27,23 @@ def get_image_metadata(image_path):
img = Image.open(image_path)
image_path_str = str(image_path)
if image_path_str.lower().endswith((".jpg", ".jpeg", ".tiff")):
exif_dict = piexif.load(str(image_path)) # Convert Path object to string
if image_path_str.lower().endswith((".jpg", ".jpeg", ".tiff", ".webp")):
try:
exif_dict = piexif.load(image_path_str)
existing_description = exif_dict["0th"].get(
piexif.ImageIFD.ImageDescription, b"{}"
)
try:
return json.loads(existing_description.decode("utf-8"))
except json.JSONDecodeError as e:
print(f"Error decoding JSON metadata for {image_path_str}: {e}")
return {}
except Exception as e:
print(f"Error decoding EXIF metadata for {image_path_str}: {e}")
return None
elif image_path_str.lower().endswith(".png"):
existing_description = img.info.get("Description", "{}")
try:
return json.loads(existing_description)
except json.JSONDecodeError as e:
print(f"Error decoding JSON metadata for {image_path_str}: {e}")
return {}
elif image_path_str.lower().endswith(".webp"):
existing_metadata = img.info.get("exif", b"{}")
try:
if isinstance(existing_metadata, bytes):
existing_metadata = existing_metadata.decode('utf-8')
return json.loads(existing_metadata)
except json.JSONDecodeError as e:
print(f"Error decoding JSON metadata for {image_path_str}: {e}")
return {}
print(f"Error decoding PNG metadata for {image_path_str}: {e}")
return None
else:
print(f"Unsupported file format: {image_path_str}")
return None