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