Fixed Timestamped hooks to work with new hook api

cooper-software · Oct 8, 2014 · 4c51724 · 4c51724
1 parent c0e86b4
commit 4c51724
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cellardoor/model.py b/cellardoor/model.py
@@ -624,13 +624,13 @@ def setup(self):
         self.hooks.pre('update', self.on_modify)
 
 
-    def on_create(self, fields):
+    def on_create(self, fields, *args, **kwargs):
         now = datetime.utcnow()
         fields['created'] = now
         fields['modified'] = now
 
 
-    def on_modify(self, id, fields, replace=False):
+    def on_modify(self, id, fields, *args, **kwargs):
         fields['modified'] = datetime.utcnow()