From: Alexander Schmidt Date: Tue, 20 Oct 2020 09:27:14 +0000 (+0200) Subject: only accept alternative files with same meta time X-Git-Url: http://git.treefish.org/~alex/photosort.git/commitdiff_plain/5ad2c3b2ac3baf7b9e64d07a7f0b37d3493a03d4 only accept alternative files with same meta time --- diff --git a/src/migrator.py b/src/migrator.py index 73650ce..7bd32e6 100644 --- a/src/migrator.py +++ b/src/migrator.py @@ -29,10 +29,11 @@ class Migrator: dst_file_path = os.path.join(dst_dir, src_file_name) if not os.path.exists(dst_file_path): - alt_dst_dir = misc.find_file(self._base_dst_dir, - src_file_name, - os.path.getsize(src_file_path), - exclude_dir=self._base_src_dir) + alt_dst_dir = misc.find_alt_file(self._base_dst_dir, + src_file_name, + os.path.getsize(src_file_path), + meta_time, + exclude_dir=self._base_src_dir) if alt_dst_dir: dst_dir = alt_dst_dir dst_file_path = os.path.join(dst_dir, src_file_name) diff --git a/src/misc.py b/src/misc.py index 93690d1..5b331b5 100644 --- 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_alt_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 alt_name in files: + if alt_name == name: + full_path = os.path.join(root, alt_name) + if os.path.getsize(full_path) == size: + alt_meta_time = extract_meta_time(full_path) + if meta_time == alt_meta_time: + return root return None def import_file(src_file_path, dst_file_path):