ordigi/elodie.py

194 lines
6.6 KiB
Python
Raw Normal View History

#!/usr/bin/env python
import os
import re
import sys
from datetime import datetime
from docopt import docopt
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():
2015-12-31 00:29:13 +01:00
"""Return usage message
"""
return """
Usage: elodie.py import --source=<s> --destination=<d> [--album-from-folder]
elodie.py import --file=<f> --destination=<d> [--album-from-folder]
elodie.py update [--time=<t>] [--location=<l>] [--album=<a>] [--title=<t>] INPUT ...
-h --help show this
2015-12-31 00:29:13 +01:00
"""
DB = Db()
FILESYSTEM = FileSystem()
2015-12-31 00:29:13 +01:00
def import_file(_file, destination, album_from_folder):
"""Set file metadata and move it to destination.
"""
media = Media.get_class_by_file(_file, (Photo, Video))
if not media:
if constants.debug:
print 'Not a supported file (%s)' % _file
print '{"source":"%s", "error_msg":"Not a supported file"}' % _file
return
if media.__name__ == 'Video':
2015-12-31 00:29:13 +01:00
FILESYSTEM.set_date_from_path_video(media)
2015-12-31 00:29:13 +01:00
if album_from_folder:
media.set_album_from_folder()
2015-12-31 00:29:13 +01:00
dest_path = FILESYSTEM.process_file(_file, destination,
media, allowDuplicate=False, move=False)
if dest_path:
print '%s -> %s' % (_file, dest_path)
def _import(params):
2015-12-31 00:29:13 +01:00
"""Import files.
"""
destination = os.path.expanduser(params['--destination'])
if params['--source']:
source = os.path.expanduser(params['--source'])
2015-12-31 00:29:13 +01:00
files = FILESYSTEM.get_all_files(source, None)
elif params['--file']:
files = [os.path.expanduser(params['--file'])]
for current_file in files:
2015-12-31 00:29:13 +01:00
import_file(current_file, destination, params['--album-from-folder'])
2015-12-30 23:52:01 +01:00
def update_location(media, file_path, location_name):
"""Update location exif metadata of media.
"""
location_coords = geolocation.coordinates_by_name(location_name)
if location_coords and 'latitude' in location_coords and \
'longitude' in location_coords:
location_status = media.set_location(location_coords[
'latitude'], location_coords['longitude'])
if not location_status:
if constants.debug:
print 'Failed to update location'
print ('{"source":"%s",' % file_path,
'"error_msg":"Failed to update location"}')
sys.exit(1)
return True
def update_time(media, file_path, time_string):
"""Update time exif metadata of media.
"""
time_format = '%Y-%m-%d %H:%M:%S'
if re.match(r'^\d{4}-\d{2}-\d{2}$', time_string):
time_string = '%s 00:00:00' % time_string
elif re.match(r'^\d{4}-\d{2}-\d{2} \d{2}:\d{2}\d{2}$', time_string):
msg = ('Invalid time format. Use YYYY-mm-dd hh:ii:ss or YYYY-mm-dd')
if constants.debug:
print msg
print '{"source":"%s", "error_msg":"%s"}' % (file_path, msg)
sys.exit(1)
time = datetime.strptime(time_string, time_format)
media.set_date_taken(time)
return True
def _update(params):
2015-12-30 23:52:01 +01:00
"""Update files.
"""
for file_path in params['INPUT']:
if not os.path.exists(file_path):
if constants.debug:
print 'Could not find %s' % file_path
print '{"source":"%s", "error_msg":"Could not find %s"}' % \
(file_path, file_path)
continue
2015-12-02 23:07:23 +01:00
file_path = os.path.expanduser(file_path)
destination = os.path.expanduser(os.path.dirname(os.path.dirname(
os.path.dirname(file_path))))
2015-12-02 23:07:23 +01:00
media = Media.get_class_by_file(file_path, (Photo, Video))
if not media:
continue
updated = False
if params['--location']:
2015-12-30 23:52:01 +01:00
updated = update_location(media, file_path, params['--location'])
if params['--time']:
2015-12-30 23:52:01 +01:00
updated = update_time(media, file_path, params['--time'])
if params['--album']:
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']:
# 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:
2015-12-31 00:29:13 +01:00
# @TODO: We should move this to a shared method since
# FileSystem.get_file_name() does it too.
original_title = re.sub(r'\W+', '-', original_title.lower())
original_base_name = metadata['base_name']
remove_old_title_from_name = True
updated = True
if updated:
updated_media = Media.get_class_by_file(file_path, (Photo, Video))
2015-12-31 00:29:13 +01:00
# See comments above on why we have to do this when titles
# get updated.
if remove_old_title_from_name and len(original_title) > 0:
updated_media.get_metadata()
updated_media.set_metadata_basename(
original_base_name.replace('-%s' % original_title, ''))
2015-12-31 00:29:13 +01:00
dest_path = FILESYSTEM.process_file(file_path, destination,
updated_media, move=True, allowDuplicate=True)
if constants.debug:
2015-12-30 01:32:29 +01:00
print u'%s -> %s' % (file_path, dest_path)
2015-12-31 00:29:13 +01:00
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)))
2015-12-31 00:29:13 +01:00
def main(argv=None):
"""Main function call elodie subcommand on files.
"""
if argv is None:
argv = sys.argv
params = docopt(usage())
2015-12-30 01:32:29 +01:00
if params['import']:
_import(params)
2015-12-30 01:32:29 +01:00
elif params['update']:
_update(params)
sys.exit(0)
2015-12-31 00:29:13 +01:00
if __name__ == '__main__':
sys.exit(main())