Adding class validator

magmax · magmax · commit 583714b31c58 · 2014-05-21T20:00:10.000+02:00
diff --git a/pluginloader/loader.py b/pluginloader/loader.py
@@ -6,11 +6,17 @@ class PluginLoader(object):
     def __init__(self):
         self.plugins = {}
 
-    def load_file(self, filename):
+    def load_file(self, filename, onlyif=True):
         context = {}
         with open(filename) as fd:
             exec(fd.read(), context)
 
         for name, clazz in context.iteritems():
-            if isinstance(clazz, type):
+            if (isinstance(clazz, type)
+                and self._apply_condition(onlyif, name, clazz)):
                 self.plugins[name] = clazz()
+
+    def _apply_condition(self, condition, *args, **kwargs):
+        if callable(condition):
+            return condition(*args, **kwargs)
+        return condition
diff --git a/tests/acceptance/test_loader.py b/tests/acceptance/test_loader.py
@@ -31,6 +31,16 @@ def test_base_case(self):
         self.assertIsInstance(sut.plugins['Foo'], object)
         self.assertEquals('Foo', sut.plugins['Foo'].__class__.__name__)
 
+    def test_ignorable_classes(self):
+        self.plugin_file.write('class Foo(object): pass')
+        self.plugin_file.flush()
+        sut = PluginLoader()
+
+        sut.load_file(self.plugin_file.name, onlyif=lambda x, y: False)
+
+        self.assertEquals({}, sut.plugins)
+
+
 
 @unittest.skip('Not ready yet')
 class plugins_in_directory(unittest.TestCase):
