Changeset 23:bc0b04a989aa in stamper

Timestamp:

Jul 24, 2014, 2:52:30 AM (10 years ago)

Author:

Óscar M. Lage <info@…>

Branch:

default

Phase:

public

Message:

Graphs, first version (limited to 15 days, without customer the graph order is inverse)

Files:

• .hgignore (modified) (1 diff)
• bin/stamps (modified) (2 diffs)
• stamper/stamper.py (modified) (8 diffs)

.hgignore

@@ -6 +6 @@
 stamper.egg-info/
 dist/
+graphs/
 

bin/stamps

@@ -17 +17 @@
     parser.add_argument('-s', '--sum', action="store_true",
                         help='Include sum of times')
+    parser.add_argument('-g', '--graph', action="store_true",
+                        help='Generate a SVG graph')
 
     args = parser.parse_args()
@@ -22 +24 @@
     s = Stamper()
     s.load_stamps()
-    s.show_stamps(args.customer, args.filter, args.verbose, args.sum)
+    s.show_stamps(args.customer, args.filter, args.verbose, args.sum,
+        args.graph)

stamper/stamper.py

@@ -2 +2 @@
 import json
 import re
+import pygal
 from datetime import datetime, timedelta
 from os.path import expanduser, exists
@@ -110 +111 @@
         details = OrderedDict()
         totals = OrderedDict()
+        total_customer = OrderedDict()
         for stamp in self.stamps:
             if stamp['customer']:
@@ -125 +127 @@
                         'action': stamp['action']
                     })
+                customer = stamp['customer']
                 totals[start_day] += worktime
+                if start_day not in total_customer:
+                    total_customer[start_day] = {}
+                if customer not in total_customer[start_day]:
+                    total_customer[start_day][customer] = 0
+                total_customer[start_day][customer] += worktime
         for day in totals:
             totals[day] = str(timedelta(seconds=totals[day]))
-        return details, totals
+        return details, totals, total_customer
 
     def details_by_customer(self, customer):
@@ -153 +161 @@
 
     def show_stamps(self, customer=None, stamp_filter=None, verbose=False,
-        sum=False):
+        sum=False, graph=False):
         if stamp_filter:
             stamp_filter = self.validate_filter(stamp_filter)
@@ -161 +169 @@
         if customer:
             seconds=totals.get(customer, 0)
-            total = timedelta(seconds)
+            total = timedelta(seconds=totals.get(customer, 0))
             print(' %(customer)s: %(total)s' % {'customer': customer,
                                                 'total': total})
@@ -167 +175 @@
             for c in totals:
                 seconds=totals[c]
-                total = timedelta(seconds)
+                total = timedelta(seconds=totals[c])
                 print(' %(customer)s: %(total)s' % {'customer': c,
                                                     'total': total})
@@ -175 +183 @@
                 details, totals = self.details_by_customer(customer)
             else:
-                details, totals = self.details()
+                details, totals, total_customer = self.details()
             for day in details:
                 print('------ %(day)s ------' % {'day': day})
@@ -201 +209 @@
                     totalDays, remainingHr, remainingMin, HOURS_DAY
                 ))
+
+        if graph:
+            DAYS = 15
+            list_days = []
+            list_tot = []
+            stackedbar_chart = pygal.StackedBar()
+            stackedbar_chart.title = 'Worked time per day (in hours)'
+
+            if customer:
+                for day, tot in totals.iteritems():
+                    list_days.append(day)
+                    (h, m, s) = tot.split(':')
+                    tot_sec = int(h) * 3600 + int(m) * 60 + int(s)
+                    tot_h = float(tot_sec / float(60) / float(60))
+                    list_tot.append(tot_h)
+                stackedbar_chart.add(customer, list_tot)
+                stackedbar_chart.x_labels = map(str, list_days)
+                stackedbar_chart.render_to_file('graphs/chart-%s.svg' % customer )
+            else:
+                all_customers = self.customers()
+                total_per_customer = {}
+                details, totals, total_customer = self.details()
+                chars = 0
+                total_customer_reverse = total_customer.items()
+                total_customer_reverse.reverse()
+                for day, tot in total_customer_reverse:
+                    if chars < DAYS:
+                        list_days.append(day)
+                        for cust in self.customers():
+                            if cust not in tot:
+                                tot[cust] = 0
+                        for cus, time in tot.iteritems():
+                                tot_h = float(time / float(60) / float(60))
+                                if cus not in total_per_customer:
+                                    total_per_customer[cus] = []
+                                total_per_customer[cus].append(tot_h)
+                    chars = chars + 1
+                for ccus, ctime in total_per_customer.iteritems():
+                    stackedbar_chart.add(ccus, ctime)
+                stackedbar_chart.x_labels = map(str, list_days[-DAYS:])
+                stackedbar_chart.render_to_file('graphs/chart-all.svg')