graveldb: assign default per-property

webgravel · Jul 15, 2013 · 0b1510f · 0b1510f
1 parent 3269ae8
commit 0b1510f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/dbtest.py b/dbtest.py
@@ -1,7 +1,7 @@
 import graveldb
 
 class Screw(graveldb.Table('Screw', '/tmp')):
-    default = lambda self: graveldb.Object(size=1)
+    default = dict(size=1)
 
 s = Screw('foo')
 with s:

diff --git a/graveldb.py b/graveldb.py
@@ -19,7 +19,13 @@ def Table(name, path):
 class _Table(object):
     def __init__(self, name):
         self.name = name
-        self.data = self.table.get(name, self.default())
+        self.data = self.table.get(name, Object())
+        self.setup()
+
+    def setup(self):
+        for k, v in self.default.items():
+            if not hasattr(self.data, k):
+                setattr(self.data, k, v)
 
     def save(self):
         self.table[self.name] = self.data
@@ -30,7 +36,7 @@ def __enter__(self):
     def __exit__(self, *args):
         self.table.unlock_all()
 
-    default = lambda self: Object()
+    default = {}
 
 class Object(object):
     def __init__(self, **kwargs):