Add foreign key restraints, expand and fix tests, add deletion and forking.

[misc] / calories.py

diff --git a/calories.py b/calories.py
index 28b1a16e0e99dfea186d6ff91f7dd8c006d1dad0..9951ce5c98896ef91319282143b2925b657ec280 100644 (file)
--- a/calories.py
+++ b/calories.py
@@ -233,7 +233,7 @@ class ConsumptionsHandler(PlomHandler):
         return 'consumptions', default_path 
 
     def do_POST(self):
-        self.write_db()
+        self.try_do(self.write_db)
 
     def write_db(self):
         from uuid import uuid4
@@ -289,15 +289,12 @@ class ConsumptionsHandler(PlomHandler):
                 default_slots -= 1
                 if (default_slots <= 0):
                     break
-        try:
-            db.write()
-            homepage = self.apps['consumptions'] if hasattr(self, 'apps') else self.homepage
-            self.redirect(homepage)
-        except PlomException as e:
-            self.fail_400(e) 
+        db.write()
+        homepage = self.apps['consumptions'] if hasattr(self, 'apps') else self.homepage
+        self.redirect(homepage)
 
     def do_GET(self):
-        self.show_db()
+        self.try_do(self.show_db)
 
     def show_db(self):
         db = CaloriesDB()