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
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
|
#!/usr/bin/python
import os
import shutil
import signal
import string
import subprocess
import sys
import xml.etree.ElementTree
from distutils.spawn import find_executable
def findparentfiles(fname):
filelist = []
newlist = []
args = ['grep', '-rl', '--exclude-dir=.git', fname]
proc = subprocess.Popen(args,
stdin=subprocess.PIPE,
stdout=subprocess.PIPE,
universal_newlines=False,
preexec_fn=lambda:
signal.signal(signal.SIGPIPE, signal.SIG_DFL))
data = proc.communicate()[0]
if proc.returncode != 0:
return filelist
for filename in data.splitlines():
if filename.endswith('.yaml') and '/' not in filename:
filelist.append(filename)
else:
newlist = findparentfiles(filename)
for tempname in newlist:
filelist.append(tempname)
return filelist
jjb_cmd = find_executable('jenkins-jobs') or sys.exit('jenkins-jobs is not found.')
jjb_args = [jjb_cmd]
jjb_user = os.environ.get('JJB_USER')
jjb_password = os.environ.get('JJB_PASSWORD')
if jjb_user is not None and jjb_password is not None:
jenkins_jobs_ini = ('[job_builder]\n'
'ignore_cache=True\n'
'keep_descriptions=False\n'
'\n'
'[jenkins]\n'
'user=%s\n'
'password=%s\n'
'url=https://ci.linaro.org/\n' % (jjb_user, jjb_password))
with open('jenkins_jobs.ini', 'w') as f:
f.write(jenkins_jobs_ini)
jjb_args.append('--conf=jenkins_jobs.ini')
jjb_test_args = list(jjb_args)
jjb_delete_args = list(jjb_args)
# !!! "update" below and through out this file is replaced by "test" (using sed)
# !!! in the sanity-check job.
main_action = 'update'
jjb_args.extend([main_action, 'template.yaml'])
jjb_test_args.extend(['test', '-o', 'out/', 'template.yaml'])
jjb_delete_args.extend(['delete'])
if main_action == 'test':
# Dry-run, don't delete jobs.
jjb_delete_args.insert(0, 'echo')
try:
git_args = ['git', 'diff', '--raw',
os.environ.get('GIT_PREVIOUS_COMMIT'),
os.environ.get('GIT_COMMIT')]
proc = subprocess.Popen(git_args,
stdin=subprocess.PIPE,
stdout=subprocess.PIPE,
universal_newlines=False,
preexec_fn=lambda:
signal.signal(signal.SIGPIPE, signal.SIG_DFL))
except (OSError, ValueError) as e:
raise ValueError("%s" % e)
data = proc.communicate()[0]
if proc.returncode != 0:
raise ValueError("command has failed with code '%s'" % proc.returncode)
filelist = []
deletelist = []
files = []
for line in data.splitlines():
# Format of the git-diff; we only need OPERATION and FILE1
#
# :<OLD MODE> <NEW MODE> <OLD REF> <NEW REF> <OPERATION> <FILE1> <FILE2>
elems = line.split()
operation = elems[4][0]
filename = elems[5]
if filename.endswith('.yaml') and '/' not in filename:
# No point trying to test deleted jobs because they don't exist any
# more.
if operation == 'D':
deletelist.append(filename[:-5])
continue
# operation R100 is 100% rename, which means sixth element is the renamed file
if operation == 'R':
filename = elems[6]
# delete old job name
deletelist.append(elems[5][:-5])
filelist.append(filename)
else:
files = findparentfiles(filename)
for tempname in files:
filelist.append(tempname)
# Remove duplicate entries in the list
filelist = list(set(filelist))
for conf_filename in filelist:
with open(conf_filename) as f:
buffer = f.read()
template = string.Template(buffer)
buffer = template.safe_substitute(
AUTH_TOKEN=os.environ.get('AUTH_TOKEN'),
LT_QCOM_KEY=os.environ.get('LT_QCOM_KEY'),
LAVA_USER=os.environ.get('LAVA_USER'),
LAVA_TOKEN=os.environ.get('LAVA_TOKEN'))
with open('template.yaml', 'w') as f:
f.write(buffer)
try:
proc = subprocess.Popen(jjb_args,
stdin=subprocess.PIPE,
stdout=subprocess.PIPE,
universal_newlines=False,
preexec_fn=lambda:
signal.signal(signal.SIGPIPE, signal.SIG_DFL))
except (OSError, ValueError) as e:
raise ValueError("%s" % e)
data = proc.communicate()[0]
if proc.returncode != 0:
raise ValueError("command has failed with code '%s'" % proc.returncode)
try:
shutil.rmtree('out/', ignore_errors=True)
proc = subprocess.Popen(jjb_test_args,
stdin=subprocess.PIPE,
stdout=subprocess.PIPE,
universal_newlines=False,
preexec_fn=lambda:
signal.signal(signal.SIGPIPE, signal.SIG_DFL))
data = proc.communicate()[0]
if proc.returncode != 0:
raise ValueError("command has failed with code '%s'" % proc.returncode)
proc = subprocess.Popen(['ls', 'out/'],
stdin=subprocess.PIPE,
stdout=subprocess.PIPE,
universal_newlines=False,
preexec_fn=lambda:
signal.signal(signal.SIGPIPE, signal.SIG_DFL))
data = proc.communicate()[0]
if proc.returncode != 0:
raise ValueError("command has failed with code '%s'" % proc.returncode)
for filename in data.splitlines():
# old job conf might have been removed because the job is now generated through the template
# do not delete the job in this case
if filename in deletelist:
deletelist.remove(filename)
conf_name=os.path.splitext(conf_filename)[0]
conf_name=conf_name[:len(filename)]
if not filename.startswith(conf_name):
raise ValueError("Job name %s does not match the file it is in: %s" % (filename, conf_name))
try:
xmlroot = xml.etree.ElementTree.parse('out/' + filename).getroot()
disabled = next(xmlroot.iterfind('disabled')).text
if disabled != 'true':
continue
displayName = next(xmlroot.iterfind('displayName')).text
if displayName != 'DELETE ME':
continue
except:
continue
deletelist.append(conf_name)
except (OSError, ValueError) as e:
raise ValueError("%s" % e)
shutil.rmtree('out/', ignore_errors=True)
os.remove('template.yaml')
for deletejob in deletelist:
delete_args = list(jjb_delete_args)
delete_args.extend([deletejob])
try:
proc = subprocess.Popen(delete_args,
stdin=subprocess.PIPE,
stdout=subprocess.PIPE,
universal_newlines=False,
preexec_fn=lambda:
signal.signal(signal.SIGPIPE, signal.SIG_DFL))
data = proc.communicate()[0]
if proc.returncode != 0:
raise ValueError("command has failed with code '%s'" % proc.returncode)
print data
except (OSError, ValueError) as e:
raise ValueError("%s" % e)
if os.path.exists('jenkins_jobs.ini'):
os.remove('jenkins_jobs.ini')
|
