added missing import

[photosort.git] / src / misc.py

diff --git a/src/misc.py b/src/misc.py
index 9f5d677ff8abce6a477a1973dc174f68ad7ebe9c..93690d10cd1596513ed947dbf50292b27bc0f8dd 100644 (file)
--- a/src/misc.py
+++ b/src/misc.py
@@ -1,29 +1,28 @@
 import datetime
+import logging
 import mimetypes
 import os
 import PIL.Image
+import re
 import shutil
+import subprocess
 
 def walk_media_files(dir_path):
     for root, dirs, files in os.walk(dir_path):
         for f in files:
             file_path = os.path.join(root, f)
-            if _is_media_file(file_path):
+            if is_media_file(file_path):
                 yield (f, file_path)
 
-def extract_timestamp(file_path, use_exif=False):
-    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 extract_meta_time(file_path):
+    if is_media_file(file_path, types=['image']):
+        return _extract_image_timestamp(file_path)
+    elif is_media_file(file_path, types=['video']):
+        return _extract_video_timestamp(file_path)
 
 def find_file(dir_path, file_name, file_size, exclude_dir):
     for root, dirs, files in os.walk(dir_path):
-        if root == exclude_dir:
+        if root.startswith(exclude_dir):
             continue
         for f in files:
             if f == file_name:
@@ -38,15 +37,7 @@ def import_file(src_file_path, dst_file_path):
     dst_stat = os.stat(dst_file_path)
     os.utime( dst_file_path, ns=(dst_stat.st_atime_ns, src_stat.st_mtime_ns) )
 
-def delete_dir_contents(dir_path):
-    for file_name in os.listdir(dir_path):
-        file_path = os.path.join(dir_path, file_name)
-        if os.path.isfile(file_path) or os.path.islink(file_path):
-            os.unlink(file_path)
-        elif os.path.isdir(file_path):
-            shutil.rmtree(file_path)
-
-def _is_media_file(file_path, types=['image', 'video']):
+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]
@@ -55,3 +46,27 @@ def _is_media_file(file_path, types=['image', 'video']):
     if not mime_type.split('/')[0] in types:
         return False
     return True
+
+def _extract_image_timestamp(file_path):
+    time = None
+    try:
+        with PIL.Image.open(file_path) as image:
+            exif = image._getexif()
+            if exif and 36867 in exif:
+                time = datetime.datetime\
+                               .strptime(exif[36867], '%Y:%m:%d %H:%M:%S')\
+                               .timestamp()
+    except Exception as e:
+        logging.warn("Error extracting exif for %s: %s", file_path, str(e))
+    return time
+
+def _extract_video_timestamp(file_path):
+    p = subprocess.run(['ffmpeg', '-i', file_path],
+                       capture_output=True, encoding='UTF-8')
+    for line in p.stderr.splitlines():
+        m = re.search('^.*creation_time.*: ([^ ]+)$', line)
+        if m:
+            return datetime.datetime\
+                           .strptime(m.group(1), '%Y-%m-%dT%H:%M:%S.%fZ')\
+                           .timestamp()
+    return None