diff donations/management/commands/donations_report.py @ 621:08d83015e15d

For issue #21, add a management command to display donation stats.
author Brian Neal <bgneal@gmail.com>
date Tue, 30 Oct 2012 18:55:16 -0500
parents
children 161b56849114
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/donations/management/commands/donations_report.py	Tue Oct 30 18:55:16 2012 -0500
@@ -0,0 +1,63 @@
+"""donations_report.py
+
+A management command to display donation statistics for a given year.
+
+"""
+import datetime
+import decimal
+
+from django.core.management.base import LabelCommand, CommandError
+from django.db.models import Sum
+
+from donations.models import Donation
+
+ROW_FMT = '%10s %7s %7s'
+
+
+class Command(LabelCommand):
+    help = "Display donation statistics for a given year"
+    args = "<year>"
+
+    def handle_label(self, year, **kwargs):
+        """Display donation statistics for a given year"""
+
+        try:
+            year = int(year)
+        except ValueError:
+            raise CommandError("invalid year")
+
+        today = datetime.date.today()
+        if year < 2011 or year > today.year:
+            raise CommandError("invalid year")
+
+        max_month = 12 if year != today.year else today.month
+
+        print ROW_FMT % ('Month', 'Gross', 'Net')
+        sep = ROW_FMT % ('-' * 10, '-' * 7, '-' * 7)
+        print sep
+
+        total_gross = decimal.Decimal()
+        total_net = decimal.Decimal()
+
+        for month in range(1, max_month + 1):
+            r = Donation.objects.filter(
+                    payment_date__year=year,
+                    payment_date__month=month).aggregate(
+                            Sum('mc_gross'), Sum('mc_fee'))
+
+            gross, fee = r['mc_gross__sum'], r['mc_fee__sum']
+
+            gross = gross if gross else decimal.Decimal()
+            fee = fee if fee else decimal.Decimal()
+
+            net = gross - fee
+
+            d = datetime.date(year=year, month=month, day=1)
+
+            print ROW_FMT % (d.strftime('%b %Y'), gross, net)
+
+            total_gross += gross
+            total_net += net
+
+        print sep
+        print ROW_FMT % ('Total:', total_gross, total_net)