From 816a95d0bc29c07118efff196b17c7229be295a4 Mon Sep 17 00:00:00 2001 From: Jaisen Mathai Date: Sat, 31 Oct 2015 01:31:35 -0700 Subject: [PATCH] Combining import.py and update.py into elodie.py --- .gitignore | 3 ++ README.md | 1 + elodie.py | 153 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 157 insertions(+) create mode 100755 elodie.py diff --git a/.gitignore b/.gitignore index 6236fcb..ac687d3 100644 --- a/.gitignore +++ b/.gitignore @@ -2,3 +2,6 @@ **/*.pyc **/config.ini **/node_modules/** +dist/** +build/** +elodie.spec diff --git a/README.md b/README.md index 21bd45b..c155efe 100644 --- a/README.md +++ b/README.md @@ -161,6 +161,7 @@ The commands on this page assume you're running them from the root of this repos ``` pip install LatLon pip install requests +pip install docopt ``` You'll need to install *exiftool* *pyexiv2* using `homebrew` on OSX. If you're running another operating system you're sort of on your own but my pal Google should be able to help. Some folks may be able to simply run these commands. Installing *boost* is a drag and can take up to 30 minutes. Don't say I didn't warn you. diff --git a/elodie.py b/elodie.py new file mode 100755 index 0000000..8ef5202 --- /dev/null +++ b/elodie.py @@ -0,0 +1,153 @@ +#!/usr/bin/env python + +import os +import pyexiv2 +import re +import shutil +import sys +import time + +from datetime import datetime +from docopt import docopt + +from elodie import arguments +from elodie import constants +from elodie import geolocation +from elodie.media.photo import Media +from elodie.media.photo import Photo +from elodie.media.video import Video +from elodie.filesystem import FileSystem +from elodie.localstorage import Db + +def usage(): + return """Usage: main.py import --source= --destination= + main.py import --file= --destination= + main.py update [--time=] [--location=] [--album=] [--title=] INPUT ... + + -h --help show this + """ + +def _import(params): + destination = params['--destination'] + + if(params['--source'] is not None): + source = params['--source'] + + for current_file in filesystem.get_all_files(source, media_type.get_valid_extensions()): + media = Media.get_class_by_file(current_file, [Photo, Video]) + if(media is None): + continue + + if(media_type.__name__ == 'Video'): + filesystem.set_date_from_path_video(media) + + dest_path = filesystem.process_file(current_file, destination, media, allowDuplicate=False, move=False) + if(dest_path is not None): + print '%s -> %s' % (current_file, dest_path) + elif(params['--file'] is not None): + current_file = params['--file'] + media = Media.get_class_by_file(current_file) + if(media.__name__ == 'Video'): + filesystem.set_date_from_path_video(media) + + dest_path = filesystem.process_file(current_file, destination, media, allowDuplicate=False, move=False) + if(dest_path is not None): + print '%s -> %s' % (current_file, dest_path) + +def _update(params): + location_coords = None + for file_path in params['INPUT']: + if(not os.path.exists(file_path)): + if(constants.debug == True): + print 'Could not find %s' % file_path + print '{"source":"%s", "error_msg":"Could not find %s"}' % (file_path, file_path) + continue + + destination = os.path.dirname(os.path.dirname(os.path.dirname(file_path))) + + media = Media.get_class_by_file(file_path, [Photo, Video]) + if(media is None): + continue + + updated = False + if(params['--location'] is not None): + if(location_coords is None): + location_coords = geolocation.coordinates_by_name(params['--location']) + + if(location_coords is not None and 'latitude' in location_coords and 'longitude' in location_coords): + location_status = media.set_location(location_coords['latitude'], location_coords['longitude']) + if(location_status != True): + if(constants.debug == True): + print 'Failed to update location' + print '{"source":"%s", "error_msg":"Failed to update location"}' % file_path + sys.exit(1) + updated = True + + + if(params['--time'] is not None): + time_string = params['--time'] + time_format = '%Y-%m-%d %H:%M:%S' + if(re.match('^\d{4}-\d{2}-\d{2}$', time_string)): + time_string = '%s 00:00:00' % time_string + + if(re.match('^\d{4}-\d{2}-\d{2}$', time_string) is None and re.match('^\d{4}-\d{2}-\d{2} \d{2}:\d{2}\d{2}$', time_string)): + if(constants.debug == True): + print 'Invalid time format. Use YYYY-mm-dd hh:ii:ss or YYYY-mm-dd' + print '{"source":"%s", "error_msg":"Invalid time format. Use YYYY-mm-dd hh:ii:ss or YYYY-mm-dd"}' % file_path + sys.exit(1) + + if(time_format is not None): + time = datetime.strptime(time_string, time_format) + media.set_datetime(time) + updated = True + + if(params['--album'] is not None): + media.set_album(params['--album']) + updated = True + + # Updating a title can be problematic when doing it 2+ times on a file. + # You would end up with img_001.jpg -> img_001-first-title.jpg -> img_001-first-title-second-title.jpg. + # To resolve that we have to track the prior title (if there was one. + # Then we massage the updated_media's metadata['base_name'] to remove the old title. + # Since FileSystem.get_file_name() relies on base_name it will properly rename the file by updating the title + # instead of appending it. + remove_old_title_from_name = False + if(params['--title'] is not None): + # We call get_metadata() to cache it before making any changes + metadata = media.get_metadata() + title_update_status = media.set_title(params['--title']) + original_title = metadata['title'] + if(title_update_status and original_title is not None): + # @TODO: We should move this to a shared method since FileSystem.get_file_name() does it too. + original_title = re.sub('\W+', '-', original_title.lower()) + original_base_name = metadata['base_name'] + remove_old_title_from_name = True + updated = True + + if(updated == True): + updated_media = Media.get_class_by_file(file_path, [Photo, Video]) + # See comments above on why we have to do this when titles get updated. + if(remove_old_title_from_name is True and len(original_title) > 0): + updated_media.get_metadata() + updated_media.set_metadata_basename(original_base_name.replace('-%s' % original_title, '')) + + dest_path = filesystem.process_file(file_path, destination, updated_media, move=True, allowDuplicate=True) + if(constants.debug == True): + print '%s -> %s' % (file_path, dest_path) + + print '{"source":"%s", "destination":"%s"}' % (file_path, dest_path) + # If the folder we moved the file out of or its parent are empty we delete it. + filesystem.delete_directory_if_empty(os.path.dirname(file_path)) + filesystem.delete_directory_if_empty(os.path.dirname(os.path.dirname(file_path))) + + +db = Db() +filesystem = FileSystem() + +if __name__ == '__main__': + params = docopt(usage()) + if(params['import'] == True): + _import(params) + elif(params['update'] == True): + _update(params) + sys.exit(0)