allow binary only python plugins

[enigma2.git] / lib / python / Components / PluginComponent.py

diff --git a/lib/python/Components/PluginComponent.py b/lib/python/Components/PluginComponent.py
index 5315d3d94c68c0dd630951eed9406092bca278c6..145f3d90a7b3a8ee24eeba097be50669894cbf25 100644 (file)
--- a/lib/python/Components/PluginComponent.py
+++ b/lib/python/Components/PluginComponent.py
@@ -1,52 +1,99 @@
 import os
 
 from Tools.Directories import *
-from Screens.Menu import menuupdater
-#import Plugins
+from Plugins.Plugin import PluginDescriptor
+
+def my_import(name):
+       print name
+       mod = __import__(name)
+       components = name.split('.')
+       for comp in components[1:]:
+               mod = getattr(mod, comp)
+       return mod
 
 class PluginComponent:
        def __init__(self):
-               self.plugins = []
+               self.plugins = {}
+               self.pluginList = [ ]
                self.setPluginPrefix("Plugins.")
-               self.menuEntries = []
                
        def setPluginPrefix(self, prefix):
                self.prefix = prefix
-
-       def getPluginList(self):
-               list = []
-               dir = os.listdir("/usr/lib/enigma2/python/Plugins/")
-               self.menuDelete()
-               self.menuEntries = []
-               for x in dir:
-                       if x[-3:] == ".py" and x[:-3] != "__init__":
-                               print "trying to import " + self.prefix + x[:-3]
-                               exec "import " + self.prefix + x[:-3]
-                               picturepath = eval(self.prefix + x[:-3]).getPicturePath()
-                               pluginname = eval(self.prefix + x[:-3]).getPluginName()
-                               try:
-                                       for menuEntry in eval(self.prefix + x[:-3]).getMenuRegistrationList():
-                                               self.menuEntries.append([menuEntry, self.prefix + x[:-3]])
-                               except:
-                                       pass
-
-                               list.append((picturepath, pluginname , x[:-3]))
-               self.menuUpdate()
-               return list
        
-       def menuDelete(self):
-               for menuEntry in self.menuEntries:
-                       menuupdater.delMenuItem(menuEntry[0][0], menuEntry[0][2], menuEntry[1], menuEntry[0][3])
+       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 x in os.listdir(directory_category):
+                               path = directory_category + "/" + x + "/"
+                               if os.path.isdir(path):
+                                       if fileExists(path + "plugin.pyc") or fileExists(path + "plugin.py"):
+                                               plugin = my_import('.'.join(["Plugins", c, x, "plugin"]))
+
+                                               if not plugin.__dict__.has_key("Plugins"):
+                                                       print "Plugin %s doesn't have 'Plugin'-call." % (x)
+                                                       continue
 
-       def menuUpdate(self):
-               for menuEntry in self.menuEntries:
-                       menuupdater.addMenuItem(menuEntry[0][0], menuEntry[0][2], menuEntry[1], menuEntry[0][3])
+                                               plugins = plugin.Plugins()
+
+                                               # 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 = [ ]
+               for x in where:
+                       for p in self.plugins.get(x, [ ]):
+                               res.append(p)
+               return res
        
-       def runPlugin(self, plugin, session):
-               try:
-                       exec "import " + self.prefix + plugin[2]
-                       eval(self.prefix + plugin[2]).main(session)
-               except:
-                       print "exec of plugin failed!"
+       def clearPluginList(self):
+               self.pluginList = []
+               self.plugins = {}
+
+       def shutdown(self):
+               for p in self.pluginList[:]:
+                       self.removePlugin(p)
 
 plugins = PluginComponent()