From 9d7e84eda7fdd5168c10961f0f571c10e9ef92bf Mon Sep 17 00:00:00 2001 From: davebshow <davebshow@gmail.com> Date: Fri, 22 Jul 2016 14:55:09 -0400 Subject: [PATCH] combined two inits into base class --- goblin/element.py | 2 +- goblin/manager.py | 16 ++++++++-------- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/goblin/element.py b/goblin/element.py index 63b3822..71bc5c6 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 3f48104..236448a 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): -- GitLab