diff --git a/goblin/element.py b/goblin/element.py index 63b382272d84c3503a14d41bdbd151cd24b9e4ef..71bc5c6631ff6551455ff8946481b3db65facab5 100644 --- a/goblin/element.py +++ b/goblin/element.py @@ -150,7 +150,7 @@ class VertexPropertyDescriptor: class VertexProperty(Vertex, abc.BaseProperty): - """Base class for user defined vertex properties. Not yet supported.""" + """Base class for user defined vertex properties.""" __descriptor__ = VertexPropertyDescriptor diff --git a/goblin/manager.py b/goblin/manager.py index 3f4810449853d5277dd1c0ca19666b644a0f731a..236448ae67c29b20731503021110dc52424e6efd 100644 --- a/goblin/manager.py +++ b/goblin/manager.py @@ -3,6 +3,12 @@ class VertexPropertyManager: + def __init__(self, data_type, vertex_prop, card): + self._data_type = data_type + self._vertex_prop = vertex_prop + self._card = card + self._mapper_func = vertex_prop.__mapping__.mapper_func + @property def mapper_func(self): return self._mapper_func @@ -22,10 +28,7 @@ class VertexPropertyManager: class ListVertexPropertyManager(list, VertexPropertyManager): def __init__(self, data_type, vertex_prop, card, obj): - self._data_type = data_type - self._vertex_prop = vertex_prop - self._card = card - self._mapper_func = vertex_prop.__mapping__.mapper_func + VertexPropertyManager.__init__(self, data_type, vertex_prop, card) list.__init__(self, obj) def append(self, val): @@ -37,10 +40,7 @@ class ListVertexPropertyManager(list, VertexPropertyManager): class SetVertexPropertyManager(set, VertexPropertyManager): def __init__(self, data_type, vertex_prop, card, obj): - self._data_type = data_type - self._vertex_prop = vertex_prop - self._card = card - self._mapper_func = vertex_prop.__mapping__.mapper_func + VertexPropertyManager.__init__(self, data_type, vertex_prop, card) set.__init__(self, obj) def add(self, val):