blob: 53a3a04f3663636862e3bd4c068c4c28570dc38d [file] [log] [blame]
import logging
import os
from autotest_lib.client.common_lib import error
from autotest_lib.client.common_lib.cros.graphite import stats
from autotest_lib.server.cros import provision
# A string of the form 'label1,label2:value,label3'.
job_labels = locals().get('job_labels') or ','.join(args)
labels_list = [l.strip() for l in job_labels.split(',') if l]
def cleanup(machine):
timer = None
try:
job.record('START', None, 'cleanup')
host = hosts.create_host(machine, initialize=False, auto_monitor=False)
timer = stats.Timer('cleanup_time')
timer.start()
# Try to save /var/log files. If the dut is not sshable, try to restart
# with servo. This is a temp fix to collect log for test failed with dut
# not returning from reboot.
# TODO(dshi): This temp fix should be removed after crash collect work
# is completed, crbug.com/336985
try:
host.ssh_ping()
except error.AutoservSshPingHostError:
# Try to restart dut with servo.
host._servo_repair_power()
local_log_dir = os.path.join(job.resultdir, machine)
host.collect_logs('/var/log', local_log_dir, ignore_errors=True)
host.cleanup()
provision.run_special_task_actions(job, host, labels_list,
provision.Cleanup)
except Exception as e:
logging.exception(e)
job.record('END FAIL', None, 'cleanup')
# See the provision control segment for the explanation of why we're
# doing this.
raise Exception('')
else:
job.record('END GOOD', None, 'cleanup',
'%s cleaned successfully' % machine)
finally:
if timer:
timer.stop()
job.parallel_simple(cleanup, machines, log=False)
# vim: set syntax=python :