autobuild: added much better email reporting
[nivanova/samba-autobuild/.git] / script / autobuild.py
1 #!/usr/bin/env python
2 # run tests on all Samba subprojects and push to a git tree on success
3 # Copyright Andrew Tridgell 2010
4 # released under GNU GPL v3 or later
5
6 from subprocess import Popen, PIPE
7 import os, signal, tarfile, sys, time
8 from optparse import OptionParser
9 import smtplib
10 from email.mime.text import MIMEText
11
12 samba_master = os.getenv('SAMBA_MASTER', 'git://git.samba.org/samba.git')
13 samba_master_ssh = os.getenv('SAMBA_MASTER_SSH', 'git+ssh://git.samba.org/data/git/samba.git')
14
15 cleanup_list = []
16
17 os.putenv('CC', "ccache gcc")
18
19 tasks = {
20     "source3" : [ "./autogen.sh",
21                   "./configure.developer ${PREFIX}",
22                   "make basics",
23                   "make -j 4 everything", # don't use too many processes
24                   "make install",
25                   "TDB_NO_FSYNC=1 make test FAIL_IMMEDIATELY=1" ],
26
27     "source4" : [ "./autogen.sh",
28                   "./configure.developer ${PREFIX}",
29                   "make -j",
30                   "make install",
31                   "TDB_NO_FSYNC=1 make test FAIL_IMMEDIATELY=1" ],
32
33     "source4/lib/ldb" : [ "./autogen-waf.sh",
34                           "./configure --enable-developer -C ${PREFIX}",
35                           "make -j",
36                           "make install",
37                           "make test" ],
38
39     "lib/tdb" : [ "./autogen-waf.sh",
40                   "./configure --enable-developer -C ${PREFIX}",
41                   "make -j",
42                   "make install",
43                   "make test" ],
44
45     "lib/talloc" : [ "./autogen-waf.sh",
46                      "./configure --enable-developer -C ${PREFIX}",
47                      "make -j",
48                      "make install",
49                      "make test" ],
50
51     "lib/replace" : [ "./autogen-waf.sh",
52                       "./configure --enable-developer -C ${PREFIX}",
53                       "make -j",
54                       "make install",
55                       "make test" ],
56
57     "lib/tevent" : [ "./autogen-waf.sh",
58                      "./configure --enable-developer -C ${PREFIX}",
59                      "make -j",
60                      "make install",
61                      "make test" ],
62 }
63
64 retry_task = [ '''set -e
65                 git remote add -t master master %s
66                 git fetch master
67                 while :; do
68                   sleep 60
69                   git describe master/master > old_master.desc
70                   git fetch master
71                   git describe master/master > master.desc
72                   diff old_master.desc master.desc
73                 done
74                ''' % samba_master]
75
76 def run_cmd(cmd, dir=".", show=None, output=False, checkfail=True):
77     cwd = os.getcwd()
78     os.chdir(dir)
79     if show is None:
80         show = options.verbose
81     if show:
82         print("Running: '%s' in '%s'" % (cmd, dir))
83     if output:
84         ret = Popen([cmd], shell=True, stdout=PIPE).communicate()[0]
85         os.chdir(cwd)
86         return ret
87     ret = os.system(cmd)
88     os.chdir(cwd)
89     if checkfail and ret != 0:
90         raise Exception("FAILED %s: %d" % (cmd, ret))
91     return ret
92
93 class builder:
94     '''handle build of one directory'''
95     def __init__(self, name, sequence):
96         self.name = name
97
98         if name in ['pass', 'fail', 'retry']:
99             self.dir = "."
100         else:
101             self.dir = self.name
102
103         self.tag = self.name.replace('/', '_')
104         self.sequence = sequence
105         self.next = 0
106         self.stdout_path = "%s/%s.stdout" % (gitroot, self.tag)
107         self.stderr_path = "%s/%s.stderr" % (gitroot, self.tag)
108         if options.verbose:
109             print("stdout for %s in %s" % (self.name, self.stdout_path))
110             print("stderr for %s in %s" % (self.name, self.stderr_path))
111         run_cmd("rm -f %s %s" % (self.stdout_path, self.stderr_path))
112         self.stdout = open(self.stdout_path, 'w')
113         self.stderr = open(self.stderr_path, 'w')
114         self.stdin  = open("/dev/null", 'r')
115         self.sdir = "%s/%s" % (testbase, self.tag)
116         self.prefix = "%s/prefix/%s" % (testbase, self.tag)
117         run_cmd("rm -rf %s" % self.sdir)
118         cleanup_list.append(self.sdir)
119         cleanup_list.append(self.prefix)
120         os.makedirs(self.sdir)
121         run_cmd("rm -rf %s" % self.sdir)
122         run_cmd("git clone --shared %s %s" % (gitroot, self.sdir))
123         self.start_next()
124
125     def start_next(self):
126         if self.next == len(self.sequence):
127             print '%s: Completed OK' % self.name
128             self.done = True
129             return
130         self.cmd = self.sequence[self.next].replace("${PREFIX}", "--prefix=%s" % self.prefix)
131         print '%s: Running %s' % (self.name, self.cmd)
132         cwd = os.getcwd()
133         os.chdir("%s/%s" % (self.sdir, self.dir))
134         self.proc = Popen(self.cmd, shell=True,
135                           stdout=self.stdout, stderr=self.stderr, stdin=self.stdin)
136         os.chdir(cwd)
137         self.next += 1
138
139
140 class buildlist:
141     '''handle build of multiple directories'''
142     def __init__(self, tasklist, tasknames):
143         global tasks
144         self.tlist = []
145         self.tail_proc = None
146         self.retry = None
147         if tasknames == ['pass']:
148             tasks = { 'pass' : [ '/bin/true' ]}
149         if tasknames == ['fail']:
150             tasks = { 'fail' : [ '/bin/false' ]}
151         if tasknames == []:
152             tasknames = tasklist
153         for n in tasknames:
154             b = builder(n, tasks[n])
155             self.tlist.append(b)
156         if options.retry:
157             self.retry = builder('retry', retry_task)
158             self.need_retry = False
159
160     def kill_kids(self):
161         if self.tail_proc is not None:
162             self.tail_proc.terminate()
163             self.tail_proc.wait()
164             self.tail_proc = None
165         if self.retry is not None:
166             self.retry.proc.terminate()
167             self.retry.proc.wait()
168             self.retry = None
169         for b in self.tlist:
170             if b.proc is not None:
171                 run_cmd("killbysubdir %s > /dev/null 2>&1" % b.sdir, checkfail=False)
172                 b.proc.terminate()
173                 b.proc.wait()
174                 b.proc = None
175
176     def wait_one(self):
177         while True:
178             none_running = True
179             for b in self.tlist:
180                 if b.proc is None:
181                     continue
182                 none_running = False
183                 b.status = b.proc.poll()
184                 if b.status is None:
185                     continue
186                 b.proc = None
187                 return b
188             if options.retry:
189                 ret = self.retry.proc.poll()
190                 if ret is not None:
191                     self.need_retry = True
192                     self.retry = None
193                     return None
194             if none_running:
195                 return None
196             time.sleep(0.1)
197
198     def run(self):
199         while True:
200             b = self.wait_one()
201             if options.retry and self.need_retry:
202                 self.kill_kids()
203                 print("retry needed")
204                 return (0, "retry")
205             if b is None:
206                 break
207             if os.WIFSIGNALED(b.status) or os.WEXITSTATUS(b.status) != 0:
208                 self.kill_kids()
209                 return (b.status, b.name, b.tag, "%s: failed '%s' with status %d" % (b.name, b.cmd, b.status))
210             b.start_next()
211         self.kill_kids()
212         return (0, None, None, "All OK")
213
214     def tarlogs(self, fname):
215         tar = tarfile.open(fname, "w:gz")
216         for b in self.tlist:
217             tar.add(b.stdout_path, arcname="%s.stdout" % b.tag)
218             tar.add(b.stderr_path, arcname="%s.stderr" % b.tag)
219         tar.close()
220
221     def remove_logs(self):
222         for b in self.tlist:
223             os.unlink(b.stdout_path)
224             os.unlink(b.stderr_path)
225
226     def start_tail(self):
227         cwd = os.getcwd()
228         cmd = "tail -f *.stdout *.stderr"
229         os.chdir(testbase)
230         self.tail_proc = Popen(cmd, shell=True)
231         os.chdir(cwd)
232
233
234 def cleanup():
235     if options.nocleanup:
236         return
237     print("Cleaning up ....")
238     for d in cleanup_list:
239         run_cmd("rm -rf %s" % d)
240
241
242 def find_git_root():
243     '''get to the top of the git repo'''
244     cwd=os.getcwd()
245     while os.getcwd() != '/':
246         try:
247             os.stat(".git")
248             ret = os.getcwd()
249             os.chdir(cwd)
250             return ret
251         except:
252             os.chdir("..")
253             pass
254     os.chdir(cwd)
255     return None
256
257 def rebase_tree(url):
258     print("Rebasing on %s" % url)
259     run_cmd("git remote add -t master master %s" % url, show=True, dir=test_master)
260     run_cmd("git fetch master", show=True, dir=test_master)
261     if options.fix_whitespace:
262         run_cmd("git rebase --whitespace=fix master/master", show=True, dir=test_master)
263     else:
264         run_cmd("git rebase master/master", show=True, dir=test_master)
265     diff = run_cmd("git --no-pager diff HEAD master/master", dir=test_master, output=True)
266     if diff == '':
267         print("No differences between HEAD and master/master - exiting")
268         sys.exit(0)
269
270 def push_to(url):
271     print("Pushing to %s" % url)
272     if options.mark:
273         run_cmd("EDITOR=script/commit_mark.sh git notes edit HEAD", dir=test_master)
274     run_cmd("git remote add -t master pushto %s" % url, show=True, dir=test_master)
275     run_cmd("git push pushto +HEAD:master", show=True, dir=test_master)
276
277 def_testbase = os.getenv("AUTOBUILD_TESTBASE", "/memdisk/%s" % os.getenv('USER'))
278
279 parser = OptionParser()
280 parser.add_option("", "--tail", help="show output while running", default=False, action="store_true")
281 parser.add_option("", "--keeplogs", help="keep logs", default=False, action="store_true")
282 parser.add_option("", "--nocleanup", help="don't remove test tree", default=False, action="store_true")
283 parser.add_option("", "--testbase", help="base directory to run tests in (default %s)" % def_testbase,
284                   default=def_testbase)
285 parser.add_option("", "--passcmd", help="command to run on success", default=None)
286 parser.add_option("", "--verbose", help="show all commands as they are run",
287                   default=False, action="store_true")
288 parser.add_option("", "--rebase", help="rebase on the given tree before testing",
289                   default=None, type='str')
290 parser.add_option("", "--rebase-master", help="rebase on %s before testing" % samba_master,
291                   default=False, action='store_true')
292 parser.add_option("", "--pushto", help="push to a git url on success",
293                   default=None, type='str')
294 parser.add_option("", "--push-master", help="push to %s on success" % samba_master_ssh,
295                   default=False, action='store_true')
296 parser.add_option("", "--mark", help="add a Tested-By signoff before pushing",
297                   default=False, action="store_true")
298 parser.add_option("", "--fix-whitespace", help="fix whitespace on rebase",
299                   default=False, action="store_true")
300 parser.add_option("", "--retry", help="automatically retry if master changes",
301                   default=False, action="store_true")
302 parser.add_option("", "--email", help="send email to the given address on failure",
303                   type='str', default=None)
304
305
306 def email_failure(status, failed_task, failed_tag, errstr):
307     '''send an email to options.email about the failure'''
308     user = os.getenv("USER")
309     text = '''
310 Dear Developer,
311
312 Your autobuild failed when trying to test %s with the following error:
313    %s
314
315 the autobuild has been abandoned. Please fix the error and resubmit.
316
317 You can see logs of the failed task here:
318
319   http://git.samba.org/%s/samba-autobuild/%s.stdout
320   http://git.samba.org/%s/samba-autobuild/%s.stderr
321
322 or you can get full logs of all tasks in this job here:
323
324   http://git.samba.org/%s/samba-autobuild/logs.tar.gz
325
326 ''' % (failed_task, errstr, user, failed_tag, user, failed_tag, user)
327     msg = MIMEText(text)
328     msg['Subject'] = 'autobuild failure for task %s' % failed_task
329     msg['From'] = 'autobuild@samba.org'
330     msg['To'] = options.email
331
332     s = smtplib.SMTP()
333     s.connect()
334     s.sendmail(msg['From'], [msg['To']], msg.as_string())
335     s.quit()
336
337 (options, args) = parser.parse_args()
338
339 if options.retry:
340     if not options.rebase_master and options.rebase is None:
341         raise Exception('You can only use --retry if you also rebase')
342
343 testbase = "%s/b%u" % (options.testbase, os.getpid())
344 test_master = "%s/master" % testbase
345
346 gitroot = find_git_root()
347 if gitroot is None:
348     raise Exception("Failed to find git root")
349
350 try:
351     os.makedirs(testbase)
352 except Exception, reason:
353     raise Exception("Unable to create %s : %s" % (testbase, reason))
354 cleanup_list.append(testbase)
355
356 while True:
357     try:
358         run_cmd("rm -rf %s" % test_master)
359         cleanup_list.append(test_master)
360         run_cmd("git clone --shared %s %s" % (gitroot, test_master))
361     except:
362         cleanup()
363         raise
364
365     try:
366         if options.rebase is not None:
367             rebase_tree(options.rebase)
368         elif options.rebase_master:
369             rebase_tree(samba_master)
370         blist = buildlist(tasks, args)
371         if options.tail:
372             blist.start_tail()
373         (status, failed_task, failed_tag, errstr) = blist.run()
374         if status != 0 or errstr != "retry":
375             break
376         cleanup()
377     except:
378         cleanup()
379         raise
380
381 blist.kill_kids()
382 if options.tail:
383     print("waiting for tail to flush")
384     time.sleep(1)
385
386 if status == 0:
387     print errstr
388     if options.passcmd is not None:
389         print("Running passcmd: %s" % options.passcmd)
390         run_cmd(options.passcmd, dir=test_master)
391     if options.pushto is not None:
392         push_to(options.pushto)
393     elif options.push_master:
394         push_to(samba_master_ssh)
395     if options.keeplogs:
396         blist.tarlogs("logs.tar.gz")
397         print("Logs in logs.tar.gz")
398     blist.remove_logs()
399     cleanup()
400     print(errstr)
401     sys.exit(0)
402
403 # something failed, gather a tar of the logs
404 blist.tarlogs("logs.tar.gz")
405
406 if options.email is not None:
407     email_failure(status, failed_task, failed_tag, errstr)
408
409 cleanup()
410 print(errstr)
411 print("Logs in logs.tar.gz")
412 sys.exit(status)