Replace dependency on pre-built wheels with py_import dependency.

PiperOrigin-RevId: 693088159

opensource_only.files

@@ -109,6 +109,7 @@ third_party/py/python_init_repositories.bzl:
 third_party/py/python_init_rules.bzl:
 third_party/py/python_init_toolchains.bzl:
 third_party/py/python_repo.bzl:
+third_party/py/python_wheel_version.bzl:
 third_party/py/verify_manylinux_compliance.py:
 third_party/pybind11.BUILD:
 third_party/pybind11_bazel/BUILD:

third_party/py/python_wheel_version.bzl

@@ -0,0 +1,24 @@
+""" Repository rule to generate a file with python wheel version. """
+
+def _python_wheel_version_repository_impl(repository_ctx):
+    file_content = repository_ctx.read(
+        repository_ctx.path(repository_ctx.attr.file_with_version),
+    )
+    version_line_start_index = file_content.find(repository_ctx.attr.version_key)
+    version_line_end_index = version_line_start_index + file_content[version_line_start_index:].find("\n")
+    repository_ctx.file(
+        "wheel_version.bzl",
+        file_content[version_line_start_index:version_line_end_index].replace(
+            repository_ctx.attr.version_key,
+            "WHEEL_VERSION",
+        ),
+    )
+    repository_ctx.file("BUILD", "")
+
+python_wheel_version_repository = repository_rule(
+    implementation = _python_wheel_version_repository_impl,
+    attrs = {
+        "file_with_version": attr.label(mandatory = True, allow_single_file = True),
+        "version_key": attr.string(mandatory = True),
+    },
+)