1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
import os
from Tools.Directories import *
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.pluginList = [ ]
self.setPluginPrefix("Plugins.")
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 reReadPluginList(self, directory, depth = 1):
print "re-reading plugin list"
self.clearPluginList()
self.readPluginList(directory=directory, depth=depth)
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:
open(path + "__init__.py", "a").close()
self.readPluginList(path, modules + [x], depth - 1)
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 clearPluginList(self):
self.pluginList = []
self.plugins = {}
def shutdown(self):
for p in self.pluginList[:]:
self.removePlugin(p)
plugins = PluginComponent()
|
