but not edit

1 files changed, 8 insertions, 1 deletions

diff --git a/controller.py b/controller.py
index 4aea9cd..5312b5a 100644
--- a/controller.py
+++ b/controller.py
@@ -68,7 +68,7 @@ class Add:
             e.expense = decimal.Decimal(f.expense.value)
             e.description = f.description.value
 
-            raise web.seeother("/add")
+            self.return_()
         else:
             return self.render(f)
 
@@ -78,6 +78,9 @@ class Add:
     def render(self, f):
         return render(self.template, form = f)
 
+    def return_(self):
+        raise web.seeother("/add")
+
     def form(self):
         return form.Form(
                 form.Textbox(
@@ -118,8 +121,12 @@ class Edit (Add):
     def POST(self, id):
         exp = SingleExpense.get(id)
         self.get_expense = lambda *x: exp
+
         return Add.POST(self)
 
+    def return_(self):
+        raise web.seeother("/")
+
 class Const:
     def GET(self):
         return "Const"