mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-10-05 22:02:55 +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.
113 lines
3.3 KiB
Python
Executable File
113 lines
3.3 KiB
Python
Executable File
#!/usr/bin/env -S uv run --script
|
|
|
|
# /// script
|
|
# requires-python = ">=3.11"
|
|
# ///
|
|
|
|
"""Autoupdates testdata in toolchain."""
|
|
|
|
__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 re
|
|
import shlex
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
def main() -> None:
|
|
bazel = str(Path(__file__).parents[1] / "scripts" / "run_bazel.py")
|
|
configs = []
|
|
# Use the most recently used build mode, or `fastbuild` if missing
|
|
# `bazel-bin`.
|
|
build_mode = "fastbuild"
|
|
workspace = subprocess.check_output(
|
|
[
|
|
bazel,
|
|
"info",
|
|
"workspace",
|
|
"--ui_event_filters=stdout",
|
|
],
|
|
encoding="utf-8",
|
|
).strip()
|
|
bazel_bin_path = Path(workspace).joinpath("bazel-bin")
|
|
if bazel_bin_path.exists():
|
|
link = str(bazel_bin_path.readlink())
|
|
m = re.search(r"-(\w+)/bin$", link)
|
|
if m:
|
|
build_mode = m[1]
|
|
else:
|
|
exit(f"Build mode not found in `bazel-bin` symlink: {link}")
|
|
|
|
# Parse arguments.
|
|
parser = argparse.ArgumentParser(__doc__)
|
|
parser.add_argument("--non-fatal-checks", action="store_true")
|
|
parser.add_argument(
|
|
"--print_slowest_tests", default=0, help="Forwarded to file_test"
|
|
)
|
|
parser.add_argument("--threads", help="Forwarded to file_test")
|
|
parser.add_argument("files", nargs="*")
|
|
args = parser.parse_args()
|
|
|
|
if args.non_fatal_checks:
|
|
if build_mode == "optimize":
|
|
exit(
|
|
"`--non-fatal-checks` is incompatible with inferred "
|
|
"`-c optimize` build mode"
|
|
)
|
|
configs.append("--config=non-fatal-checks")
|
|
|
|
argv = [
|
|
bazel,
|
|
"run",
|
|
"-c",
|
|
build_mode,
|
|
*configs,
|
|
"--experimental_convenience_symlinks=ignore",
|
|
"--ui_event_filters=-info,-stdout,-stderr,-finish",
|
|
"//toolchain/testing:file_test",
|
|
"--",
|
|
"--autoupdate",
|
|
"--print_slowest_tests",
|
|
str(args.print_slowest_tests),
|
|
]
|
|
if args.threads:
|
|
argv += ["--threads", args.threads]
|
|
# Support specifying tests to update, such as:
|
|
# ./autoupdate_testdata.py lex/**/*
|
|
if args.files:
|
|
repo_root = Path(__file__).parents[1]
|
|
file_tests = []
|
|
# Filter down to just test files.
|
|
for f in args.files:
|
|
if f.endswith(".carbon"):
|
|
path = str(Path(f).resolve().relative_to(repo_root))
|
|
if path.count("/testdata/"):
|
|
file_tests.append(path)
|
|
if not file_tests:
|
|
sys.exit(
|
|
"Args do not seem to be test files; for example, "
|
|
f"{args.files[0]}"
|
|
)
|
|
argv.append("--file_tests=" + ",".join(file_tests))
|
|
# Provide an empty stdin so that the driver tests that read from stdin
|
|
# don't block waiting for input. This matches the behavior of `bazel test`.
|
|
result = subprocess.run(argv)
|
|
if result.returncode != 0:
|
|
sys.exit(
|
|
f"Command `{shlex.join(argv)}` failed with exit code "
|
|
f"`{result.returncode}`"
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
main()
|
|
except KeyboardInterrupt:
|
|
sys.exit(1)
|