From 013f21f9d76701fead641d534c8ba2ca2d48590c Mon Sep 17 00:00:00 2001
From: Stefano Borini <sborini@enthought.com>
Date: Fri, 29 Jun 2018 10:18:30 +0100
Subject: [PATCH] Passing through the information to traits
---
force_bdss/core/base_factory.py | 4 ++--
force_bdss/data_sources/base_data_source_factory.py | 7 +++++--
2 files changed, 7 insertions(+), 4 deletions(-)
diff --git a/force_bdss/core/base_factory.py b/force_bdss/core/base_factory.py
index 0171bdf..e8daa73 100644
--- a/force_bdss/core/base_factory.py
+++ b/force_bdss/core/base_factory.py
@@ -17,8 +17,8 @@ class BaseFactory(HasStrictTraits):
#: in your subclass.
plugin = Instance(Plugin, allow_none=False)
- def __init__(self, plugin):
- super(BaseFactory, self).__init__(plugin=plugin)
+ def __init__(self, plugin, *args, **kwargs):
+ super(BaseFactory, self).__init__(plugin=plugin, *args, **kwargs)
self.name = self.get_name()
identifier = self.get_identifier()
diff --git a/force_bdss/data_sources/base_data_source_factory.py b/force_bdss/data_sources/base_data_source_factory.py
index 206321c..3793fb4 100644
--- a/force_bdss/data_sources/base_data_source_factory.py
+++ b/force_bdss/data_sources/base_data_source_factory.py
@@ -38,8 +38,11 @@ class BaseDataSourceFactory(BaseFactory):
#: Define this to your DataSourceModel
model_class = Type(BaseDataSourceModel, allow_none=False)
- def __init__(self, plugin):
- super(BaseDataSourceFactory, self).__init__(plugin=plugin)
+ def __init__(self, plugin, *args, **kwargs):
+ super(BaseDataSourceFactory, self).__init__(
+ plugin=plugin,
+ *args,
+ **kwargs)
self.data_source_class = self.get_data_source_class()
self.model_class = self.get_model_class()
--
GitLab