From 1cf6e64bfae6d8ad190cf1a2e496da57c178a93a Mon Sep 17 00:00:00 2001 From: Ethan Dalool Date: Fri, 27 Nov 2020 15:39:58 -0800 Subject: [PATCH] Add remove_tag command. --- frontends/etiquette_cli.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/frontends/etiquette_cli.py b/frontends/etiquette_cli.py index 884595f..3116743 100644 --- a/frontends/etiquette_cli.py +++ b/frontends/etiquette_cli.py @@ -126,7 +126,7 @@ def search_by_argparse(args, yield_albums=False, yield_photos=False): #################################################################################################### -def add_tag_argparse(args): +def add_remove_tag_argparse(args, action): photodb = find_photodb() tag = photodb.get_tag(name=args.tag_name) @@ -138,7 +138,10 @@ def add_tag_argparse(args): photos = search_in_cwd(yield_photos=True, yield_albums=False) for photo in photos: - photo.add_tag(tag) + if action == 'add': + photo.add_tag(tag) + elif action == 'remove': + photo.remove_tag(tag) if args.autoyes or getpermission.getpermission('Commit?'): photodb.commit() @@ -299,7 +302,13 @@ def main(argv): p_add_tag.add_argument('tag_name') p_add_tag.add_argument('globs', nargs='*') p_add_tag.add_argument('--yes', dest='autoyes', action='store_true') - p_add_tag.set_defaults(func=add_tag_argparse) + p_add_tag.set_defaults(func=lambda args: add_remove_tag_argparse(args, action='add')) + + p_remove_tag = subparsers.add_parser('remove_tag', aliases=['remove-tag']) + p_remove_tag.add_argument('tag_name') + p_remove_tag.add_argument('globs', nargs='*') + p_remove_tag.add_argument('--yes', dest='autoyes', action='store_true') + p_remove_tag.set_defaults(func=lambda args: add_remove_tag_argparse(args, action='remove')) p_easybake = subparsers.add_parser('easybake') p_easybake.add_argument('eb_strings', nargs='+')