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
|
# The following function is a thin wrapper around iter_entry_points. The reason it
# is in this separate file is that when making the Mac app, py2app doesn't
# support entry points, so we replace this function with a version that has the
# entry points we want hardcoded. If this function was in glue/main.py, the
# reference to the iter_plugin_entry_points function in load_plugin would be
# evaluated at compile time rather than at runtime, so the patched version
# wouldn't be used.
import os
from collections import defaultdict
def iter_plugin_entry_points():
from pkg_resources import iter_entry_points
return iter_entry_points(group='glue.plugins', name=None)
class PluginConfig(object):
def __init__(self, plugins={}):
self.plugins = defaultdict(lambda: True)
self.plugins.update(plugins)
def __str__(self):
string = ""
for plugin in sorted(self.plugins):
string += "{0}: {1}\n".format(plugin, self.plugins[plugin])
return string
@classmethod
def load(cls):
# Import at runtime because some tests change this value. We also don't
# just import the variable directly otherwise it is cached.
from glue import config
cfg_dir = config.CFG_DIR
plugin_cfg = os.path.join(cfg_dir, 'plugins.cfg')
import configparser
config = configparser.ConfigParser()
read = config.read(plugin_cfg)
if len(read) == 0 or not config.has_section('plugins'):
return cls()
plugins = {}
for name, enabled in config.items('plugins'):
plugins[name] = bool(int(enabled))
self = cls(plugins=plugins)
return self
def save(self):
# Import at runtime because some tests change this value. We also don't
# just import the variable directly otherwise it is cached.
from glue import config
cfg_dir = config.CFG_DIR
plugin_cfg = os.path.join(cfg_dir, 'plugins.cfg')
import configparser
config = configparser.ConfigParser()
config.add_section('plugins')
for key in sorted(self.plugins):
config.set('plugins', key, value=str(int(self.plugins[key])))
if not os.path.exists(cfg_dir):
os.mkdir(cfg_dir)
with open(plugin_cfg, 'w') as fout:
config.write(fout)
def filter(self, keep):
"""
Keep only certain plugins.
This is used to filter out plugins that are not installed.
"""
for key in list(self.plugins.keys())[:]:
if key not in keep:
self.plugins.pop(key)
|
