from bluechips.lib.base import *
-from pylons import request
+from pylons import request, app_globals as g
from pylons.decorators import validate
+from pylons.decorators.secure import authenticate_form
+from pylons.controllers.util import abort
from formencode import Schema, validators
+from mailer import Message
+
log = logging.getLogger(__name__)
else:
c.title = 'Edit a Transfer'
c.transfer = meta.Session.query(model.Transfer).get(id)
+ if c.transfer is None:
+ abort(404)
return render('/transfer/index.mako')
+ @redirect_on_get('edit')
+ @authenticate_form
@validate(schema=TransferSchema(), form='edit')
def update(self, id=None):
if id is None:
t = model.Transfer()
meta.Session.add(t)
+ op = 'created'
else:
t = meta.Session.query(model.Transfer).get(id)
+ if t is None:
+ abort(404)
+ op = 'updated'
update_sar(t, self.form_result)
meta.Session.commit()
- if id is None:
- h.flash('Transfer created.')
- else:
- h.flash('Transfer updated.')
-
+ show = ('Transfer of %s from %s to %s %s.' %
+ (t.amount, t.debtor, t.creditor, op))
+ h.flash(show)
+
+ # Send email notification to involved users if they have an email set.
+ body = render('/emails/transfer.txt', extra_vars={'transfer': t,
+ 'op': op})
+ g.handle_notification((t.debtor, t.creditor), show, body)
+
+ return h.redirect_to('/')
+
+ def delete(self, id):
+ c.title = 'Delete a Transfer'
+ c.transfer = meta.Session.query(model.Transfer).get(id)
+ if c.transfer is None:
+ abort(404)
+
+ return render('/transfer/delete.mako')
+
+ @redirect_on_get('delete')
+ @authenticate_form
+ def destroy(self, id):
+ t = meta.Session.query(model.Transfer).get(id)
+ if t is None:
+ abort(404)
+
+ if 'delete' in request.params:
+ meta.Session.delete(t)
+
+ meta.Session.commit()
+ show = ("Transfer of %s from %s to %s deleted." %
+ (t.amount, t.debtor, t.creditor))
+ h.flash(show)
+
+ body = render('/emails/transfer.txt',
+ extra_vars={'transfer': t,
+ 'op': 'deleted'})
+ g.handle_notification((t.debtor, t.creditor), show, body)
+
return h.redirect_to('/')