clean registry only once after migration

[photosort.git] / src / migrator.py

diff --git a/src/migrator.py b/src/migrator.py
index 1bfaacdafe7f7574fa80f8668c16a8962af9a6ed..d48022698b00e7e59ec908d872ab475ea4d11c1b 100644 (file)
--- a/src/migrator.py
+++ b/src/migrator.py
@@ -3,6 +3,7 @@ import logging
 import os
 
 import misc
+from registry import Registry
 
 class Migrator:
     def __init__(self, src_dir, dst_dir, reg_db=None):
@@ -16,12 +17,12 @@ class Migrator:
             try:
                 if not self._reg.is_registered(src_file_path):
                     self._migrate_single(src_file_name, src_file_path, remove)
-                    self._reg.clean()
                     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))
+        self._reg.clean()
 
     def _migrate_single(self, src_file_name, src_file_path, remove):
         meta_time = misc.extract_meta_time(src_file_path)