Add VAT summary

1 files changed, 16 insertions, 1 deletions

diff --git a/fracture/invoice.py b/fracture/invoice.py
index 295b7a8..0ccfec2 100644
--- a/fracture/invoice.py
+++ b/fracture/invoice.py
@@ -363,6 +363,20 @@ class Invoice:
     def total(self):
         return sum(p.total           for p in self.products) + \
                sum(self.base*t.ratio for t in self.taxes)
+    def summarize_vat(self):
+        pvat = lambda p: p.vat
+        vs = groupby(sorted(self.products, key=pvat), key=pvat)
+
+        res = ()
+
+        for vat, ps in vs:
+            ps = tuple(ps)
+            res += ({
+                    "vat":      vat,
+                    "base":     Conf.round(sum(p.base        for p in ps)),
+                    "charged":  Conf.round(sum(p.charged_vat for p in ps)),
+            },)
+        return res
 
     def to_dict(self):
         return {
@@ -379,7 +393,8 @@ class Invoice:
             "base":              self.base,
             "notes":             self.notes,
             "total":             Conf.round(self.total),
-            "vat-charged":       self.vat_charged
+            "vat-charged":       self.vat_charged,
+            "vat-summary":       self.summarize_vat()
         }
     def to_json(self):
         return json.dumps(self.to_dict())