from bluechips.lib.base import *
-from pylons import request
+from pylons import request, app_globals as g
from pylons.decorators.rest import dispatch_on
from pylons.decorators import validate
from formencode.foreach import ForEach
from formencode.variabledecode import NestedVariables
+from mailer import Message
+
log = logging.getLogger(__name__)
allow_extra_fields = False
pre_validators = [NestedVariables()]
spender_id = validators.Int(not_empty=True)
- amount = validators.Number(not_empty=True)
+ amount = model.types.CurrencyValidator(not_empty=True)
description = validators.UnicodeString()
date = validators.DateConverter()
shares = ForEach(ShareSchema)
c.title = 'Add a New Expenditure'
c.expenditure = model.Expenditure()
c.expenditure.spender_id = request.environ['user'].id
+
+ num_residents = meta.Session.query(model.User).\
+ filter_by(resident=True).count()
+ # Pre-populate split percentages for an even split.
+ c.values = {}
+ for ii, user_row in enumerate(c.users):
+ user_id, user = user_row
+ if user.resident:
+ val = Decimal(100) / Decimal(num_residents)
+ else:
+ val = 0
+ c.values['shares-%d.amount' % ii] = val
else:
c.title = 'Edit an Expenditure'
c.expenditure = meta.Session.query(model.Expenditure).get(id)
if id is None:
e = model.Expenditure()
meta.Session.add(e)
+ op = 'created'
else:
e = meta.Session.query(model.Expenditure).get(id)
+ op = 'updated'
# Set the fields that were submitted
shares = self.form_result.pop('shares')
- e.amount = Decimal(self.form_result.pop('amount') * 100)
update_sar(e, self.form_result)
if e.id is not None:
e.update_split()
e.split(split_dict)
meta.Session.commit()
-
- h.flash('Expenditure updated.')
+ show = ("Expenditure of %s paid for by %s %s." %
+ (e.amount, e.spender, op))
+ h.flash(show)
+
+ # Send email notification to involved users if they have an email set.
+ involved_users = set(sp.user for sp in e.splits if sp.share != 0)
+ involved_users.add(e.spender)
+ body = render('/emails/expenditure.txt',
+ extra_vars={'expenditure': e,
+ 'op': op})
+ g.handle_notification(involved_users, show, body)
+
return h.redirect_to('/')