Merge pull request mogui#187 from kensho/add_abstract_field

Add abstract field to class objects

pyorient/ogm/graph.py

@@ -183,6 +183,7 @@ def extract_properties(property_schema, is_edge):
             props.update(extract_properties(class_def['properties'], is_edge))
 
             props['class_fields'] = class_def.get('customFields', None) or {}
+            props['abstract'] = class_def.get('abstract', False)
 
             if bases:
                 # Create class for the graph type

tests/test_ogm.py

@@ -789,3 +789,23 @@ def testCustomFields(self):
         self.assertEquals(
             {'test_field_1': 'test string two'},
             g.registry['classfieldedge'].class_fields)
+
+
+class OGMTestAbstractField(unittest.TestCase):
+    def __init__(self, *args, **kwargs):
+        super(OGMTestAbstractField, self).__init__(*args, **kwargs)
+        self.g = None
+
+    def setUp(self):
+        g = self.g = Graph(Config.from_url('abstract_classes', 'root', 'root'
+                                           , initial_drop=True))
+        g.client.command('CREATE CLASS AbstractClass EXTENDS V ABSTRACT')
+        g.client.command('CREATE CLASS ConcreteClass EXTENDS V')
+
+    def testAbstractFlag(self):
+        g = self.g
+
+        database_registry = g.build_mapping(
+            declarative_node(), declarative_relationship(), auto_plural=True)
+        self.assertTrue(database_registry['AbstractClass'].abstract)
+        self.assertFalse(database_registry['ConcreteClass'].abstract)