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
|
#!/usr/bin/env python
import os
import sys
import glob
import lxml.etree
import Bcfg2.Options
def main():
opts = dict(repo=Bcfg2.Options.SERVER_REPOSITORY)
setup = Bcfg2.Options.OptionParser(opts)
setup.parse(sys.argv[1:])
files = []
for plugin in ['Bundler', 'Rules', 'Default']:
files.extend(glob.glob(os.path.join(setup['repo'], plugin, "*")))
for bfile in files:
bdata = lxml.etree.parse(bfile)
changed = False
for svc in bdata.xpath("//Service|//BoundService"):
if "mode" not in svc.attrib:
continue
mode = svc.get("mode")
del svc.attrib["mode"]
if mode not in ["default", "supervised", "interactive_only",
"manual"]:
print("Unrecognized mode on Service:%s: %s. Assuming default" %
(svc.get("name"), mode))
mode = "default"
if mode == "default" or mode == "supervised":
svc.set("restart", "true")
svc.set("install", "true")
elif mode == "interactive_only":
svc.set("restart", "interactive")
svc.set("install", "true")
elif mode == "manual":
svc.set("restart", "false")
svc.set("install", "false")
changed = True
if changed:
print("Writing %s" % bfile)
try:
open(bfile, "w").write(lxml.etree.tostring(bdata))
except IOError:
err = sys.exc_info()[1]
print("Could not write %s: %s" % (bfile, err))
if __name__ == '__main__':
sys.exit(main())
|
