First step to adding filters to accounts.

2 files changed, 7 insertions, 6 deletions

diff --git a/ledger.py b/ledger.py
index a418e65..d5e9b72 100755
--- 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
index e39d2fe..f479628 100644
--- 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???