alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-25 15:15:44 -03:00
Code Issues Packages Projects Releases Wiki Activity

fix: Prevent EXIF metadata operations on video files by adding type checks.

Browse Source
This commit is contained in:
Will Miao
2026-01-02 20:18:03 +08:00
parent ab85ba54a9
commit 4317b06049
1 changed files with 28 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
py/utils/exif_utils.py
View File

@@ -22,6 +22,12 @@ class ExifUtils:
Optional[str]: Extracted metadata or None if not found Optional[str]: Extracted metadata or None if not found
""" """
try: try:
# Skip for video files
if image_path:
ext = os.path.splitext(image_path)[1].lower()
if ext in ['.mp4', '.webm']:
return None
# First try to open the image # First try to open the image
with Image.open(image_path) as img: with Image.open(image_path) as img:
# Method 1: Check for parameters in image info # Method 1: Check for parameters in image info
@@ -80,6 +86,12 @@ class ExifUtils:
str: Path to the updated image str: Path to the updated image
""" """
try: try:
# Skip for video files
if image_path:
ext = os.path.splitext(image_path)[1].lower()
if ext in ['.mp4', '.webm']:
return image_path
# Load the image and check its format # Load the image and check its format
with Image.open(image_path) as img: with Image.open(image_path) as img:
img_format = img.format img_format = img.format
@@ -133,6 +145,12 @@ class ExifUtils:
def append_recipe_metadata(image_path, recipe_data) -> str: def append_recipe_metadata(image_path, recipe_data) -> str:
"""Append recipe metadata to an image's EXIF data""" """Append recipe metadata to an image's EXIF data"""
try: try:
# Skip for video files
if image_path:
ext = os.path.splitext(image_path)[1].lower()
if ext in ['.mp4', '.webm']:
return image_path
# First, extract existing metadata # First, extract existing metadata
metadata = ExifUtils.extract_image_metadata(image_path) metadata = ExifUtils.extract_image_metadata(image_path)
@@ -242,6 +260,16 @@ class ExifUtils:
Tuple of (optimized_image_data, extension) Tuple of (optimized_image_data, extension)
""" """
try: try:
# Skip for video files early if it's a file path
if isinstance(image_data, str) and os.path.exists(image_data):
ext = os.path.splitext(image_data)[1].lower()
if ext in ['.mp4', '.webm']:
try:
with open(image_data, 'rb') as f:
return f.read(), ext
except Exception:
return image_data, ext
# First validate the image data is usable # First validate the image data is usable
img = None img = None
if isinstance(image_data, str) and os.path.exists(image_data): if isinstance(image_data, str) and os.path.exists(image_data):