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
|
#!/usr/bin/python
#coding:utf-8
import os, sys
import yum, pickle
import tempfile
usages = \
"""
Usage: sunyums [OPTION]... [FILE]...
Output and match all dependent installation packages
Example:
sunyums packname1 packname2
sunyums packname1 packname2 --config=file --comps=comps.xml
sunyums packname1 packname2 --mandatory=1 --default=1 --options=0
Options:
--config=file.conf supply an yum config file, default: optional
--comps=comps.xml supply an parsed comps.xml default: optional
--mandatory=True include mandatory packages default: True
--default=True include mandatory packages default: True
--options=False include mandatory packages default: False
"""
class Application(object):
def __init__(self, args):
self.args = args[1:]
self.yums = yum.YumBase()
self.comps = None
self.config = None
self.groups = []
self.mandatory = True
self.default = True
self.options = False
self.basePacks = []
self.origPacks = []
self.packages = []
def str2bool(self, s):
"""Converts an on/off, yes/no, true/false string to 1/0."""
if s and s.upper() in [ 'ON', 'YES', 'Y', 'TRUE', '1', 'ENABLED', 'ENABLE']:
return True
else:
return False
def usages(self):
print usages
sys.exit(0)
def parseArgs(self):
if not self.args:
self.usages()
for arg in self.args:
if arg in [ '-h', '--help']:
self.usages()
elif arg.startswith('--comps='):
self.comps = arg.split('=')[1]
elif arg.startswith('--config='):
self.config = arg.split('=')[1]
elif arg.startswith('--mandatory='):
self.mandatory = self.str2bool(arg.split('=')[1])
elif arg.startswith('--default='):
self.default = self.str2bool(arg.split('=')[1])
elif arg.startswith('--options='):
self.options = self.str2bool(arg.split('=')[1])
else:
self.groups.append(arg)
def depends(self):
pkgs = []
avail = self.yums.pkgSack.returnNewestByNameArch()
for p in avail:
if p.name in self.basePacks:
pkgs.append(p)
done = 0
while not done:
done = 1
results = self.yums.findDeps(pkgs)
for pkg in results.keys():
for req in results[pkg].keys():
reqlist = results[pkg][req]
for r in reqlist:
if r.name not in self.basePacks:
self.basePacks.append(r.name)
pkgs.append(r)
done = 0
def allgroups(self):
for grp in self.yums.comps.groups:
self.packages.extend(grp.packages)
def handerPackages(self, name):
if not self.packages:
self.allgroups()
if name in self.packages and \
name not in self.basePacks:
self.basePacks.append(name)
if name not in self.origPacks:
self.origPacks.append(name)
def handerGroups(self, name):
groups = []
if not self.yums.comps.has_group(name):
return
valid_groups = self.yums.comps.return_group(name.encode('utf-8'))
if self.mandatory:
groups.extend(valid_groups.mandatory_packages.keys())
if self.default:
groups.extend(valid_groups.default_packages.keys())
if self.options:
groups.extend(valid_groups.options_packages.keys())
for package in groups:
self.handerPackages(package)
def handerEnviron(self, name):
groups = []
if not self.yums.comps.has_environment(name):
return
valid_environ = self.yums.comps.return_environment(name)
for grp in valid_environ.groups:
self.handerGroups(grp)
def run(self):
if self.comps and os.path.exists(self.comps):
self.yums.comps.add(self.comps)
if self.config and os.path.exists(self.config):
self.yums.doConfigSetup(fn=self.config, init_plugins=False)
self.yums.conf.cache = 0
for rpm in self.groups:
if rpm[0] == '@':
self.handerGroups(rpm[1:])
elif rpm[0] == '^':
self.handerEnviron(rpm[1:])
else:
self.handerPackages(rpm)
self.depends()
for o in self.origPacks:
if o not in self.basePacks:
print '#%s' % o
for p in self.basePacks:
print p
if __name__ == "__main__":
app = Application(sys.argv)
app.parseArgs()
app.allgroups()
app.run()
|
