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
|
#!/usr/bin/python
import copy
import time
import yaml
import os
import sys
class KSWriter():
def __init__(self, im, rep, out):
self.image_filename = im
self.repo_filename = rep
self.outdir = out
self.image_stream = file(self.image_filename, 'r')
self.repo_stream = file(self.repo_filename, 'r')
self.extra = {}
self.repo_meta = yaml.load(self.repo_stream)
self.image_meta = yaml.load(self.image_stream)
pass
def merge(*input):
return list(reduce(set.union, input, set()))
def dump(self):
print yaml.dump(yaml.load(self.stream))
def parse(self, img):
conf = copy.copy(self.image_meta['Default'])
plat = copy.copy(self.image_meta[img['Platform']])
conf.update(plat)
conf.update(img)
lval = ['Repos', 'Groups', 'PostScripts', 'NoChrootScripts', 'RemovePackages', 'ExtraPackages']
lvald = {}
for l in lval:
full = []
if self.image_meta['Default'].has_key(l) and self.image_meta['Default'][l]:
full = full + self.image_meta['Default'][l]
if plat.has_key(l) and plat[l]:
full = full + plat[l]
if img.has_key(l) and img[l]:
full = full + img[l]
lvald[l] = sorted(set(full), key=full.index)
#print full
conf.update(lvald)
#print conf
postscript = ""
for scr in conf['PostScripts']:
if os.path.exists('./custom/scripts/%s.post' %scr):
f = open('./custom/scripts/%s.post' %scr, 'r')
postscript += f.read()
postscript += "\n"
f.close()
else:
print './custom/scripts/%s.post not found, skipping.' %scr
nochrootscript = ""
for scr in conf['NoChrootScripts']:
if os.path.exists('./custom/scripts/%s.nochroot' %scr):
f = open('./custom/scripts/%s.nochroot' %scr, 'r')
nochrootscript += f.read()
nochrootscript += "\n"
f.close()
else:
print './custom/scripts/%s.nochroot not found, skipping.' %scr
ptab = ""
for g in [ plat, img ]:
if g.has_key("Part"):
f = open("./custom/part/%s" %g['Part'] )
ptab = f.read()
f.close()
conf['Part'] = ptab
conf['Post'] = postscript
conf['NoChroot'] = nochrootscript
return conf
def process_files(self, meta, repos):
new_repos = []
if meta.has_key("Architecture") and meta['Architecture']:
for repo in repos:
r = {}
r['Name'] = repo['Name']
if repo.has_key('Options'):
r['Options'] = repo['Options']
r['Url'] = repo['Url'].replace("@ARCH@", meta['Architecture'])
r['Url'] = r['Url'].replace("@RELEASE@", meta['Baseline'])
new_repos.append(r)
else:
new_repos = repos
nameSpace = {'metadata': meta, 'repos': new_repos}
t = kickstart(searchList=[nameSpace])
a = str(t)
if meta.has_key('FileName') and meta['FileName']:
f = None
if meta.has_key("Baseline"):
mkdir_p(meta['Baseline'])
f = open("%s/%s/%s.ks" %( self.outdir, meta['Baseline'], meta['FileName'] ), 'w')
else:
f = open("%s/%s.ks" %( self.outdir, meta['FileName'] ), 'w')
f.write(a)
f.close()
|
