bluechips/controllers/spend.py

   1 """
   2 Handle expenditures
   3 """
   4
   5 import logging
   6
   7 from decimal import Decimal, InvalidOperation
   8
   9 from bluechips.lib.base import *
  10
  11 from pylons import request, app_globals as g
  12 from pylons.decorators.rest import dispatch_on
  13 from pylons.decorators import validate
  14 from pylons.controllers.util import abort
  15
  16 from formencode import validators, Schema
  17 from formencode.foreach import ForEach
  18 from formencode.variabledecode import NestedVariables
  19
  20 from mailer import Message
  21
  22 log = logging.getLogger(__name__)
  23
  24
  25 class ShareSchema(Schema):
  26     "Validate individual user shares."
  27     allow_extra_fields = False
  28     user_id = validators.Int(not_empty=True)
  29     amount = validators.Number(not_empty=True)
  30
  31
  32 class ExpenditureSchema(Schema):
  33     "Validate an expenditure."
  34     allow_extra_fields = False
  35     pre_validators = [NestedVariables()]
  36     spender_id = validators.Int(not_empty=True)
  37     amount = model.types.CurrencyValidator(not_empty=True)
  38     description = validators.UnicodeString()
  39     date = validators.DateConverter()
  40     shares = ForEach(ShareSchema)
  41
  42
  43 class SpendController(BaseController):
  44     def index(self):
  45         return self.edit()
  46
  47     def edit(self, id=None):
  48         c.users = meta.Session.query(model.User.id, model.User)
  49         if id is None:
  50             c.title = 'Add a New Expenditure'
  51             c.expenditure = model.Expenditure()
  52             c.expenditure.spender_id = request.environ['user'].id
  53
  54             num_residents = meta.Session.query(model.User).\
  55                     filter_by(resident=True).count()
  56             # Pre-populate split percentages for an even split.
  57             c.values = {}
  58             for ii, user_row in enumerate(c.users):
  59                 user_id, user = user_row
  60                 if user.resident:
  61                     val = Decimal(100) / Decimal(num_residents)
  62                 else:
  63                     val = 0
  64                 c.values['shares-%d.amount' % ii] = val
  65         else:
  66             c.title = 'Edit an Expenditure'
  67             c.expenditure = meta.Session.query(model.Expenditure).get(id)
  68             if c.expenditure is None:
  69                 abort(404)
  70             c.values = {}
  71             for ii, user_row in enumerate(c.users):
  72                 user_id, user = user_row
  73                 try:
  74                     share = [s.share for s in c.expenditure.splits
  75                              if s.user == user][0]
  76                     percent = (Decimal(100) * Decimal(int(share)) /
  77                                Decimal(int(c.expenditure.amount))).\
  78                             quantize(Decimal("0.001"))
  79                 except IndexError:
  80                     percent = 0
  81                 c.values['shares-%d.amount' % ii] = percent
  82
  83         return render('/spend/index.mako')
  84
  85     @validate(schema=ExpenditureSchema(), form='edit', variable_decode=True)
  86     def update(self, id=None):
  87         # Either create a new object, or, if we're editing, get the
  88         # old one
  89         if id is None:
  90             e = model.Expenditure()
  91             meta.Session.add(e)
  92             op = 'created'
  93         else:
  94             e = meta.Session.query(model.Expenditure).get(id)
  95             if e is None:
  96                 abort(404)
  97             op = 'updated'
  98
  99         # Set the fields that were submitted
 100         shares = self.form_result.pop('shares')
 101         update_sar(e, self.form_result)
 102         if e.id is not None:
 103             e.update_split()
 104
 105         users = dict(meta.Session.query(model.User.id, model.User).all())
 106         split_dict = {}
 107         for share_params in shares:
 108             user = users[share_params['user_id']]
 109             split_dict[user] = Decimal(str(share_params['amount']))
 110         e.split(split_dict)
 111
 112         meta.Session.commit()
 113
 114         show = ("Expenditure of %s paid for by %s %s." %
 115                 (e.amount, e.spender, op))
 116         h.flash(show)
 117
 118         # Send email notification to involved users if they have an email set.
 119         involved_users = set(sp.user for sp in e.splits if sp.share != 0)
 120         involved_users.add(e.spender)
 121         body = render('/emails/expenditure.txt',
 122                       extra_vars={'expenditure': e,
 123                                   'op': op})
 124         g.handle_notification(involved_users, show, body)
 125
 126         return h.redirect_to('/')