-import os
+from os import path as os_path, listdir as os_listdir
+from traceback import print_exc
+from sys import stdout
-from Tools.Directories import *
-
-def my_import(name):
- mod = __import__(name)
- components = name.split('.')
- for comp in components[1:]:
- mod = getattr(mod, comp)
- return mod
+from Tools.Directories import fileExists
+from Tools.Import import my_import
+from Plugins.Plugin import PluginDescriptor
class PluginComponent:
def __init__(self):
self.plugins = {}
+ self.pluginList = [ ]
self.setPluginPrefix("Plugins.")
-
+ self.resetWarnings()
+
def setPluginPrefix(self, prefix):
self.prefix = prefix
- def readPluginList(self, runAutostartPlugins=False, runAutoendPlugins=False):
- """enumerates plugins"""
+ def addPlugin(self, plugin):
+ self.pluginList.append(plugin)
+ for x in plugin.where:
+ self.plugins.setdefault(x, []).append(plugin)
+ if x == PluginDescriptor.WHERE_AUTOSTART:
+ plugin(reason=0)
+
+ def removePlugin(self, plugin):
+ self.pluginList.remove(plugin)
+ for x in plugin.where:
+ self.plugins[x].remove(plugin)
+ if x == PluginDescriptor.WHERE_AUTOSTART:
+ plugin(reason=1)
- directories = os.listdir(resolveFilename(SCOPE_PLUGINS))
+ def readPluginList(self, directory):
+ """enumerates plugins"""
- for x in directories:
- path = resolveFilename(SCOPE_PLUGINS, x) + "/"
- if os.path.exists(path):
- if fileExists(path + "plugin.py"):
- plugin = my_import('.'.join(("Plugins", x, "plugin")))
-
- if not plugin.__dict__.has_key("Plugins"):
- print "Plugin %s doesn't have 'Plugin'-call." % (x)
- continue
-
- print "plugin", plugin
- plugins = plugin.Plugins()
-
- # allow single entry not to be a list
- if type(plugins) is not list:
- plugins = [ plugins ]
-
- for p in plugins:
- print "imported plugin %s" % (p.name)
-
- for x in p.where:
- self.plugins.setdefault(x, []).append(p)
+ categories = os_listdir(directory)
+
+ new_plugins = [ ]
+
+ for c in categories:
+ directory_category = directory + c
+ if not os_path.isdir(directory_category):
+ continue
+ open(directory_category + "/__init__.py", "a").close()
+ for pluginname in os_listdir(directory_category):
+ path = directory_category + "/" + pluginname
+ if os_path.isdir(path):
+ if fileExists(path + "/plugin.pyc") or fileExists(path + "/plugin.py"):
+ try:
+ plugin = my_import('.'.join(["Plugins", c, pluginname, "plugin"]))
+
+ if not plugin.__dict__.has_key("Plugins"):
+ print "Plugin %s doesn't have 'Plugin'-call." % (pluginname)
+ continue
+
+ plugins = plugin.Plugins(path=path)
+ except Exception, exc:
+ print "Plugin ", c + "/" + pluginname, "failed to load:", exc
+ print_exc(file=stdout)
+ print "skipping plugin."
+ self.warnings.append( (c + "/" + pluginname, str(exc)) )
+ continue
+
+ # allow single entry not to be a list
+ if type(plugins) is not list:
+ plugins = [ plugins ]
+
+ for p in plugins:
+ p.updateIcon(path)
+ new_plugins.append(p)
+
+ # build a diff between the old list of plugins and the new one
+ # internally, the "fnc" argument will be compared with __eq__
+ plugins_added = [p for p in new_plugins if p not in self.pluginList]
+ plugins_removed = [p for p in self.pluginList if p not in new_plugins]
+
+ for p in plugins_removed:
+ self.removePlugin(p)
+
+ for p in plugins_added:
+ self.addPlugin(p)
def getPlugins(self, where):
"""Get list of plugins in a specific category"""
-
+
if type(where) is not list:
where = [ where ]
res = [ ]
res.append(p)
return res
+ def getPluginsForMenu(self, menuid):
+ res = [ ]
+ for p in self.getPlugins(PluginDescriptor.WHERE_SETUP):
+ res += p(menuid)
+ return res
+
+ def clearPluginList(self):
+ self.pluginList = []
+ self.plugins = {}
+
+ def shutdown(self):
+ for p in self.pluginList[:]:
+ self.removePlugin(p)
+
+ def resetWarnings(self):
+ self.warnings = [ ]
+
plugins = PluginComponent()