add missing arguments

[photosort.git] / src / photosort.py

diff --git a/src/photosort.py b/src/photosort.py
index 8ed4ee7120fe77e49b46de4829b198fa9823aa13..872724ee49a8a688b234b12556515f221b15f510 100755 (executable)
--- a/src/photosort.py
+++ b/src/photosort.py
@@ -4,7 +4,7 @@ import argparse
 import datetime
 import logging
 
-from dirhandler import DirHandler
+from migrator import Migrator
 
 parser = argparse.ArgumentParser(description='Process some integers.')
 parser.add_argument('SOURCE_DIR', type=str, help='source directory')
@@ -28,7 +28,11 @@ else:
 logging.basicConfig(format='[%(asctime)s] %(levelname)s: %(message)s',
                     level=log_level, datefmt='%m/%d/%Y %H:%M:%S')
 
-handler = DirHandler(args.SOURCE_DIR, args.DEST_DIR)
-handler.migrate(args.cleanup)
+migrator = Migrator(args.SOURCE_DIR, args.DEST_DIR)
+
+migrator.migrate(args.cleanup)
+
 if args.cleanup:
-    handler.cleanup()
+    migrator.cleanup()
+
+migrator.close()