Changeset 338

Timestamp:

06/13/08 18:35:55 (5 years ago)

Author:

ged

Message:

* Added setattr method on Metaclass so that you can add properties

slightly more easily after class definition (but before setup_all):

class A(Entity):

pass

A.name = Field(String(32))

Location:

elixir/trunk

Files:

• elixir/entity.py (modified) (1 diff)
• tests/test_properties.py (modified) (1 diff)

elixir/trunk/elixir/entity.py

@@ -726 +726 @@
             elixir.setup_all()
         return type.__call__(cls, *args, **kwargs)
+
+    def __setattr__(cls, key, value):
+        if isinstance(value, Property):
+            if hasattr(cls, '_setup_done'):
+                raise Exception('Cannot set attribute on a class after '
+                                'setup_all')
+            else:
+                value.attach(cls, key)
+        else:
+            type.__setattr__(cls, key, value)
 
 

elixir/trunk/tests/test_properties.py

@@ -198 +198 @@
         assert p.primary_email == 'new@z.com'
 
+    def test_setattr(self):
+        class A(Entity):
+            pass
+
+        A.name = Field(String(30))
+
+        setup_all(True)
+
+        a1 = A(name='a1')
+
+        session.flush(); session.clear()
+
+        a = A.query.one()
+
+        assert a.name == 'a1'
+