Adding arguments for plugin consructor

Author: magmax

pluginloader/loader.py

@@ -6,15 +6,17 @@ class PluginLoader(object):
     def __init__(self):
         self.plugins = {}
 
-    def load_file(self, filename, onlyif=True):
+    def load_file(self, filename, onlyif=True, args=None, kwargs=None):
+        args = args or []
+        kwargs = kwargs or {}
         context = {}
         with open(filename) as fd:
             exec(fd.read(), context)
 
         for name, clazz in context.iteritems():
             if (isinstance(clazz, type)
                 and self._apply_condition(onlyif, name, clazz)):
-                self.plugins[name] = clazz()
+                self.plugins[name] = clazz(*args, **kwargs)
 
     def _apply_condition(self, condition, *args, **kwargs):
         if callable(condition):

tests/acceptance/test_loader.py

@@ -40,6 +40,32 @@ def test_ignorable_classes(self):
 
         self.assertEquals({}, sut.plugins)
 
+    def test_parameters_for_constructor(self):
+        self.plugin_file.write(
+            'class Foo(object):\n'
+            '  def __init__(self, a):\n'
+            '    self.a = a'
+            )
+        self.plugin_file.flush()
+        sut = PluginLoader()
+
+        sut.load_file(self.plugin_file.name, args=[5])
+
+        self.assertEquals(5, sut.plugins['Foo'].a)
+
+    def test_namedparameters_for_constructor(self):
+        self.plugin_file.write(
+            'class Foo(object):\n'
+            '  def __init__(self, a):\n'
+            '    self.a = a'
+            )
+        self.plugin_file.flush()
+        sut = PluginLoader()
+
+        sut.load_file(self.plugin_file.name, kwargs={'a': 5})
+
+        self.assertEquals(5, sut.plugins['Foo'].a)
+
 
 
 @unittest.skip('Not ready yet')