Changeset dd2c3d9 for remit/finance_core/reporting.py

Timestamp:

Mar 27, 2010, 2:47:54 PM (15 years ago)

Author:

Alex Dehnert <adehnert@…>

Branches:

master, client

Children:

8a0d18c

Parents:

ca03565

git-author:

Alex Dehnert <adehnert@…> (03/27/10 14:47:54)

git-committer:

Alex Dehnert <adehnert@…> (03/27/10 14:47:54)

Message:

Allow filtering by term

File:

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

remit/finance_core/reporting.py

@@ -45 +45 @@
 
 
-def get_primary_axis(slug, base_area):
+def get_primary_axis(slug, base_area, term, ):
     if slug in axes and axes[slug][1]:
-        return axes[slug][0](base_area)
+        return axes[slug][0](base_area, term, )
     else:
         raise NotImplementedError
 
-def get_secondary_axis(slug, base_area):
+def get_secondary_axis(slug, base_area, term, ):
     if slug in axes and axes[slug][2]:
-        return axes[slug][0](base_area)
+        return axes[slug][0](base_area, term, )
     else:
         raise NotImplementedError
 
-def get_budget_areas(base_area):
+def get_budget_areas(base_area, term, ):
     name = 'Budget Areas'
     base_area_depth = base_area.depth
+    areas = base_area.get_descendants()
+    if term:
+        areas = areas.filter(Q(always=True) or Q(budget_term=term))
     axis = [
         (
@@ -67 +70 @@
             Q(lineitem__budget_area=area),
         )
-        for area in base_area.get_descendants()
+        for area in areas
     ]
-    axis_objs = base_area.get_descendants()
+    axis_objs = areas
     return name, axis, axis_objs,
 
-def get_budget_terms(base_area):
+def get_budget_terms(base_area, term, ):
     name = 'Budget Terms'
-    terms = finance_core.models.BudgetTerm.objects.all()
+    if term:
+        terms = finance_core.models.BudgetTerm.objects.filter(pk=term.pk)
+    else:
+        terms = finance_core.models.BudgetTerm.objects.all()
     axis = [
         (
@@ -86 +92 @@
     return name, axis, terms
 
-def get_layers(base_area):
+def get_layers(base_area, term, ):
     name = 'Layers'
     axis = [