itdominator
/
terminator
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

handle import failures and add some very borked doctesting

This commit is contained in:
Chris Jones 2009-12-17 12:54:42 +00:00
parent 209c37aa34
commit 6088084fdf
1 changed files with 24 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
terminatorlib/plugin.py
View File

@ -1,12 +1,21 @@
#!/usr/bin/python #!/usr/bin/python
# Terminator by Chris Jones <cmsj@tenshu.net> # Terminator by Chris Jones <cmsj@tenshu.net>
# GPL v2 only # GPL v2 only
"""plugin.py - Base plugin system""" """plugin.py - Base plugin system
>>> registry = PluginRegistry()
>>> registry.load_plugins()
>>> registry.get_plugins_by_capability('test')
[<testplugin.TestPlugin object at ...>]
>>> registry.get_plugins_by_capability('this_should_not_ever_exist')
[]
"""
import sys import sys
import os import os
import borg import borg
from util import dbg from util import dbg, err
class Plugin(object): class Plugin(object):
"""Definition of our base plugin class""" """Definition of our base plugin class"""
@ -45,11 +54,17 @@ class PluginRegistry(borg.Borg):
plugin[-3:] == '.py': plugin[-3:] == '.py':
dbg('PluginRegistry::load_plugins: Importing plugin %s' % dbg('PluginRegistry::load_plugins: Importing plugin %s' %
plugin) plugin)
try:
__import__(plugin[:-3], None, None, ['']) __import__(plugin[:-3], None, None, [''])
except Exception as e:
err('PluginRegistry::load_plugins: Importing plugin %s \
failed: %s' % (plugin, e))
def get_plugins_by_capability(self, capability): def get_plugins_by_capability(self, capability):
"""Return a list of plugins with a particular capability""" """Return a list of plugins with a particular capability"""
result = [] result = []
dbg('PluginRegistry::get_plugins_by_capability: searching %d plugins \
for %s' % (len(Plugin.__subclasses__()), capability))
for plugin in Plugin.__subclasses__(): for plugin in Plugin.__subclasses__():
if capability in plugin.capabilities: if capability in plugin.capabilities:
if not plugin in self.instances: if not plugin in self.instances:
@ -57,3 +72,9 @@ class PluginRegistry(borg.Borg):
result.append(self.instances[plugin]) result.append(self.instances[plugin])
return result return result
if __name__ == '__main__':
import doctest
sys.path.insert(0, 'plugins')
import testplugin
(failed, attempted) = doctest.testmod()
print "%d/%d tests failed" % (failed, attempted)