| OLD | NEW |
| (Empty) |
| 1 # Copyright 2013 The Chromium Authors. All rights reserved. | |
| 2 # Use of this source code is governed by a BSD-style license that can be | |
| 3 # found in the LICENSE file. | |
| 4 | |
| 5 import json | |
| 6 import optparse | |
| 7 import os | |
| 8 import subprocess | |
| 9 import sys | |
| 10 | |
| 11 import bb_annotations | |
| 12 | |
| 13 sys.path.append(os.path.join(os.path.dirname(__file__), '..')) | |
| 14 import devil_chromium # pylint: disable=unused-import | |
| 15 from devil.utils import cmd_helper | |
| 16 from pylib import constants | |
| 17 | |
| 18 | |
| 19 TESTING = 'BUILDBOT_TESTING' in os.environ | |
| 20 | |
| 21 BB_BUILD_DIR = os.path.abspath( | |
| 22 os.path.join(os.path.dirname(__file__), os.pardir, os.pardir, os.pardir, | |
| 23 os.pardir, os.pardir, os.pardir, os.pardir)) | |
| 24 | |
| 25 CHROME_SRC = os.path.abspath( | |
| 26 os.path.join(os.path.dirname(__file__), '..', '..', '..')) | |
| 27 | |
| 28 # TODO: Figure out how to merge this with devil.utils.cmd_helper.OutDirectory(). | |
| 29 CHROME_OUT_DIR = os.path.join(CHROME_SRC, 'out') | |
| 30 | |
| 31 GOMA_DIR = os.environ.get('GOMA_DIR', os.path.join(BB_BUILD_DIR, 'goma')) | |
| 32 | |
| 33 GSUTIL_PATH = os.path.join(BB_BUILD_DIR, 'third_party', 'gsutil', 'gsutil') | |
| 34 | |
| 35 def CommandToString(command): | |
| 36 """Returns quoted command that can be run in bash shell.""" | |
| 37 return ' '.join(cmd_helper.SingleQuote(c) for c in command) | |
| 38 | |
| 39 | |
| 40 def SpawnCmd(command, stdout=None, cwd=CHROME_SRC): | |
| 41 """Spawn a process without waiting for termination.""" | |
| 42 print '>', CommandToString(command) | |
| 43 sys.stdout.flush() | |
| 44 if TESTING: | |
| 45 class MockPopen(object): | |
| 46 @staticmethod | |
| 47 def wait(): | |
| 48 return 0 | |
| 49 @staticmethod | |
| 50 def communicate(): | |
| 51 return '', '' | |
| 52 return MockPopen() | |
| 53 return subprocess.Popen(command, cwd=cwd, stdout=stdout) | |
| 54 | |
| 55 | |
| 56 def RunCmd(command, flunk_on_failure=True, halt_on_failure=False, | |
| 57 warning_code=constants.WARNING_EXIT_CODE, stdout=None, | |
| 58 cwd=CHROME_SRC): | |
| 59 """Run a command relative to the chrome source root.""" | |
| 60 code = SpawnCmd(command, stdout, cwd).wait() | |
| 61 print '<', CommandToString(command) | |
| 62 if code != 0: | |
| 63 print 'ERROR: process exited with code %d' % code | |
| 64 if code != warning_code and flunk_on_failure: | |
| 65 bb_annotations.PrintError() | |
| 66 else: | |
| 67 bb_annotations.PrintWarning() | |
| 68 # Allow steps to have both halting (i.e. 1) and non-halting exit codes. | |
| 69 if code != warning_code and halt_on_failure: | |
| 70 print 'FATAL %d != %d' % (code, warning_code) | |
| 71 sys.exit(1) | |
| 72 return code | |
| 73 | |
| 74 | |
| 75 def GetParser(): | |
| 76 def ConvertJson(option, _, value, parser): | |
| 77 setattr(parser.values, option.dest, json.loads(value)) | |
| 78 parser = optparse.OptionParser() | |
| 79 parser.add_option('--build-properties', action='callback', | |
| 80 callback=ConvertJson, type='string', default={}, | |
| 81 help='build properties in JSON format') | |
| 82 parser.add_option('--factory-properties', action='callback', | |
| 83 callback=ConvertJson, type='string', default={}, | |
| 84 help='factory properties in JSON format') | |
| 85 return parser | |
| 86 | |
| 87 | |
| 88 def EncodeProperties(options): | |
| 89 return ['--factory-properties=%s' % json.dumps(options.factory_properties), | |
| 90 '--build-properties=%s' % json.dumps(options.build_properties)] | |
| 91 | |
| 92 | |
| 93 def RunSteps(steps, step_cmds, options): | |
| 94 unknown_steps = set(steps) - set(step for step, _ in step_cmds) | |
| 95 if unknown_steps: | |
| 96 print >> sys.stderr, 'FATAL: Unknown steps %s' % list(unknown_steps) | |
| 97 sys.exit(1) | |
| 98 | |
| 99 exit_code = 0 | |
| 100 for step, cmd in step_cmds: | |
| 101 if step in steps: | |
| 102 exit_code = cmd(options) or exit_code | |
| 103 | |
| 104 return exit_code | |
| OLD | NEW |