summaryrefslogtreecommitdiffstats
path: root/fetch.py
diff options
context:
space:
mode:
authorGavin Mak <gavinmak@google.com>2023-03-11 06:46:20 +0000
committerLUCI <gerrit-scoped@luci-project-accounts.iam.gserviceaccount.com>2023-03-22 17:46:28 +0000
commitea2e330e43c182dc16b0111ebc69ee5a71ee4ce1 (patch)
treedc33ba0e56825b3e007d0589891756724725a465 /fetch.py
parent1604cf255f8c1786a23388db6d5277ac7949a24a (diff)
downloadgit-repo-ea2e330e43c182dc16b0111ebc69ee5a71ee4ce1.tar.gz
Format codebase with black and check formatting in CQ
Apply rules set by https://gerrit-review.googlesource.com/c/git-repo/+/362954/ across the codebase and fix any lingering errors caught by flake8. Also check black formatting in run_tests (and CQ). Bug: b/267675342 Change-Id: I972d77649dac351150dcfeb1cd1ad0ea2efc1956 Reviewed-on: https://gerrit-review.googlesource.com/c/git-repo/+/363474 Reviewed-by: Mike Frysinger <vapier@google.com> Tested-by: Gavin Mak <gavinmak@google.com> Commit-Queue: Gavin Mak <gavinmak@google.com>
Diffstat (limited to 'fetch.py')
-rw-r--r--fetch.py46
1 files changed, 25 insertions, 21 deletions
diff --git a/fetch.py b/fetch.py
index c954a9c2..31f8152f 100644
--- a/fetch.py
+++ b/fetch.py
@@ -21,25 +21,29 @@ from urllib.request import urlopen
21 21
22 22
23def fetch_file(url, verbose=False): 23def fetch_file(url, verbose=False):
24 """Fetch a file from the specified source using the appropriate protocol. 24 """Fetch a file from the specified source using the appropriate protocol.
25 25
26 Returns: 26 Returns:
27 The contents of the file as bytes. 27 The contents of the file as bytes.
28 """ 28 """
29 scheme = urlparse(url).scheme 29 scheme = urlparse(url).scheme
30 if scheme == 'gs': 30 if scheme == "gs":
31 cmd = ['gsutil', 'cat', url] 31 cmd = ["gsutil", "cat", url]
32 try: 32 try:
33 result = subprocess.run( 33 result = subprocess.run(
34 cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, 34 cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=True
35 check=True) 35 )
36 if result.stderr and verbose: 36 if result.stderr and verbose:
37 print('warning: non-fatal error running "gsutil": %s' % result.stderr, 37 print(
38 file=sys.stderr) 38 'warning: non-fatal error running "gsutil": %s'
39 return result.stdout 39 % result.stderr,
40 except subprocess.CalledProcessError as e: 40 file=sys.stderr,
41 print('fatal: error running "gsutil": %s' % e.stderr, 41 )
42 file=sys.stderr) 42 return result.stdout
43 sys.exit(1) 43 except subprocess.CalledProcessError as e:
44 with urlopen(url) as f: 44 print(
45 return f.read() 45 'fatal: error running "gsutil": %s' % e.stderr, file=sys.stderr
46 )
47 sys.exit(1)
48 with urlopen(url) as f:
49 return f.read()