pluginloader

1 files changed, 8 insertions, 4 deletions

diff --git a/run_module_examples.py b/run_module_examples.py
index 8a22c10..c6559c8 100644
--- a/run_module_examples.py
+++ b/run_module_examples.py
@@ -1,9 +1,13 @@
 #!/usr/bin/python2.7
 """calls the example_run method on all modules"""
-from photoblaster.modules import Pb
-for cls in Pb.__subclasses__():
-    print cls.__name__
-    if cls.__name__ == "PbGradient":
+import pluginbase
+from photoblaster.modules import Modules
+
+
+modules = Modules()
+for module_name in modules.list_modules():
+    if module_name == 'pbgradient':
+        cls = modules.get_module(module_name)
         instance = cls.example_run()
         instance.get_output_file().s3move()
         print instance.get_output_file().as_dict()