added missing import

[photosort.git] / src / bunch.py

diff --git a/src/bunch.py b/src/bunch.py
index 228131a9e1ec5e6b8157f86813b9fbffceb15634..2b98734f46f5cb9fa295a959cca4b10ea96f01fc 100644 (file)
--- a/src/bunch.py
+++ b/src/bunch.py
@@ -1,4 +1,5 @@
 import logging
+import os
 import time
 import threading
 
@@ -23,7 +24,8 @@ class Bunch:
                     source_idx,
                     DirTrigger(src_dir_cfg['path'], src_dir_cfg['cool_time'],
                                src_dir_cfg['max_time']),
-                    Migrator(src_dir_cfg['path'], cfg['dst_dir']['path']),
+                    Migrator(src_dir_cfg['path'], cfg['dst_dir']['path'],
+                             os.path.join(cfg['cache_dir'], "%d.db" % idx)),
                     src_dir_cfg['cleanup']
                 )
             )
@@ -40,6 +42,8 @@ class Bunch:
         logging.info("Stopping bunch #%d...", self._idx)
         self._stop = True
         self._worker_thread.join()
+        for src in self._sources:
+            src.migrator.close()
         logging.info("Stopped bunch #%d.", self._idx)
 
     def is_running(self):