Changeset 162f527 for remit/finance_core/reporting.py

Timestamp:

Mar 27, 2010, 3:48:00 AM (15 years ago)

Author:

Alex Dehnert <adehnert@…>

Branches:

master, client

Children:

eae6ea8

Parents:

83d9608

git-author:

Alex Dehnert <adehnert@…> (03/27/10 03:48:00)

git-committer:

Alex Dehnert <adehnert@…> (03/27/10 03:48:00)

Message:

Start of giving options for the axis

File:

• remit/finance_core/reporting.py (modified) (4 diffs)

remit/finance_core/reporting.py

@@ -1 +1 @@
 from decimal import Decimal
-from django.db.models import Sum
+from django.db.models import Q, Sum
+import finance_core.models
 
 def build_table_annotate(line_items, main_lineitem_limit_primary, primary_axis, primary_axis_objs, secondary_axis, ):
@@ -7 +8 @@
     table = []
     zero = Decimal('0.00')
-    for num, (pk, label, qobj, ) in enumerate(primary_axis):
+    for num, (pk, label, qobj, objrel_qobj) in enumerate(primary_axis):
         arcprimary[pk] = num
         table.append([zero]*len(secondary_axis))
@@ -17 +18 @@
     # Do the real work
     print secondary_axis
-    for num, (label, qobj_lineitem, qobj_primary) in enumerate(secondary_axis):
+    for num, (pk, label, qobj_lineitem, qobj_primary) in enumerate(secondary_axis):
         print num
         secondary_results = (primary_axis_objs.filter(main_lineitem_limit_primary, qobj_primary, ).annotate(Sum('lineitem__amount')))
@@ -37 +38 @@
     table = [ # Primary axis
             [ # Secondary axis
-                total_amount(line_items.filter(primary[2], secondary[1]))
+                total_amount(line_items.filter(primary[2], secondary[2]))
             for secondary in secondary_axis]
         for primary in primary_axis]
 
     return table
-    
 
 build_table = build_table_annotate
+
+
+def get_primary_axis(slug, base_area):
+    if slug == 'budget-areas':
+        return get_budget_areas(base_area)
+    else:
+        raise UnsupportedOperationException()
+
+def get_secondary_axis(slug, base_area):
+    if slug == 'budget-areas':
+        return get_budget_areas(base_area)
+    elif slug == 'layers':
+        return get_layers(base_area)
+    else:
+        raise NotImplementedError
+
+def get_budget_areas(base_area):
+    name = 'Budget Areas'
+    base_area_depth = base_area.depth
+    axis = [
+        (
+            area.pk,
+            area.indented_name(base_area_depth),
+            Q(budget_area=area),
+            Q(lineitem__budget_area=area),
+        )
+        for area in base_area.get_descendants()
+    ]
+    axis_objs = base_area.get_descendants()
+    return name, axis, axis_objs,
+
+def get_layers(base_area):
+    name = 'Layers'
+    axis = [
+        (
+            finance_core.models.layer_num(layer),
+            finance_core.models.layer_name(layer),
+            Q(layer=finance_core.models.layer_num(layer)),
+            Q(lineitem__layer=finance_core.models.layer_num(layer)),
+        )
+        for layer in finance_core.models.layers
+    ]
+    return name, axis, None,
+