60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
|
#!/usr/bin/python
|
||
|
|
# Terminator by Chris Jones <cmsj@tenshu.net>
|
||
|
|
# GPL v2 only
|
||
|
|
"""plugin.py - Base plugin system"""
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
import borg
|
||
|
|
from util import dbg
|
||
|
|
|
||
|
|
class Plugin(object):
|
||
|
|
"""Definition of our base plugin class"""
|
||
|
|
capabilities = None
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
"""Class initialiser."""
|
||
|
|
pass
|
||
|
|
|
||
|
|
class PluginRegistry(borg.Borg):
|
||
|
|
"""Definition of a class to store plugin instances"""
|
||
|
|
instances = None
|
||
|
|
path = None
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
"""Class initialiser"""
|
||
|
|
borg.Borg.__init__(self)
|
||
|
|
self.prepare_attributes()
|
||
|
|
|
||
|
|
def prepare_attributes(self):
|
||
|
|
"""Prepare our attributes"""
|
||
|
|
if not self.instances:
|
||
|
|
self.instances = {}
|
||
|
|
if not self.path:
|
||
|
|
(head, tail) = os.path.split(borg.__file__)
|
||
|
|
self.path = os.path.join(head, 'plugins')
|
||
|
|
dbg('PluginRegistry::prepare_attributes: Plugin path: %s' %
|
||
|
|
self.path)
|
||
|
|
|
||
|
|
def load_plugins(self):
|
||
|
|
"""Load all plugins present in the plugins/ directory in our module"""
|
||
|
|
sys.path.insert(0, self.path)
|
||
|
|
files = os.listdir(self.path)
|
||
|
|
for plugin in files:
|
||
|
|
if os.path.isfile(os.path.join(self.path, plugin)) and \
|
||
|
|
plugin[-3:] == '.py':
|
||
|
|
dbg('PluginRegistry::load_plugins: Importing plugin %s' %
|
||
|
|
plugin)
|
||
|
|
__import__(plugin[:-3], None, None, [''])
|
||
|
|
|
||
|
|
def get_plugins_by_capability(self, capability):
|
||
|
|
"""Return a list of plugins with a particular capability"""
|
||
|
|
result = []
|
||
|
|
for plugin in Plugin.__subclasses__():
|
||
|
|
if capability in plugin.capabilities:
|
||
|
|
if not plugin in self.instances:
|
||
|
|
self.instances[plugin] = plugin()
|
||
|
|
result.append(self.instances[plugin])
|
||
|
|
return result
|
||
|
|
|