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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
|
# Copyright 2007 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2
# $Id$
import os
from ConfigParser import SafeConfigParser, NoOptionError
from portage import load_mod
from portage.const import USER_CONFIG_PATH, GLOBAL_CONFIG_PATH
SETPREFIX = "@"
def get_boolean(options, name, default):
if not name in options:
return default
elif options[name].lower() in ("1", "yes", "on", "true"):
return True
elif options[name].lower() in ("0", "no", "off", "false"):
return False
else:
raise SetConfigError("invalid value '%s' for option '%s'" % (options[name], name))
class SetConfigError(Exception):
pass
class SetConfig(SafeConfigParser):
def __init__(self, paths, settings, trees):
SafeConfigParser.__init__(self)
self.read(paths)
self.errors = []
self.psets = {}
self.trees = trees
self.settings = settings
self._parsed = False
self.active = []
def update(self, setname, options):
self.errors = []
if not setname in self.psets:
options["name"] = setname
options["world-candidate"] = "False"
# for the unlikely case that there is already a section with the requested setname
import random
while setname in self.sections():
setname = "%08d" % random.randint(0, 10**10)
self.add_section(setname)
for k, v in options.items():
self.set(setname, k, v)
else:
section = self.psets[setname].creator
if self.has_option(section, "multiset") and self.getboolean(section, "multiset"):
self.errors.append("Invalid request to reconfigure set '%s' generated by multiset section '%s'" % (setname, section))
return
for k, v in options.items():
self.set(section, k, v)
self._parse(update=True)
def _parse(self, update=False):
if self._parsed and not update:
return
for sname in self.sections():
# find classname for current section, default to file based sets
if not self.has_option(sname, "class"):
classname = "portage.sets.files.StaticFileSet"
else:
classname = self.get(sname, "class")
# try to import the specified class
try:
setclass = load_mod(classname)
except (ImportError, AttributeError):
self.errors.append("Could not import '%s' for section '%s'" % (classname, sname))
continue
# prepare option dict for the current section
optdict = {}
for oname in self.options(sname):
optdict[oname] = self.get(sname, oname)
# create single or multiple instances of the given class depending on configuration
if self.has_option(sname, "multiset") and self.getboolean(sname, "multiset"):
if hasattr(setclass, "multiBuilder"):
newsets = {}
try:
newsets = setclass.multiBuilder(optdict, self.settings, self.trees)
except SetConfigError, e:
self.errors.append("Configuration error in section '%s': %s" % (sname, str(e)))
continue
for x in newsets:
if x in self.psets and not update:
self.errors.append("Redefinition of set '%s' (sections: '%s', '%s')" % (x, self.psets[x].creator, sname))
newsets[x].creator = sname
if self.has_option(sname, "world-candidate") and not self.getboolean(sname, "world-candidate"):
newsets[x].world_candidate = False
self.psets.update(newsets)
else:
self.errors.append("Section '%s' is configured as multiset, but '%s' doesn't support that configuration" % (sname, classname))
continue
else:
try:
setname = self.get(sname, "name")
except NoOptionError:
setname = sname
if setname in self.psets and not update:
self.errors.append("Redefinition of set '%s' (sections: '%s', '%s')" % (setname, self.psets[setname].creator, sname))
if hasattr(setclass, "singleBuilder"):
try:
self.psets[setname] = setclass.singleBuilder(optdict, self.settings, self.trees)
self.psets[setname].creator = sname
if self.has_option(sname, "world-candidate") and not self.getboolean(sname, "world-candidate"):
self.psets[setname].world_candidate = False
except SetConfigError, e:
self.errors.append("Configuration error in section '%s': %s" % (sname, str(e)))
continue
else:
self.errors.append("'%s' does not support individual set creation, section '%s' must be configured as multiset" % (classname, sname))
continue
self._parsed = True
def getSets(self):
self._parse()
return self.psets.copy()
def getSetAtoms(self, setname, ignorelist=None):
myset = self.getSets()[setname]
myatoms = myset.getAtoms()
extend = set()
remove = set()
intersect = set()
if ignorelist is None:
ignorelist = set()
if not setname in ignorelist:
if self.has_option(myset.creator, "extend"):
extend.update(self.get(myset.creator, "extend").split())
if self.has_option(myset.creator, "remove"):
remove.update(self.get(myset.creator, "remove").split())
if self.has_option(myset.creator, "intersect"):
intersect.update(self.get(myset.creator, "intersect").split())
ignorelist.add(setname)
for n in myset.getNonAtoms():
if n.startswith(SETPREFIX) and n[len(SETPREFIX):] in self.psets:
extend.add(n[len(SETPREFIX):])
for s in ignorelist:
extend.discard(s)
remove.discard(s)
intersect.discard(s)
for s in extend:
myatoms.update(self.getSetAtoms(s, ignorelist=ignorelist))
for s in remove:
myatoms.difference_update(self.getSetAtoms(s, ignorelist=ignorelist))
for s in intersect:
myatoms.intersection_update(self.getSetAtoms(s, ignorelist=ignorelist))
return myatoms
def load_default_config(settings, trees):
setconfigpaths = [os.path.join(GLOBAL_CONFIG_PATH, "sets.conf")]
setconfigpaths.append(os.path.join(settings["PORTDIR"], "sets.conf"))
setconfigpaths += [os.path.join(x, "sets.conf") for x in settings["PORTDIR_OVERLAY"].split()]
setconfigpaths.append(os.path.join(settings["PORTAGE_CONFIGROOT"],
USER_CONFIG_PATH.lstrip(os.path.sep), "sets.conf"))
return SetConfig(setconfigpaths, settings, trees)
# adhoc test code
if __name__ == "__main__":
import portage
sc = load_default_config(portage.settings, portage.db["/"])
l, e = sc.getSets()
for x in l:
print x+":"
print "DESCRIPTION = %s" % l[x].getMetadata("Description")
for n in sorted(l[x].getAtoms()):
print "- "+n
print
|