104 lines
3.7 KiB
Python
104 lines
3.7 KiB
Python
|
#!/usr/bin/python
|
||
|
|
#
|
||
|
|
# Author: Patrick Ohly <patrick.ohly@intel.com>
|
||
|
|
# Copyright: Copyright (C) 2015 Intel Corporation
|
||
|
|
#
|
||
|
|
# This file is licensed under the MIT license, see COPYING.MIT in
|
||
|
|
# this source distribution for the terms.
|
||
|
|
|
||
|
|
# Runs a command, pipes its output to stdout, and injects status
|
||
|
|
# reports at regular time interval.
|
||
|
|
#
|
||
|
|
# This ensures that TravisCI does not abort the command just because
|
||
|
|
# it is silent for more than 10 minutes, as it can happen with bitbake
|
||
|
|
# when working on a single complex task, like "bitbake linux-yocto".
|
||
|
|
#
|
||
|
|
# Piping bitbake stdout has the advantage that bitbake enters
|
||
|
|
# non-interactive output mode, which it would do when run by TravisCI
|
||
|
|
# directly.
|
||
|
|
#
|
||
|
|
# Finally, the default status messages give some sense of memory
|
||
|
|
# and disk usage, which is critical in the rather constrained
|
||
|
|
# TravisCI environments.
|
||
|
|
|
||
|
|
import errno
|
||
|
|
import optparse
|
||
|
|
import signal
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
import time
|
||
|
|
|
||
|
|
parser = optparse.OptionParser()
|
||
|
|
parser.add_option("-s", "--status",
|
||
|
|
help="invoked in a shell when it is time for a status report",
|
||
|
|
# 200 columns is readable in the TravisCI Web UI without wrapping.
|
||
|
|
# Depends of course on screen and font size. Resizing top output
|
||
|
|
# only works (and is needed) on the more recent Trusty TravisCI
|
||
|
|
# environment.
|
||
|
|
default="date; free; df -h .; COLUMNS=200 LINES=30 top -w -b -n 1 2>/dev/null || top -n 1; ps x --cols 200 --forest",
|
||
|
|
metavar="SHELL-CMD")
|
||
|
|
parser.add_option("-i", "--interval",
|
||
|
|
help="repeat status at intervals of this amount of seconds, 0 to disable",
|
||
|
|
default=300,
|
||
|
|
metavar="SECONDS", type="int")
|
||
|
|
parser.add_option("-d", "--deadline",
|
||
|
|
help="stop execution when reaching the given time",
|
||
|
|
default=time.time,
|
||
|
|
metavar="SECONDS-SINCE-EPOCH", type="int")
|
||
|
|
|
||
|
|
(options, args) = parser.parse_args()
|
||
|
|
|
||
|
|
def check_deadline(now):
|
||
|
|
if options.deadline > 0 and options.deadline < now:
|
||
|
|
print "\n\n*** travis-cmd-wrapper: deadline reached, shutting down ***\n\n"
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
# Set up status alarm. When we have a deadline, we need to check more often
|
||
|
|
# and/or sooner. Sending a SIGALRM manually will also trigger a status report
|
||
|
|
# (not really possible in TravisCI, but may be useful elsewhere).
|
||
|
|
now = time.time()
|
||
|
|
next_status = now + options.interval
|
||
|
|
alarm_interval = max(options.interval, 0)
|
||
|
|
if options.deadline:
|
||
|
|
check_deadline(now)
|
||
|
|
if options.deadline < now + 60:
|
||
|
|
# Wake up a little too late, to be sure that we trigger the if check.
|
||
|
|
deadline_alarm_interval = max(int(options.deadline + 2 - now), 1)
|
||
|
|
elif next_status > 60:
|
||
|
|
deadline_alarm_interval = 60
|
||
|
|
if deadline_alarm_interval < alarm_interval:
|
||
|
|
alarm_interval = deadline_alarm_interval
|
||
|
|
|
||
|
|
def status(signum, frame):
|
||
|
|
global next_status
|
||
|
|
now = time.time()
|
||
|
|
if options.interval < 0 or now >= next_status:
|
||
|
|
subprocess.call(options.status, shell=True)
|
||
|
|
next_status = now + options.interval
|
||
|
|
check_deadline(now)
|
||
|
|
if alarm_interval > 0:
|
||
|
|
signal.alarm(alarm_interval)
|
||
|
|
|
||
|
|
# Run command.
|
||
|
|
try:
|
||
|
|
cmd = subprocess.Popen(args, stdout=subprocess.PIPE)
|
||
|
|
|
||
|
|
# Arm timer and handler.
|
||
|
|
signal.signal(signal.SIGALRM, status)
|
||
|
|
if alarm_interval > 0:
|
||
|
|
signal.alarm(alarm_interval)
|
||
|
|
|
||
|
|
while cmd.poll() is None:
|
||
|
|
try:
|
||
|
|
line = cmd.stdout.readline()
|
||
|
|
sys.stdout.write(line)
|
||
|
|
sys.stdout.flush()
|
||
|
|
except IOError, ex:
|
||
|
|
if ex.errno != errno.EINTR:
|
||
|
|
raise
|
||
|
|
finally:
|
||
|
|
# If we go down, so must our child...
|
||
|
|
if cmd.poll() is None:
|
||
|
|
cmd.kill()
|
||
|
|
|
||
|
|
exit(1 if cmd.returncode else 0)
|