X-Git-Url: https://asedeno.scripts.mit.edu/gitweb/?a=blobdiff_plain;f=bluechips%2Fcontrollers%2Fspend.py;h=1a1c34612091b888df800ebeacfba199fc631da3;hb=9458527642fab328a945d06daed04404c1ae2091;hp=d39f78e900f6e388c77c31643f5087b9a1b139c7;hpb=2976c86719afddb7000dbb938f92cbc3be4c7ac3;p=bluechips.git diff --git a/bluechips/controllers/spend.py b/bluechips/controllers/spend.py index d39f78e..1a1c346 100644 --- a/bluechips/controllers/spend.py +++ b/bluechips/controllers/spend.py @@ -6,23 +6,105 @@ import logging from bluechips.lib.base import * from bluechips.widgets import spend -from bluechips.lib.split import * + +from pylons import request +from pylons.decorators.rest import dispatch_on + +from decimal import Decimal, InvalidOperation log = logging.getLogger(__name__) class SpendController(BaseController): def index(self): + c.title = 'Add a New Expenditure' + + c.expenditure = dict() + c.expenditure['spender'] = request.environ['user'] + + return render('/spend/index.mako') + + def edit(self, id): + c.title = 'Edit an Expenditure' + + c.expenditure = meta.Session.query(model.Expenditure).get(id) + return render('/spend/index.mako') - @validate(form=spend.new_spend_form, error_handler='index') - def new(self): - e = model.Expenditure() + def update(self, id=None): + # Validate the submission + if not valid(self, spend.new_spend_form): + if id is None: + return self.index() + else: + return self.edit(id) + + # Either create a new object, or, if we're editing, get the + # old one + if id is None: + e = model.Expenditure() + meta.Session.add(e) + else: + e = meta.Session.query(model.Expenditure).get(id) + + # Set the fields that were submitted update_sar(e, self.form_result) - meta.Session.save(e) - even_split(e) + if id is None: + e.even_split() + else: + e.update_split() + + meta.Session.commit() + + h.flash('Expenditure recorded.') + h.flash("""Want to do something unusual? + +