X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/2d470a436a60b917a27c2055952a0a24374b994c..4b0a216bcf413989b5a7e6c1ad0275247fabca9c:/lib/python/Components/PluginComponent.py

diff --git a/lib/python/Components/PluginComponent.py b/lib/python/Components/PluginComponent.py
index 67373ea7..ae06d776 100644
--- a/lib/python/Components/PluginComponent.py
+++ b/lib/python/Components/PluginComponent.py
@@ -1,59 +1,79 @@
 import os
 
 from Tools.Directories import *
-from Screens.Menu import menuupdater
+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 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, modules = [], depth = 1):
+		"""enumerates plugins"""
+		
+		directories = os.listdir(directory)
+		
+		for x in directories:
+			path = directory + x + "/"
+			if os.path.isdir(path):
+				if fileExists(path + "plugin.py"):
+					plugin = my_import('.'.join(["Plugins"] + modules + [x, "plugin"]))
+					
+					if not plugin.__dict__.has_key("Plugins"):
+						print "Plugin %s doesn't have 'Plugin'-call." % (x)
+						continue
+					
+					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)
+						self.addPlugin(p);
+				else:
+					self.readPluginList(path, modules + [x], depth - 1)
 
-	def getPluginList(self):
-		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)
-						picturepath = plugin.getPicturePath()
-						pluginname = plugin.getPluginName()
-						try:
-							for menuEntry in plugin.getMenuRegistrationList():
-								self.menuEntries.append([menuEntry, pluginmodule])
-						except:
-							pass
+	def getPlugins(self, where):
+		"""Get list of plugins in a specific category"""
 		
-						list.append((picturepath, pluginname , x))
-			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][2], menuEntry[1], menuEntry[0][3])
+		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 menuUpdate(self):
-		for menuEntry in self.menuEntries:
-			menuupdater.addMenuItem(menuEntry[0][0], menuEntry[0][2], menuEntry[1], menuEntry[0][3])
-	
-	def runPlugin(self, plugin, session):
-		try:
-			exec "import " + self.prefix + plugin[2] + ".plugin"
-			eval(self.prefix + plugin[2] + ".plugin").main(session)
-		except:
-			print "exec of plugin failed!"
+	def shutdown(self):
+		for p in self.pluginList[:]:
+			self.removePlugin(p)
 
 plugins = PluginComponent()