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