X-Git-Url: http://git.treefish.org/~alex/photosort.git/blobdiff_plain/5ad2c3b2ac3baf7b9e64d07a7f0b37d3493a03d4..0038b30e2f17a1726920d2d774456205d7d8b4c7:/src/migrator.py?ds=sidebyside

diff --git a/src/migrator.py b/src/migrator.py
index 7bd32e6..b406689 100644
--- a/src/migrator.py
+++ b/src/migrator.py
@@ -5,15 +5,20 @@ import os
 import misc
 
 class Migrator:
-    def __init__(self, src_dir, dst_dir):
+    def __init__(self, src_dir, dst_dir, reg_db=None):
         self._base_src_dir = src_dir
         self._base_dst_dir = dst_dir
+        self._reg = Registry(reg_db)
 
     def migrate(self, remove):
         for src_file_name, src_file_path in misc.walk_media_files(self._base_src_dir):
             logging.debug('Migrating %s...', src_file_name)
             try:
-                self._migrate_single(src_file_name, src_file_path, remove)
+                if not self._reg.is_registered(src_file_path):
+                    self._migrate_single(src_file_name, src_file_path, remove)
+                    self._reg.register(src_file_path)
+                else:
+                    self._reg.refresh(src_file_path)
             except Exception as e:
                 logging.error('Error migrating %s: %s', src_file_path, str(e))