X-Git-Url: http://git.treefish.org/~alex/photosort.git/blobdiff_plain/33289bef2b08748db1c8061ecf453b291d2dc337..d15281357553c1fe81f4408d44ade249fa1dcdd9:/src/photosort.py?ds=inline

diff --git a/src/photosort.py b/src/photosort.py
index 38bea1b..09f55d7 100755
--- a/src/photosort.py
+++ b/src/photosort.py
@@ -24,16 +24,17 @@ logging.basicConfig(format='[%(asctime)s] %(levelname)s: %(message)s',
 for src_file_name, src_file_path in misc.walk_media_files(args.SOURCE_DIR):
     logging.info('Processing %s...', src_file_name)
 
-    src_time = misc.extract_timestamp(src_file_path)
+    exif_time = misc.extract_timestamp(src_file_path, use_exif=True)
 
     dst_dir = os.path.join(args.DEST_DIR,
-                           datetime.datetime.fromtimestamp(src_time).strftime("%Y/%m"))
+                           datetime.datetime.fromtimestamp(exif_time).strftime("%Y/%m"))
     dst_file_path = os.path.join(dst_dir, src_file_name)
 
     if not os.path.exists(dst_file_path):
         alt_dst_dir = misc.find_file(args.DEST_DIR,
                                      src_file_name,
-                                     os.path.getsize(src_file_path))
+                                     os.path.getsize(src_file_path),
+                                     exclude_dir=args.SOURCE_DIR)
         if alt_dst_dir:
             dst_dir = alt_dst_dir
             dst_file_path = os.path.join(dst_dir, src_file_name)
@@ -43,6 +44,10 @@ for src_file_name, src_file_path in misc.walk_media_files(args.SOURCE_DIR):
             os.makedirs(dst_dir)
         misc.import_file(src_file_path, dst_file_path, move=args.cleanup)
     else:
+        src_time = misc.extract_timestamp(src_file_path)
         dst_time = misc.extract_timestamp(dst_file_path)
         if src_time > dst_time:
             misc.import_file(src_file_path, dst_file_path, move=args.cleanup)
+
+if args.cleanup:
+    misc.delete_dir_contents(args.SOURCE_DIR)