X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/bb438bec17665a907c28ca47d665766c7b26be79..4c1d3d2f5cf39f72bf85041a6ba6665350ea742e:/lib/python/Components/PluginComponent.py

diff --git a/lib/python/Components/PluginComponent.py b/lib/python/Components/PluginComponent.py
old mode 100644
new mode 100755
index 5315d3d9..5e439fdf
--- a/lib/python/Components/PluginComponent.py
+++ b/lib/python/Components/PluginComponent.py
@@ -1,52 +1,128 @@
-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
-#import Plugins
+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):
-		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 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]
-			eval(self.prefix + plugin[2]).main(session)
-		except:
-			print "exec of plugin failed!"
+	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.pyo") 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 not isinstance(plugins, 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 not p.internal and 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 not isinstance(where, list):
+			where = [ where ]
+		res = [ ]
+
+		for x in where:
+			res.extend(self.plugins.get(x, [ ]))
+
+		return  res
+
+	def getPluginsForMenu(self, menuid):
+		res = [ ]
+		for p in self.getPlugins(PluginDescriptor.WHERE_MENU):
+			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 = [ ]
+
+	def getNextWakeupTime(self):
+		wakeup = -1
+		for p in self.pluginList:
+			current = p.getWakeupTime()
+			if current > -1 and (wakeup > current or wakeup == -1):
+				wakeup = current
+		return int(wakeup)
 
 plugins = PluginComponent()