116 lines
3.7 KiB
Python
116 lines
3.7 KiB
Python
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
|