Add a form to change an expenditure's split

diff --git a/bluechips/controllers/spend.py b/bluechips/controllers/spend.py
index a3f502d9dbd9325465153b0bdb1932f3ec364cbb..fd884536b1d02811ce0977c40bbbe8c9c9956831 100644 (file)
--- a/bluechips/controllers/spend.py
+++ b/bluechips/controllers/spend.py
@@ -8,6 +8,9 @@ from bluechips.lib.base import *
 from bluechips.widgets import spend
 
 from pylons import request
+from pylons.decorators.rest import dispatch_on
+
+from decimal import Decimal, InvalidOperation
 
 log = logging.getLogger(__name__)
 
@@ -67,3 +70,41 @@ class SpendController(BaseController):
                                                      id=e.id))))
         
         return h.redirect_to('/')
+    
+    @dispatch_on(POST='_post_split',
+                 GET='_get_split')
+    def split(self, id):
+        abort(500)
+    
+    def _get_split(self, id):
+        c.title = 'Change Expenditure Split'
+        
+        c.expenditure = meta.Session.query(model.Expenditure).get(id)
+        c.users = meta.Session.query(model.User)
+        
+        return render('/spend/split.mako')
+    
+    def _post_split(self, id):
+        c.values = request.params
+        c.errors = dict()
+        
+        split_dict = dict()
+        
+        for username, percent in c.values.iteritems():
+            try:
+                user = meta.Session.query(model.User).\
+                    filter(model.User.username==username).one()
+                split_dict[user] = Decimal(percent)
+            except InvalidOperation:
+                c.errors[username] = 'Please enter a number'
+        if c.errors != dict():
+            return self._get_split(id)
+        
+        e = meta.Session.query(model.Expenditure).get(id)
+        e.split(split_dict)
+        
+        meta.Session.commit()
+        
+        h.flash('Expenditure redivided')
+        
+        return h.redirect_to('/')

diff --git a/bluechips/templates/spend/split.mako b/bluechips/templates/spend/split.mako
new file mode 100644 (file)
index 0000000..842466d
--- /dev/null
+++ b/bluechips/templates/spend/split.mako
@@ -0,0 +1,51 @@
+<%inherit file="/base.mako"/>
+
+<%!
+import itertools
+from decimal import Decimal
+%>
+
+<%
+form_cycle = itertools.cycle(['even', 'odd'])
+%>
+
+## ToscaWidgets doesn't support dynamically generating the form fields
+## themselves, so I'm mimicing its style but generating the form by
+## hand
+
+<p>Change how an expenditure is split up. Enter a percentage, or
+something like a percentage, for each user. They don't have to add to
+100.</p>
+
+<p>You're editing an expenditure of ${c.expenditure.amount} by
+${c.expenditure.spender.name} on ${c.expenditure.date}, described as
+"${c.expenditure.description}"</p>
+
+${h.form('', method='post')}
+<form>
+  <ul class="field_list">
+    % for user in c.users:
+    <%
+    name = user.username
+    try:
+        percent = c.values[name]
+    except TypeError:
+        try:
+            share = [s.share for s in c.expenditure.splits if s.user == user][0]
+            percent = (Decimal(100) * Decimal(int(share)) / Decimal(int(c.expenditure.amount))).quantize(Decimal("0.001"))
+        except KeyError:
+            percent = Decimal(0)
+    %>\
+    <li class="${form_cycle.next()}" id="${name}.container">
+      <label id="${name}.label" for="${name}" class="fieldlabel">${user.name}</label>
+      ${h.text(name, value=percent, class_="textfield required", id=name)}
+      % if name in c.errors:
+      <span class="fielderror">${c.errors[name]}</span>
+      % endif
+    </li>
+    % endfor
+    <li class="${form_cycle.next()}" id="submit.container">
+      ${h.submit(None, 'Submit', class_="submitbutton")}
+    </li>
+  </ul>
+</form>