X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/a63fe4fc45fc957dce2db4752839cb0191646e26..d0a84d65c3a38c3e85576159c39ac88a46bb7b4a:/lib/python/Components/PluginComponent.py diff --git a/lib/python/Components/PluginComponent.py b/lib/python/Components/PluginComponent.py index adfc98a0..44fe896a 100644 --- a/lib/python/Components/PluginComponent.py +++ b/lib/python/Components/PluginComponent.py @@ -1,76 +1,119 @@ -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 * -from Screens.Menu import menuupdater +from Tools.Directories import fileExists +from Tools.Import import my_import +from Plugins.Plugin import PluginDescriptor +import keymapparser class PluginComponent: def __init__(self): - self.plugins = [] + self.plugins = {} + self.pluginList = [ ] self.setPluginPrefix("Plugins.") - self.menuEntries = [] - + self.resetWarnings() + def setPluginPrefix(self, prefix): self.prefix = prefix - def getPluginList(self, runAutostartPlugins=False, runAutoendPlugins=False): - list = [] - dir = os.listdir(resolveFilename(SCOPE_PLUGINS)) - self.menuDelete() - self.menuEntries = [] - - for x in dir: - path = resolveFilename(SCOPE_PLUGINS, x) + "/" - try: - if os.path.exists(path): - if fileExists(path + "plugin.py"): - pluginmodule = self.prefix + x + ".plugin" - print "trying to import " + pluginmodule - exec "import " + pluginmodule - plugin = eval(pluginmodule) - plugins = plugin.getPlugins() - try: picturepaths = plugin.getPicturePaths() - except: - picturepaths = [] - for p in plugins: - picturepaths.append("") + 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) + + def readPluginList(self, directory): + """enumerates plugins""" + + 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: - for menuEntry in plugin.getMenuRegistrationList(): - self.menuEntries.append([menuEntry, pluginmodule]) - except: - pass - - for y in range(len(plugins)): - if len(plugins[y]) < 5: - list.append((path + picturepaths[y], plugins[y][0] , x, plugins[y][2], plugins[y][3], None, plugins[y][1])) - else: - list.append((path + picturepaths[y], plugins[y][0] , x, plugins[y][2], plugins[y][3], plugins[y][4], plugins[y][1])) - if runAutostartPlugins: - try: plugin.autostart() - except: pass - if runAutoendPlugins: - try: plugin.autoend() - except: pass - except: - print "Directory", path, "contains a faulty plugin" - self.menuUpdate() - return list - - def menuDelete(self): - for menuEntry in self.menuEntries: - menuupdater.delMenuItem(menuEntry[0][0], menuEntry[0][1], menuEntry[0][2], menuEntry[1], menuEntry[0][3]) - - def menuUpdate(self): - for menuEntry in self.menuEntries: - menuupdater.addMenuItem(menuEntry[0][0], menuEntry[0][1], menuEntry[0][2], menuEntry[1], menuEntry[0][3]) - - def runPlugin(self, plugin, session): - try: - exec("import " + self.prefix + plugin[2] + ".plugin") - if plugin[3] == "screen": - session.open(eval(self.prefix + plugin[2] + ".plugin." + plugin[4]), plugin[5]) - elif plugin[3] == "function": - eval(self.prefix + plugin[2] + ".plugin." + plugin[4])(session, plugin[5]) - except: - print "exec of plugin failed!" + 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) + + if fileExists(path + "/keymap.xml"): + try: + keymapparser.readKeymap(path + "/keymap.xml") + except Exception, exc: + print "keymap for plugin %s/%s failed to load: " % (c, pluginname), exc + self.warnings.append( (c + "/" + pluginname, str(exc)) ) + + # 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 = [ ] + for x in where: + for p in self.plugins.get(x, [ ]): + 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()