From 814a257da1e8aafa15f94f122d10c8f01bf2ad51 Mon Sep 17 00:00:00 2001 From: Alexander Schmidt Date: Fri, 16 Oct 2020 14:35:11 +0200 Subject: [PATCH 1/1] only extract exif for image files --- src/misc.py | 21 +++++++++------------ 1 file changed, 9 insertions(+), 12 deletions(-) diff --git a/src/misc.py b/src/misc.py index 4d39b33..adb94b1 100644 --- a/src/misc.py +++ b/src/misc.py @@ -12,16 +12,13 @@ def walk_media_files(dir_path): yield (f, file_path) def extract_timestamp(file_path, use_exif=False): - if use_exif: - try: - with PIL.Image.open(file_path) as image: - exif = image._getexif() - if exif and 36867 in exif: - return int( datetime.datetime - .strptime(exif[36867], '%Y:%m:%d %H:%M:%S') - .timestamp() ) - except PIL.UnidentifiedImageError: - pass + if use_exif and _is_media_file(file_path, types=['image']): + with PIL.Image.open(file_path) as image: + exif = image._getexif() + if exif and 36867 in exif: + return int( datetime.datetime + .strptime(exif[36867], '%Y:%m:%d %H:%M:%S') + .timestamp() ) return os.path.getmtime(file_path) def find_file(dir_path, file_name, file_size, exclude_dir): @@ -49,12 +46,12 @@ def delete_dir_contents(dir_path): elif os.path.isdir(file_path): shutil.rmtree(file_path) -def _is_media_file(file_path): +def _is_media_file(file_path, types=['image', 'video']): if not os.path.isfile(file_path): return False mime_type = mimetypes.guess_type(file_path)[0] if not mime_type: return False - if not mime_type.split('/')[0] in ['image', 'video']: + if not mime_type.split('/')[0] in types: return False return True -- 2.39.5