X-Git-Url: http://git.treefish.org/~alex/photosort.git/blobdiff_plain/e97415dff8bbc02a90428374c37d4d105d2a38ce..359ed2700cbd68627fdb5e8cd0da02b34321116a:/src/misc.py?ds=sidebyside diff --git a/src/misc.py b/src/misc.py index 08baabb..acd5c94 100644 --- a/src/misc.py +++ b/src/misc.py @@ -1,4 +1,5 @@ import datetime +import logging import mimetypes import os import PIL.Image @@ -19,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):