1 | from django.http import HttpResponse |
---|
2 | import finance_core.models |
---|
3 | from django.contrib.auth.decorators import user_passes_test |
---|
4 | from django.shortcuts import render_to_response, get_object_or_404 |
---|
5 | from django.template import RequestContext |
---|
6 | from django.http import Http404, HttpResponseRedirect |
---|
7 | from django.db.models import Q |
---|
8 | from decimal import Decimal |
---|
9 | import finance_core.reporting |
---|
10 | |
---|
11 | def display_tree(request): |
---|
12 | root = finance_core.models.BudgetArea.get_by_path(['Accounts']) |
---|
13 | return HttpResponse(root.dump_to_html()) |
---|
14 | |
---|
15 | def reporting(request): |
---|
16 | line_items = finance_core.models.LineItem.objects.all() |
---|
17 | compute_method = 'default' |
---|
18 | if 'compute_method' in request.REQUEST: |
---|
19 | compute_method = request.REQUEST['compute_method'] |
---|
20 | |
---|
21 | # Main limit to lineitems, relative to primary axis |
---|
22 | main_lineitem_limits_primary = [] |
---|
23 | if 'term' in request.REQUEST and not request.REQUEST['term'] == 'all': |
---|
24 | term_obj = get_object_or_404(finance_core.models.BudgetTerm, slug=request.REQUEST['term']) |
---|
25 | term_name = term_obj.name |
---|
26 | line_items = line_items.filter(budget_term=term_obj) |
---|
27 | main_lineitem_limits_primary.append(Q(lineitem__budget_term=term_obj)) |
---|
28 | else: |
---|
29 | term_obj = None |
---|
30 | term_name = 'All' |
---|
31 | if 'area' in request.REQUEST: |
---|
32 | base_area_obj = get_object_or_404(finance_core.models.BudgetArea, pk=request.REQUEST['area']) |
---|
33 | else: |
---|
34 | base_area_obj = finance_core.models.BudgetArea.get_by_path(['Accounts']) |
---|
35 | all_relevant_areas = base_area_obj.get_descendants() |
---|
36 | line_items = line_items.filter(budget_area__in=all_relevant_areas) |
---|
37 | main_lineitem_limits_primary.append(Q(lineitem__budget_area__in=all_relevant_areas)) |
---|
38 | if 'layer' in request.REQUEST and request.REQUEST['layer'] != 'all': |
---|
39 | try: |
---|
40 | layer_id = int(request.REQUEST['layer']) |
---|
41 | layer = finance_core.models.get_layer_by_num(layer_id) |
---|
42 | except KeyError: |
---|
43 | raise Http404("Invalid layer %s request" % request.REQUEST['layer']) |
---|
44 | line_items = line_items.filter(layer=layer_id) |
---|
45 | main_lineitem_limits_primary.append(Q(lineitem__layer=layer_id)) |
---|
46 | else: |
---|
47 | layer = 'all' |
---|
48 | main_lineitem_limit_primary = Q(*main_lineitem_limits_primary) |
---|
49 | |
---|
50 | # Initialize the axis |
---|
51 | # Primary |
---|
52 | if 'primary' in request.REQUEST: |
---|
53 | primary_slug = request.REQUEST['primary'] |
---|
54 | else: |
---|
55 | primary_slug = 'budget-areas' |
---|
56 | try: |
---|
57 | primary_name, primary_axis, primary_axis_objs = finance_core.reporting.get_primary_axis(primary_slug, base_area_obj, term_obj, ) |
---|
58 | except NotImplementedError: |
---|
59 | raise Http404("Primary axis %s is not implemented" % primary_slug) |
---|
60 | |
---|
61 | # Secondary |
---|
62 | if 'secondary' in request.REQUEST: |
---|
63 | secondary_slug = request.REQUEST['secondary'] |
---|
64 | else: |
---|
65 | secondary_slug = 'layers' |
---|
66 | try: |
---|
67 | secondary_name, secondary_axis, secondary_axis_obj = finance_core.reporting.get_secondary_axis(secondary_slug, base_area_obj, term_obj, ) |
---|
68 | except NotImplementedError: |
---|
69 | raise Http404("Secondary axis %s is not implemented" % secondary_slug) |
---|
70 | secondary_axis.append((None, 'Total', Q(), Q())) |
---|
71 | |
---|
72 | primary_labels = [ ] |
---|
73 | for num, (pk, label, qobj, objrel_qobj, ) in enumerate(primary_axis): |
---|
74 | primary_labels.append(label) |
---|
75 | secondary_labels = [ secondary[1] for secondary in secondary_axis ] |
---|
76 | |
---|
77 | # Do the computation |
---|
78 | compute_methods = { |
---|
79 | 'default': finance_core.reporting.build_table, |
---|
80 | 'aggregate': finance_core.reporting.build_table_aggregate, |
---|
81 | 'annotate': finance_core.reporting.build_table_annotate, |
---|
82 | } |
---|
83 | if compute_method in compute_methods: |
---|
84 | build_table = compute_methods[compute_method] |
---|
85 | else: |
---|
86 | raise Http404("Unknown compute_method selected") |
---|
87 | table = build_table(line_items, main_lineitem_limit_primary, primary_axis, primary_axis_objs, secondary_axis, ) |
---|
88 | |
---|
89 | debug = False |
---|
90 | if debug: |
---|
91 | from django.db import connection |
---|
92 | print connection.queries |
---|
93 | print "Number of queries:\t%d" % (len(connection.queries),) |
---|
94 | print "Table size:\t%dx%d" % (len(primary_labels), len(secondary_labels), ) |
---|
95 | |
---|
96 | term_options = finance_core.models.BudgetTerm.objects.all() |
---|
97 | area_options = finance_core.models.BudgetArea.objects.filter(always=True) |
---|
98 | context = { |
---|
99 | 'pagename':'reporting', |
---|
100 | 'term_name': term_name, |
---|
101 | 'term_options': term_options, |
---|
102 | 'area': base_area_obj, |
---|
103 | 'area_options': area_options, |
---|
104 | 'layer': layer, |
---|
105 | 'layer_options': finance_core.models.layers, |
---|
106 | 'axes': finance_core.reporting.axes, |
---|
107 | 'primary_name': primary_name, |
---|
108 | 'secondary_name': secondary_name, |
---|
109 | 'primary_labels': primary_labels, |
---|
110 | 'secondary_labels': secondary_labels, |
---|
111 | 'table': table, |
---|
112 | 'table_with_row_labels': zip(primary_labels, table), |
---|
113 | } |
---|
114 | return render_to_response('finance_core/reporting.html', context, context_instance=RequestContext(request), ) |
---|