X-Git-Url: https://asedeno.scripts.mit.edu/gitweb/?a=blobdiff_plain;f=bluechips%2Fcontrollers%2Fstatus.py;h=faadfc5085b318fbaf2ed65dc74563eff14d9888;hb=512db6643a148932d6f734972f23ddf55a2e91ac;hp=491493e1ec48cda16d8434bb28181411af4d22ed;hpb=0ac94d84ed130a99acb3007faae3dff747f34b86;p=bluechips.git

diff --git a/bluechips/controllers/status.py b/bluechips/controllers/status.py
index 491493e..faadfc5 100644
--- a/bluechips/controllers/status.py
+++ b/bluechips/controllers/status.py
@@ -5,9 +5,51 @@ Calculate the current state of the books
 import logging
 
 from bluechips.lib.base import *
+from bluechips.lib.totals import *
+
+import sqlalchemy
+
+from datetime import date, timedelta
+from decimal import Decimal
+
+from pylons import request
 
 log = logging.getLogger(__name__)
 
 class StatusController(BaseController):
     def index(self):
-        return 'Hello World'
+        c.debts = debts()
+        c.settle = settle(c.debts)
+        
+        c.total = self._total(True)
+        
+        year = date.today() - timedelta(days=365)
+        this_year = date.today().replace(month=1, day=1)
+        this_month = date.today().replace(day=1)
+        last_month = (date.today() - timedelta(days=30)).replace(day=1)
+        
+        c.year_total, c.this_year_total, c.this_month_total =\
+            [self._total(model.expenditures.c.date >= i)
+             for i in [year, this_year, this_month]]
+        
+
+        c.last_month_total = self._total(sqlalchemy.and_(
+                    model.expenditures.c.date >= last_month,
+                    model.expenditures.c.date < this_month))
+        
+        c.expenditures = meta.Session.query(model.Expenditure).\
+            filter(model.Expenditure.spender==request.environ['user']).\
+            limit(10).all()
+        c.transfers = meta.Session.query(model.Transfer).\
+            filter(sqlalchemy.or_(
+                model.Transfer.debtor==request.environ['user'],
+                model.Transfer.creditor==request.environ['user'])).\
+                limit(10).all()
+        
+        return render('/status/index.mako')
+    
+    def _total(self, where):
+        return (meta.Session.execute(sqlalchemy.sql.select([
+                sqlalchemy.func.sum(model.expenditures.c.amount).\
+                    label('total')]).\
+                    where(where)).scalar() or Decimal("0.00")) / 100