OLD | NEW |
1 # Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 import fnmatch | 5 import fnmatch |
6 import json | 6 import json |
7 import os | 7 import os |
8 import pipes | 8 import pipes |
9 import shlex | 9 import shlex |
10 import shutil | 10 import shutil |
(...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
71 | 71 |
72 if not only_if_changed or old_dump != new_dump: | 72 if not only_if_changed or old_dump != new_dump: |
73 with open(path, 'w') as outfile: | 73 with open(path, 'w') as outfile: |
74 outfile.write(new_dump) | 74 outfile.write(new_dump) |
75 | 75 |
76 def ReadJson(path): | 76 def ReadJson(path): |
77 with open(path, 'r') as jsonfile: | 77 with open(path, 'r') as jsonfile: |
78 return json.load(jsonfile) | 78 return json.load(jsonfile) |
79 | 79 |
80 | 80 |
81 # This can be used in most cases like subprocess.check_call. The output, | 81 class CalledProcessError(Exception): |
| 82 """This exception is raised when the process run by CheckOutput |
| 83 exits with a non-zero exit code.""" |
| 84 |
| 85 def __init__(self, cwd, args, output): |
| 86 self.cwd = cwd |
| 87 self.args = args |
| 88 self.output = output |
| 89 |
| 90 def __str__(self): |
| 91 # A user should be able to simply copy and paste the command that failed |
| 92 # into their shell. |
| 93 copyable_command = '( cd {}; {} )'.format(os.path.abspath(self.cwd), |
| 94 ' '.join(map(pipes.quote, self.args))) |
| 95 return 'Command failed: {}\n{}'.format(copyable_command, self.output) |
| 96 |
| 97 |
| 98 # This can be used in most cases like subprocess.check_output(). The output, |
82 # particularly when the command fails, better highlights the command's failure. | 99 # particularly when the command fails, better highlights the command's failure. |
83 # This call will directly exit on a failure in the subprocess so that no python | 100 # If the command fails, raises a build_utils.CalledProcessError. |
84 # stacktrace is printed after the output of the failed command (and will | 101 def CheckOutput(args, cwd=None, print_stdout=False, print_stderr=True, |
85 # instead print a python stack trace before the output of the failed command) | 102 fail_if_stderr=False): |
86 def CheckCallDie(args, suppress_output=False, cwd=None, fail_if_stderr=False): | |
87 if not cwd: | 103 if not cwd: |
88 cwd = os.getcwd() | 104 cwd = os.getcwd() |
89 | 105 |
90 child = subprocess.Popen(args, | 106 child = subprocess.Popen(args, |
91 stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd=cwd) | 107 stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd=cwd) |
92 | |
93 stdout, stderr = child.communicate() | 108 stdout, stderr = child.communicate() |
94 | 109 |
95 returncode = child.returncode | 110 if child.returncode or (stderr and fail_if_stderr): |
96 if fail_if_stderr and stderr and returncode == 0: | 111 raise CalledProcessError(cwd, args, stdout + stderr) |
97 returncode = 1 | |
98 | 112 |
99 if returncode: | 113 if print_stdout: |
100 stacktrace = traceback.extract_stack() | 114 sys.stdout.write(stdout) |
101 print >> sys.stderr, ''.join(traceback.format_list(stacktrace)) | 115 if print_stderr: |
102 # A user should be able to simply copy and paste the command that failed | 116 sys.stderr.write(stderr) |
103 # into their shell. | |
104 copyable_command = ' '.join(map(pipes.quote, args)) | |
105 copyable_command = ('( cd ' + os.path.abspath(cwd) + '; ' | |
106 + copyable_command + ' )') | |
107 print >> sys.stderr, 'Command failed:', copyable_command, '\n' | |
108 | 117 |
109 if stdout: | 118 return stdout |
110 print stdout, | |
111 if stderr: | |
112 print >> sys.stderr, stderr, | |
113 | |
114 # Directly exit to avoid printing stacktrace. | |
115 sys.exit(returncode) | |
116 | |
117 else: | |
118 if not suppress_output: | |
119 if stdout: | |
120 print stdout, | |
121 if stderr: | |
122 print >> sys.stderr, stderr, | |
123 return stdout + stderr | |
124 | 119 |
125 | 120 |
126 def GetModifiedTime(path): | 121 def GetModifiedTime(path): |
127 # For a symlink, the modified time should be the greater of the link's | 122 # For a symlink, the modified time should be the greater of the link's |
128 # modified time and the modified time of the target. | 123 # modified time and the modified time of the target. |
129 return max(os.lstat(path).st_mtime, os.stat(path).st_mtime) | 124 return max(os.lstat(path).st_mtime, os.stat(path).st_mtime) |
130 | 125 |
131 | 126 |
132 def IsTimeStale(output, inputs): | 127 def IsTimeStale(output, inputs): |
133 if not os.path.exists(output): | 128 if not os.path.exists(output): |
134 return True | 129 return True |
135 | 130 |
136 output_time = GetModifiedTime(output) | 131 output_time = GetModifiedTime(output) |
137 for input in inputs: | 132 for input in inputs: |
138 if GetModifiedTime(input) > output_time: | 133 if GetModifiedTime(input) > output_time: |
139 return True | 134 return True |
140 return False | 135 return False |
141 | 136 |
142 | 137 |
143 def IsDeviceReady(): | 138 def IsDeviceReady(): |
144 device_state = CheckCallDie(['adb', 'get-state'], suppress_output=True) | 139 device_state = CheckOutput(['adb', 'get-state']) |
145 return device_state.strip() == 'device' | 140 return device_state.strip() == 'device' |
146 | 141 |
147 | 142 |
148 def PrintWarning(message): | 143 def PrintWarning(message): |
149 print 'WARNING: ' + message | 144 print 'WARNING: ' + message |
150 | 145 |
151 | 146 |
152 def PrintBigWarning(message): | 147 def PrintBigWarning(message): |
153 print '***** ' * 8 | 148 print '***** ' * 8 |
154 PrintWarning(message) | 149 PrintWarning(message) |
155 print '***** ' * 8 | 150 print '***** ' * 8 |
OLD | NEW |