import logging
+from datetime import date
+
from bluechips.lib.base import *
-from bluechips.widgets import transfer
from pylons import request
+from pylons.decorators import validate
+
+from formencode import Schema, validators
log = logging.getLogger(__name__)
+
+class TransferSchema(Schema):
+ "Validate a transfer."
+ allow_extra_fields = False
+ debtor_id = validators.Int(not_empty=True)
+ creditor_id = validators.Int(not_empty=True)
+ amount = model.types.CurrencyValidator(not_empty=True)
+ description = validators.UnicodeString()
+ date = validators.DateConverter()
+
+
class TransferController(BaseController):
def index(self):
- c.title = 'Add a New Transfer'
-
- c.transfer = dict()
- c.transfer['debtor'] = request.environ['user']
-
+ return self.edit()
+
+ def edit(self, id=None):
+ c.users = meta.Session.query(model.User.id, model.User.name)
+ if id is None:
+ c.title = 'Add a New Transfer'
+ c.transfer = model.Transfer()
+ c.transfer.debtor_id = request.environ['user'].id
+ c.transfer.date = date.today()
+ else:
+ c.title = 'Edit a Transfer'
+ c.transfer = meta.Session.query(model.Transfer).get(id)
return render('/transfer/index.mako')
- @validate(form=transfer.new_transfer_form, error_handler='index')
- def new(self):
- t = model.Transfer()
+ @validate(schema=TransferSchema(), form='edit')
+ def update(self, id=None):
+ if id is None:
+ t = model.Transfer()
+ meta.Session.add(t)
+ else:
+ t = meta.Session.query(model.Transfer).get(id)
+
update_sar(t, self.form_result)
- meta.Session.save(t)
meta.Session.commit()
- h.flash('Transfer recorded.')
+ h.flash('Transfer updated.')
return h.redirect_to('/')