From 071711dbdbfd443dff46c88019244659fa474c83 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lu=C3=ADs=20Murta?= Date: Fri, 30 Dec 2022 16:09:40 +0000 Subject: [PATCH] Fix CategoryRule and TagRule initializations and re-adds the remove option to the modify category rule cli. --- pfbudget/__main__.py | 4 ++-- pfbudget/cli/runnable.py | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) 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):