anki/python.bzl
2021-10-04 15:05:15 +10:00

53 lines
1.2 KiB
Python

def _impl(rctx):
# locate python on path, and export it
names = [
"python3.9",
"python3.10",
"python.exe",
]
path = None
if rctx.os.environ.get("PYO3_PYTHON"):
path = rctx.os.environ.get("PYO3_PYTHON")
else:
for name in names:
path = rctx.which(name)
if path:
break
if not path:
fail("python3 or python.exe not found on path, and PYO3_PYTHON not set")
rctx.symlink(path, "python")
rctx.file("BUILD.bazel", """
load("@rules_python//python:defs.bzl", "py_runtime_pair")
py_runtime(
name = "python_runtime",
interpreter_path = "{path}",
python_version = "PY3",
visibility = ["//visibility:public"],
)
py_runtime_pair(
name = "python3_runtime_pair",
py2_runtime = None,
py3_runtime = ":python_runtime",
)
toolchain(
name = "python3_toolchain",
toolchain = ":python3_runtime_pair",
toolchain_type = "@bazel_tools//tools/python:toolchain_type",
visibility = ["//visibility:public"],
)
exports_files(["python"])
""".format(path = path))
setup_local_python = repository_rule(
implementation = _impl,
local = True,
attrs = {},
)