personal-finance/budget
2
1
Fork 0
Code Issues 11 Pull Requests Projects 6 Releases 2 Wiki Activity

fix: add category/tag rules

Browse Source
This commit is contained in:
Luís Murta 2025-10-09 22:05:41 +01:00
parent aab728392c
commit 7b4815dee5
Signed by: satprog
GPG Key ID: 169EF1BBD7049F94
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
pfbudget/__main__.py
View File

@ -161,9 +161,9 @@ if __name__ == "__main__":
} }
assert args.keys() >= keys, f"missing {args.keys() - keys}" assert args.keys() >= keys, f"missing {args.keys() - keys}"
params = [ params = []
type.CategoryRule( for cat in args["category"]:
cat, rule = type.CategoryRule(
start=args["start"][0] if args["start"] else None, start=args["start"][0] if args["start"] else None,
end=args["end"][0] if args["end"] else None, end=args["end"][0] if args["end"] else None,
description=args["description"][0] if args["description"] else None, description=args["description"][0] if args["description"] else None,
@ -172,8 +172,9 @@ if __name__ == "__main__":
min=args["min"][0] if args["min"] else None, min=args["min"][0] if args["min"] else None,
max=args["max"][0] if args["max"] else None, max=args["max"][0] if args["max"] else None,
) )
for cat in args["category"] # associate the rule to the category name (mapped col has init=False)
] rule.name = cat
params.append(rule)
case Operation.RuleRemove | Operation.TagRuleRemove: case Operation.RuleRemove | Operation.TagRuleRemove:
keys = {"id"} keys = {"id"}
@ -213,9 +214,9 @@ if __name__ == "__main__":
keys = {"tag", "start", "end", "description", "regex", "bank", "min", "max"} keys = {"tag", "start", "end", "description", "regex", "bank", "min", "max"}
assert args.keys() >= keys, f"missing {args.keys() - keys}" assert args.keys() >= keys, f"missing {args.keys() - keys}"
params = [ params = []
type.TagRule( for tag in args["tag"]:
tag, rule = type.TagRule(
start=args["start"][0] if args["start"] else None, start=args["start"][0] if args["start"] else None,
end=args["end"][0] if args["end"] else None, end=args["end"][0] if args["end"] else None,
description=args["description"][0] if args["description"] else None, description=args["description"][0] if args["description"] else None,
@ -224,8 +225,9 @@ if __name__ == "__main__":
min=args["min"][0] if args["min"] else None, min=args["min"][0] if args["min"] else None,
max=args["max"][0] if args["max"] else None, max=args["max"][0] if args["max"] else None,
) )
for tag in args["tag"] # associate the rule to the tag name (mapped col has init=False)
] rule.tag = tag
params.append(rule)
case Operation.TagRuleModify: case Operation.TagRuleModify:
keys = {"id", "tag", "date", "description", "bank", "min", "max", "remove"} keys = {"id", "tag", "date", "description", "bank", "min", "max", "remove"}