jmathai / elodie

    return True

@click.command('update')

@click.option('--album', help='Update the image album.')

@click.option('--location', help=('Update the image location. Location '

                                  'should be the name of a place, like "Las '

                                  'Vegas, NV".'))

@click.option('--time', help=('Update the image time. Time should be in '

                              'YYYY-mm-dd hh:ii:ss or YYYY-mm-dd format.'))

@click.option('--title', help='Update the image title.')

@click.option('--debug', default=False, is_flag=True,

              help='Override the value in constants.py with True.')

@click.argument('paths', nargs=-1,

                required=True)

def _update(album, location, time, title, paths, debug):

    """Update a file's EXIF. Automatically modifies the file's location and file name accordingly.

    """

    constants.debug = debug

    has_errors = False

    result = Result()

    files = set()

    for path in paths:

        path = os.path.expanduser(path)

        if os.path.isdir(path):

            files.update(FILESYSTEM.get_all_files(path, None))

        else:

            files.add(path)

    for current_file in files:

        if not os.path.exists(current_file):

            has_errors = True

            result.append((current_file, False))

            log.warn('Could not find %s' % current_file)

            log.all('{"source":"%s", "error_msg":"Could not find %s"}' %

                      (current_file, current_file))

            continue

        current_file = os.path.expanduser(current_file)

        # The destination folder structure could contain any number of levels

        #  So we calculate that and traverse up the tree.

        # '/path/to/file/photo.jpg' -> '/path/to/file' ->

        #  ['path','to','file'] -> ['path','to'] -> '/path/to'

        current_directory = os.path.dirname(current_file)

        destination_depth = -1 * len(FILESYSTEM.get_folder_path_definition())

        destination = os.sep.join(

                          os.path.normpath(

                              current_directory

                          ).split(os.sep)[:destination_depth]

                      )

        media = Media.get_class_by_file(current_file, get_all_subclasses())

        if not media:

            continue

        updated = False

        if location:

            update_location(media, current_file, location)

            updated = True

        if time:

            update_time(media, current_file, time)

            updated = True

        if album:

            media.set_album(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 title:

            # We call get_metadata() to cache it before making any changes

            metadata = media.get_metadata()

            title_update_status = media.set_title(title)

            original_title = metadata['title']

            if title_update_status and original_title:

                # @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(current_file,

                                                    get_all_subclasses())

            # 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, ''))

            dest_path = FILESYSTEM.process_file(current_file, destination,

                updated_media, move=True, allowDuplicate=True)

            log.info(u'%s -> %s' % (current_file, dest_path))

            log.all('{"source":"%s", "destination":"%s"}' % (current_file,

                                                               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(current_file))

            FILESYSTEM.delete_directory_if_empty(

                os.path.dirname(os.path.dirname(current_file)))

            result.append((current_file, dest_path))

            # Trip has_errors to False if it's already False or dest_path is.

            has_errors = has_errors is True or not dest_path

        else:

            has_errors = False

            result.append((current_file, False))

    result.write()

    if has_errors:

        sys.exit(1)

@click.group()

    return True

@click.command('update')

@click.option('--album', help='Update the image album.')

@click.option('--location', help=('Update the image location. Location '

                                  'should be the name of a place, like "Las '

                                  'Vegas, NV".'))

@click.option('--time', help=('Update the image time. Time should be in '

                              'YYYY-mm-dd hh:ii:ss or YYYY-mm-dd format.'))

@click.option('--title', help='Update the image title.')

@click.option('--debug', default=False, is_flag=True,

              help='Override the value in constants.py with True.')

@click.argument('paths', nargs=-1,

                required=True)

def _update(album, location, time, title, paths, debug):

    """Update a file's EXIF. Automatically modifies the file's location and file name accordingly.

    """

    constants.debug = debug

    has_errors = False

    result = Result()

    files = set()

    for path in paths:

        path = os.path.expanduser(path)

        if os.path.isdir(path):

            files.update(FILESYSTEM.get_all_files(path, None))

        else:

            files.add(path)

    for current_file in files:

        if not os.path.exists(current_file):

            has_errors = True

            result.append((current_file, False))

            log.warn('Could not find %s' % current_file)

            log.all('{"source":"%s", "error_msg":"Could not find %s"}' %

                      (current_file, current_file))

            continue

        current_file = os.path.expanduser(current_file)

        # The destination folder structure could contain any number of levels

        #  So we calculate that and traverse up the tree.

        # '/path/to/file/photo.jpg' -> '/path/to/file' ->

        #  ['path','to','file'] -> ['path','to'] -> '/path/to'

        current_directory = os.path.dirname(current_file)

        destination_depth = -1 * len(FILESYSTEM.get_folder_path_definition())

        destination = os.sep.join(

                          os.path.normpath(

                              current_directory

                          ).split(os.sep)[:destination_depth]

                      )

        media = Media.get_class_by_file(current_file, get_all_subclasses())

        if not media:

            continue

        updated = False

        if location:

            update_location(media, current_file, location)

            updated = True

        if time:

            update_time(media, current_file, time)

            updated = True

        if album:

            media.set_album(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 title:

            # We call get_metadata() to cache it before making any changes

            metadata = media.get_metadata()

            title_update_status = media.set_title(title)

            original_title = metadata['title']

            if title_update_status and original_title:

                # @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(current_file,

                                                    get_all_subclasses())

            # 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, ''))

            dest_path = FILESYSTEM.process_file(current_file, destination,

                updated_media, move=True, allowDuplicate=True)

            log.info(u'%s -> %s' % (current_file, dest_path))

            log.all('{"source":"%s", "destination":"%s"}' % (current_file,

                                                               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(current_file))

            FILESYSTEM.delete_directory_if_empty(

                os.path.dirname(os.path.dirname(current_file)))

            result.append((current_file, dest_path))

            # Trip has_errors to False if it's already False or dest_path is.

            has_errors = has_errors is True or not dest_path

        else:

            has_errors = False

            result.append((current_file, False))

    result.write()

    if has_errors:

        sys.exit(1)

@click.group()