Changeset 332 for elixir/trunk/elixir/__init__.py

Timestamp:

05/14/08 18:55:49 (5 years ago)

Author:

cleverdevil

Message:

Made the elixir.entities list into a useful way to lookup entities by name.
You can do "from elixir import entities" and then get any registered entity
through "entities.MyEntity".

Files:

• elixir/trunk/elixir/__init__.py (modified) (2 diffs)

elixir/trunk/elixir/__init__.py

@@ -39 +39 @@
 __version__ = '0.6.0'
 
-__all__ = ['Entity', 'EntityMeta',
+__all__ = ['Entity', 'EntityMeta', 'entities',
            'Field', 'has_field', 'with_fields', 
            'has_property', 'GenericProperty', 'ColumnProperty', 'Synonym',
@@ -91 +91 @@
 
 # default entity collection
-entities = list()
+class AttributeEntityList(list):
+    
+    def __init__(self):
+        self._entity_map = None
+        list.__init__(self)
+    
+    def __getattr__(self, key):
+        if self._entity_map is None:
+            self._entity_map = {}
+            for entity in self:
+                self._entity_map[entity.__name__] = entity
+        
+        return self._entity_map.get(key)
+            
+entities = AttributeEntityList()