Changes in bin/stamps [10:d3a34cb9b65f:3:4eb41ee30ee9] in stamper

File:

• bin/stamps (modified) (1 diff)

bin/stamps

@@ -1 +1 @@
 #!/usr/bin/env python
 
-import argparse
+import sys
 from stamper import Stamper
+
+
+def usage(name):
+    msg = """
+    Usage:
+
+    To gather information of recorded times: %(name)s
+
+    Recorded times for a given customer: %(name)s customer
+
+    Detailed times for a given customer: %(name)s customer details
+    """ % {'name': name}
+    print(msg)
 
 
 if __name__ == '__main__':
 
-    parser = argparse.ArgumentParser(
-        description='stamps: show recorded times information')
-    parser.add_argument('customer', action="store", nargs='?',
-                        help='Show times only for this customer')
-    parser.add_argument('filter', action="store", nargs='?',
-                        help='Filter the times by date range')
-    parser.add_argument('-v', '--verbose', action="store_true",
-                        help='Include detailed times information')
-
-    args = parser.parse_args()
-
     s = Stamper()
     s.load_stamps()
-    s.show_stamps(args.customer, args.filter, args.verbose)
+
+    if len(sys.argv) == 1:
+        totals = s.totals()
+        print('Total times for all customers:')
+        for t in totals:
+            print(' -> %(customer)s: %(total)s' % {'customer': t,
+                                                   'total': totals[t]})
+    elif len(sys.argv) == 2:
+        if sys.argv[1] in s.customers():
+            print('Total time for %(customer)s: %(total)s' % {
+                'customer': sys.argv[1],
+                'total': s.total_by_customer(sys.argv[1])
+                })
+
+    elif len(sys.argv) == 3:
+        if sys.argv[1] in s.customers():
+            print('Detailed time for %(customer)s' % {
+                'customer': sys.argv[1],
+                })
+            details, totals = s.details_by_customer(sys.argv[1])
+            for day in details:
+                print('------ %(day)s ------' % {'day': day})
+                for line in details[day]:
+                    print(line)
+                print(' Total: %(total)s' % {'total': totals[day]})
+
+    else:
+        usage(sys.argv[0])
+        sys.exit(1)