Passing through the information to traits

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()

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()