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                     if c.expenditure.amount == 0:
  77                         percent = 0
  78                     else:
  79                         percent = (Decimal(100) * Decimal(int(share)) /
  80                                    Decimal(int(c.expenditure.amount))).\
  81                                 quantize(Decimal("0.001"))
  82                 except IndexError:
  83                     percent = 0
  84                 c.values['shares-%d.amount' % ii] = percent
  85
  86         return render('/spend/index.mako')
  87
  88     @validate(schema=ExpenditureSchema(), form='edit', variable_decode=True)
  89     def update(self, id=None):
  90         # Either create a new object, or, if we're editing, get the
  91         # old one
  92         if id is None:
  93             e = model.Expenditure()
  94             meta.Session.add(e)
  95             op = 'created'
  96         else:
  97             e = meta.Session.query(model.Expenditure).get(id)
  98             if e is None:
  99                 abort(404)
 100             op = 'updated'
 101
 102         # Set the fields that were submitted
 103         shares = self.form_result.pop('shares')
 104         update_sar(e, self.form_result)
 105
 106         users = dict(meta.Session.query(model.User.id, model.User).all())
 107         split_dict = {}
 108         for share_params in shares:
 109             user = users[share_params['user_id']]
 110             split_dict[user] = Decimal(str(share_params['amount']))
 111         e.split(split_dict)
 112
 113         meta.Session.commit()
 114
 115         show = ("Expenditure of %s paid for by %s %s." %
 116                 (e.amount, e.spender, op))
 117         h.flash(show)
 118
 119         # Send email notification to involved users if they have an email set.
 120         involved_users = set(sp.user for sp in e.splits if sp.share != 0)
 121         involved_users.add(e.spender)
 122         body = render('/emails/expenditure.txt',
 123                       extra_vars={'expenditure': e,
 124                                   'op': op})
 125         g.handle_notification(involved_users, show, body)
 126
 127         return h.redirect_to('/')