mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 22:00:12 +01:00
Chandler Carruth
b79ce84f33
This removes the need to install any specific version of Python or figure out how to configure it by instead asking users to install `uv` and letting it manage Python. Among other advantages, `uv` is designed to be fast enough to embed directly into our scripts. We were already using this in `bench_runner.py` so that the script could import non standard library dependencies. Moving to it for the rest of our Python unifies the approach and will also enable dependencies whenever needed. I've left `github_tools` alone as it has special handling with its own Bazel setup. I've updated the contributing tools to explain the approach here.
95 lines
2.8 KiB
Python
Executable File
95 lines
2.8 KiB
Python
Executable File
#!/usr/bin/env -S uv run --script
|
|
|
|
# /// script
|
|
# requires-python = ">=3.11"
|
|
# ///
|
|
|
|
|
|
"""Runs bazel on arguments.
|
|
|
|
This is provided for other scripts to run bazel without requiring it be
|
|
manually installed.
|
|
"""
|
|
|
|
__copyright__ = """
|
|
Part of the Carbon Language project, under the Apache License v2.0 with LLVM
|
|
Exceptions. See /LICENSE for license information.
|
|
SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
"""
|
|
|
|
import argparse
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
|
|
import scripts_utils
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description="Runs bazel.")
|
|
parser.add_argument(
|
|
"--attempts",
|
|
metavar="COUNT",
|
|
type=int,
|
|
default=1,
|
|
help="The number of attempts to execute the command, automatically "
|
|
"retrying errors that may be transient.",
|
|
)
|
|
parser.add_argument(
|
|
"--jobs-on-last-attempt",
|
|
metavar="COUNT",
|
|
type=int,
|
|
help="Sets the number of jobs in user.bazelrc on the last attempt. If "
|
|
"there is only one attempt, this will be set immediately.",
|
|
)
|
|
parser.add_argument(
|
|
"--retry-all-errors",
|
|
action="store_true",
|
|
help="Retries permanent errors in addition to transient.",
|
|
)
|
|
script_args, bazel_args = parser.parse_known_args()
|
|
|
|
bazel = scripts_utils.locate_bazel()
|
|
attempt = 0
|
|
while True:
|
|
attempt += 1
|
|
if attempt == script_args.attempts and script_args.jobs_on_last_attempt:
|
|
with open("user.bazelrc", "a") as bazelrc:
|
|
bazelrc.write(
|
|
f"build --jobs={script_args.jobs_on_last_attempt}\n"
|
|
)
|
|
|
|
p = subprocess.run([bazel] + bazel_args)
|
|
|
|
# If this was the last attempt, or it succeeded, we're done.
|
|
if attempt == script_args.attempts or p.returncode == 0:
|
|
exit(p.returncode)
|
|
|
|
# Several error codes are reliably permanent, break immediately.
|
|
# `1` -- The build failed.
|
|
# `2` -- Command line or environment problem.
|
|
# `3` -- Tests failed or timed out, we don't retry at this layer
|
|
# on execution timeout.
|
|
# `4` -- Test command but no tests found.
|
|
# `8` -- Explicitly interrupted build.
|
|
#
|
|
# Note that `36` is documented as "likely permanent", but we retry
|
|
# it as most of our transient failures actually produce that error
|
|
# code.
|
|
perm_error = (1, 2, 3, 4, 8)
|
|
if not script_args.retry_all_errors and p.returncode in perm_error:
|
|
exit(p.returncode)
|
|
|
|
print(
|
|
f"Retrying exit code {p.returncode} because it may be transient..."
|
|
)
|
|
# Also sleep a bit to try to skip over transient machine load.
|
|
time.sleep(attempt)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
main()
|
|
except KeyboardInterrupt:
|
|
sys.exit(1)
|