Issue

Code backup
2026-05-10 16:59:01 +02:00
commit 368d6fafea
796 changed files with 315310 additions and 0 deletions
@@ -0,0 +1,115 @@
+import os
+from PIL import Image, ExifTags, UnidentifiedImageError
+from pillow_heif import register_heif_opener
+from datetime import datetime
+import piexif
+import fnmatch
+
+register_heif_opener(allow_incorrect_headers=True)
+
+
+def get_file_list(dir_of_interest, recursive):
+    """
+    Get a list of all files in the directory of interest
+
+    :param dir_of_interest: the directory to search
+    :param recursive: search subdirectories
+
+    :return: a list of files
+    """
+    file_list = []
+
+    if os.path.isdir(dir_of_interest):
+        for root, dirs, files in os.walk(dir_of_interest):
+            if not recursive:
+                dirs.clear()
+            for file in files:
+                if fnmatch.fnmatch(file.lower(), '*.heic'):
+                    file_list.append([os.path.normpath(root), file])
+        return file_list
+    else:
+        print("Path {} is not a valid directory.".format(dir_of_interest))
+        return None
+
+
+def convert_heic_file(source_file, target_file, overwrite, remove):
+    """
+    Convert a single heic file to jpeg
+
+    :param source_file: the source file
+    :param target_file: the target file
+    :param overwrite: overwrite existing jpeg files
+    :param remove: remove converted heic files
+
+    :return: True if successful, False otherwise
+    """
+
+    if os.path.exists(target_file) and not overwrite:
+        print(f'File {target_file} already exists, skip')
+        return False
+
+    try:
+        image = Image.open(source_file)
+        image_exif = image.getexif()
+        if image_exif:
+            # Make a map with tag names and grab the datetime
+            exif = {ExifTags.TAGS[k]: v for k, v in image_exif.items() if k in ExifTags.TAGS and type(v) is not bytes}
+            if 'DateTime' in exif:
+                date = datetime.strptime(exif['DateTime'], '%Y:%m:%d %H:%M:%S')
+            else:
+                date = datetime.now()
+
+            # Load exif data via piexif
+            exif_dict = piexif.load(image.info["exif"])
+
+            # Update exif data with orientation and datetime
+            exif_dict["0th"][piexif.ImageIFD.DateTime] = date.strftime("%Y:%m:%d %H:%M:%S")
+            exif_dict["0th"][piexif.ImageIFD.Orientation] = 1
+            exif_bytes = piexif.dump(exif_dict)
+
+            # Save image as jpeg
+            image.save(target_file, "jpeg", exif=exif_bytes)
+            print(f'Converted image: {source_file}')
+            if remove:
+                os.remove(source_file)
+
+            return True
+        else:
+            print(f"Unable to get exif data for {source_file}")
+            
+    except UnidentifiedImageError as e:
+        print(f"{source_file} is not a valid image : {e}")
+    except Exception as e:
+        print(f"Unable to convert {source_file}: {e}")
+
+    return False
+
+
+def convert_heic_to_jpeg(dir_of_interest, recursive, overwrite, remove):
+    """
+    Convert all heic files in the directory of interest to jpeg
+
+    :param dir_of_interest: The directory to search
+    :param recursive: search subdirectories
+    :param overwrite: overwrite existing jpeg files
+    :param remove: remove converted heic files
+
+    :return: a list of successfully converted files
+    """
+    heic_files = get_file_list(dir_of_interest, recursive)
+
+    # Extract files of interest
+    success_files = []
+
+    print(f'Found {len(heic_files)} files to convert in folder {dir_of_interest}')
+
+    # Convert files to jpg while keeping the timestamp
+    for root, filename in heic_files:
+
+        target_filename = os.path.splitext(filename)[0] + ".jpg"
+        target_file = os.path.join(root, target_filename)
+        source_file = os.path.join(root, filename)
+        if convert_heic_file(source_file, target_file, overwrite, remove):
+            success_files.append(target_filename)
+
+    return success_files