[f468e6d] | 1 | from django.http import HttpResponse |
---|
[86817cf] | 2 | import finance_core.models |
---|
[ed4ccc7] | 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 |
---|
[13e7c01] | 7 | from django.db.models import Q |
---|
[ed4ccc7] | 8 | from decimal import Decimal |
---|
[13e7c01] | 9 | import finance_core.reporting |
---|
[f468e6d] | 10 | |
---|
| 11 | def display_tree(request): |
---|
[86817cf] | 12 | root = finance_core.models.BudgetArea.get_by_path(['Accounts']) |
---|
[f468e6d] | 13 | return HttpResponse(root.dump_to_html()) |
---|
[ed4ccc7] | 14 | |
---|
| 15 | def reporting(request): |
---|
| 16 | line_items = finance_core.models.LineItem.objects.all() |
---|
[13e7c01] | 17 | compute_method = 'default' |
---|
| 18 | if 'compute_method' in request.REQUEST: |
---|
| 19 | compute_method = request.REQUEST['compute_method'] |
---|
[c968555] | 20 | |
---|
| 21 | # Main limit to lineitems, relative to primary axis |
---|
| 22 | main_lineitem_limits_primary = [] |
---|
[8a0d18c] | 23 | if 'term' in request.REQUEST and not request.REQUEST['term'] == 'all': |
---|
[ed4ccc7] | 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) |
---|
[c968555] | 27 | main_lineitem_limits_primary.append(Q(lineitem__budget_term=term_obj)) |
---|
[dd2c3d9] | 28 | else: |
---|
| 29 | term_obj = None |
---|
| 30 | term_name = 'All' |
---|
[ed4ccc7] | 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']) |
---|
[c968555] | 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) |
---|
[ed4ccc7] | 49 | |
---|
[13e7c01] | 50 | # Initialize the axis |
---|
[162f527] | 51 | # Primary |
---|
| 52 | if 'primary' in request.REQUEST: |
---|
| 53 | primary_slug = request.REQUEST['primary'] |
---|
| 54 | else: |
---|
| 55 | primary_slug = 'budget-areas' |
---|
| 56 | try: |
---|
[dd2c3d9] | 57 | primary_name, primary_axis, primary_axis_objs = finance_core.reporting.get_primary_axis(primary_slug, base_area_obj, term_obj, ) |
---|
[162f527] | 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: |
---|
[dd2c3d9] | 67 | secondary_name, secondary_axis, secondary_axis_obj = finance_core.reporting.get_secondary_axis(secondary_slug, base_area_obj, term_obj, ) |
---|
[162f527] | 68 | except NotImplementedError: |
---|
| 69 | raise Http404("Secondary axis %s is not implemented" % secondary_slug) |
---|
| 70 | secondary_axis.append((None, 'Total', Q(), Q())) |
---|
[ed4ccc7] | 71 | |
---|
[3f1e074] | 72 | primary_labels = [ ] |
---|
[162f527] | 73 | for num, (pk, label, qobj, objrel_qobj, ) in enumerate(primary_axis): |
---|
[3f1e074] | 74 | primary_labels.append(label) |
---|
[162f527] | 75 | secondary_labels = [ secondary[1] for secondary in secondary_axis ] |
---|
[ed4ccc7] | 76 | |
---|
[13e7c01] | 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, ) |
---|
[ed4ccc7] | 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 | |
---|
[8a0d18c] | 96 | term_options = finance_core.models.BudgetTerm.objects.all() |
---|
[c968555] | 97 | area_options = finance_core.models.BudgetArea.objects.filter(always=True) |
---|
[ed4ccc7] | 98 | context = { |
---|
| 99 | 'pagename':'reporting', |
---|
| 100 | 'term_name': term_name, |
---|
[8a0d18c] | 101 | 'term_options': term_options, |
---|
[ed4ccc7] | 102 | 'area': base_area_obj, |
---|
[c968555] | 103 | 'area_options': area_options, |
---|
| 104 | 'layer': layer, |
---|
| 105 | 'layer_options': finance_core.models.layers, |
---|
[8a0d18c] | 106 | 'axes': finance_core.reporting.axes, |
---|
[ed4ccc7] | 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), ) |
---|