commit 269eee248d24acbb12f13ab519dfbe1c881d3bb8
parent 8aa967f32d8e7ca6da2d759bd49f0bb4ee2021b0
Author: Adrián Oliva <[email protected]>
Date: Mon, 8 May 2023 19:22:43 -0600
First step to adding filters to accounts.
Diffstat:
2 files changed, 7 insertions(+), 6 deletions(-)
diff --git a/ledger.py b/ledger.py
@@ -17,14 +17,14 @@ def main():
if args.sort:
sort_entries(result, args.sort)
- if args.verb == 'print':
+ if args.verb[0] == 'print':
for ent in result:
print(ent)
- elif args.verb in ['register', 'reg', 'r']:
+ elif args.verb[0] in ['register', 'reg', 'r']:
print_register(result)
- elif args.verb in ['balance', 'bal', 'b']:
+ elif args.verb[0] in ['balance', 'bal', 'b']:
print_balance(result)
diff --git a/utils/args.py b/utils/args.py
@@ -14,7 +14,8 @@ def get_arguments():
parser.add_argument('-v', '--version',
action='version', version='%(prog)s 0.1')
- parser.add_argument('verb', metavar='Action', action='store',
+ parser.add_argument('verb', metavar='Action',
+ action='store', nargs='+',
help='Specify an action between balance, report and \
print.')
parser.add_argument('-S', '--sort', metavar='value-expression',
@@ -40,8 +41,8 @@ def test_args(my_args):
'register', 'reg', 'r',
'print'
]
- if my_args.verb not in valid_verbs:
- raise Exception(f'{my_args.verb} is NOT a valid verb! Valid verbs are: {", ".join(valid_verbs)}.')
+ if my_args.verb[0] not in valid_verbs:
+ raise Exception(f'{my_args.verb[0]} is NOT a valid verb! Valid verbs are: {", ".join(valid_verbs)}.')
# Test if expression for sorting is valid.
# TODO: How can we validate an expression???
