]> git.treefish.org Git - photosort.git/blobdiff - src/migrator.py
add missing import
[photosort.git] / src / migrator.py
index b406689346d3da777233da4b60174061f1d6b010..c2fde409fefad59aa9fe936f65c58553b5189fa3 100644 (file)
@@ -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,6 +17,7 @@ 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)