diff --git a/pfbudget/__main__.py b/pfbudget/__main__.py index c62c96f..baa1049 100644 --- a/pfbudget/__main__.py +++ b/pfbudget/__main__.py @@ -57,13 +57,13 @@ if __name__ == "__main__": params = [ pfbudget.types.CategoryRule( - cat, args["date"][0] if args["date"] else None, args["description"][0] if args["description"] else None, args["regex"][0] if args["regex"] else None, args["bank"][0] if args["bank"] else None, args["min"][0] if args["min"] else None, args["max"][0] if args["max"] else None, + cat, ) for cat in args["category"] ] @@ -109,13 +109,13 @@ if __name__ == "__main__": params = [ pfbudget.types.TagRule( - tag, args["date"][0] if args["date"] else None, args["description"][0] if args["description"] else None, args["regex"][0] if args["regex"] else None, args["bank"][0] if args["bank"] else None, args["min"][0] if args["min"] else None, args["max"][0] if args["max"] else None, + tag, ) for tag in args["tag"] ] diff --git a/pfbudget/cli/runnable.py b/pfbudget/cli/runnable.py index aa780f2..bdb1cfd 100644 --- a/pfbudget/cli/runnable.py +++ b/pfbudget/cli/runnable.py @@ -376,6 +376,7 @@ def category_rule(parser: argparse.ArgumentParser, universal: argparse.ArgumentP modify.add_argument("id", nargs="+", type=int) modify.add_argument("--category", nargs=1, type=str) rules(modify) + modify.add_argument("--remove", nargs="*", default=[], type=str) def tags(parser: argparse.ArgumentParser, universal: argparse.ArgumentParser):