use locator db to speed up file search

[photosort.git] / src / misc.py

diff --git a/src/misc.py b/src/misc.py
index 93690d10cd1596513ed947dbf50292b27bc0f8dd..acd5c944c48451ec62a448d406b671fcf0fa0f66 100644 (file)
--- a/src/misc.py
+++ b/src/misc.py
@@ -20,15 +20,17 @@ def extract_meta_time(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):
+def find_file(base_dir, name, size, meta_time, exclude_dir):
+    for root, dirs, files in os.walk(base_dir):
         if root.startswith(exclude_dir):
             continue
-        for f in files:
-            if f == file_name:
-                full_path = os.path.join(root, f)
-                if os.path.getsize(full_path) == file_size:
-                    return root
+        for other_name in files:
+            if other_name == name:
+                full_path = os.path.join(root, other_name)
+                if os.path.getsize(full_path) == size:
+                    other_meta_time = extract_meta_time(full_path)
+                    if meta_time == other_meta_time:
+                        return root
     return None
 
 def import_file(src_file_path, dst_file_path):