feat(rules): add py_dist_info rule for importlib.metadata

docs/BUILD.bazel

@@ -112,6 +112,7 @@ sphinx_stardocs(
         "//python:proto",
         "//python:py_binary",
         "//python:py_cc_link_params_info",
+        "//python:py_dist_info",
         "//python:py_exec_tools_info",
         "//python:py_exec_tools_toolchain",
         "//python:py_executable_info",
@@ -139,6 +140,7 @@ sphinx_stardocs(
         "//python/private:builders_util",
         "//python/private:py_binary_rule",
         "//python/private:py_cc_toolchain_rule",
+        "//python/private:py_dist_info",
         "//python/private:py_info",
         "//python/private:py_library_rule",
         "//python/private:py_runtime_rule",

news/4357.added.md

@@ -0,0 +1,5 @@
+(rules) Added {obj}`py_dist_info` in `@rules_python//python:py_dist_info.bzl`
+to generate `.dist-info` metadata (`METADATA` and `entry_points.txt`) in
+runfiles so `importlib.metadata` can discover package versions and entry points
+at runtime, and added `dist_info` support to {obj}`py_wheel` so wheels can
+reuse a {obj}`py_dist_info` target.

python/BUILD.bazel

@@ -52,6 +52,7 @@ filegroup(
         "defs.bzl",
         "packaging.bzl",
         "pip.bzl",
+        "py_dist_info.bzl",
         "pytest_test.bzl",
         "repositories.bzl",
         "versions.bzl",
@@ -65,6 +66,7 @@ filegroup(
 
 exports_files([
     "defs.bzl",
+    "py_dist_info.bzl",
     "pytest_test.bzl",
     "python.bzl",  # Deprecated, please use defs.bzl
 ])
@@ -207,6 +209,7 @@ bzl_library(
     deps = [
         ":py_binary",
         "//python/private:bzlmod_enabled",
+        "//python/private:py_dist_info",
         "//python/private:py_package",
         "//python/private:py_wheel",
         "//python/private:util",
@@ -276,6 +279,13 @@ alias(
     deprecation = "Use //python:py_cc_link_params_info instead",
 )
 
+# keep
+bzl_library(
+    name = "py_dist_info",
+    srcs = ["py_dist_info.bzl"],
+    deps = ["//python/private:py_dist_info"],
+)
+
 bzl_library(
     name = "py_exec_tools_info",
     srcs = ["py_exec_tools_info.bzl"],

python/features.bzl

@@ -121,6 +121,7 @@ _TARGETS = {
     "//python:proto": True,
     "//python:py_binary": True,
     "//python:py_cc_link_params_info": True,
+    "//python:py_dist_info": True,
     "//python:py_exec_tools_info": True,
     "//python:py_exec_tools_toolchain": True,
     "//python:py_executable_info": True,
@@ -140,6 +141,11 @@ _LOADABLE_SYMBOLS = {
         # keep sorted
         "py_extension",
     ],
+    "//python:py_dist_info.bzl": [
+        # keep sorted
+        "PyDistInfo",
+        "py_dist_info",
+    ],
     "//python:py_info.bzl": [
         # keep sorted
         "PyInfo",

python/packaging.bzl

@@ -17,6 +17,11 @@
 load("@bazel_skylib//rules:native_binary.bzl", "native_binary")
 load("//python:py_binary.bzl", "py_binary")
 load("//python/private:bzlmod_enabled.bzl", "BZLMOD_ENABLED")
+load(
+    "//python/private:py_dist_info.bzl",
+    "DIST_INFO_ATTR_NAMES",
+    "py_dist_info",
+)
 load("//python/private:py_package.bzl", "py_package_lib")
 load("//python/private:py_wheel.bzl", _PyWheelInfo = "PyWheelInfo", _py_wheel = "py_wheel")
 load("//python/private:util.bzl", "copy_propagating_kwargs")
@@ -169,6 +174,12 @@ def py_wheel(
         //path/to:my_wheel.publish --repository testpypi
     ```
 
+    :::{versionchanged} VERSION_NEXT_FEATURE
+    Added support for passing a {obj}`py_dist_info` target via `dist_info`, and
+    automatically generates `<name>.dist_info` when `dist_info` is not
+    specified.
+    :::
+
     Args:
         name:  A unique name for this target.
         twine: A label of the external location of the py_library target for twine
@@ -190,6 +201,22 @@ def py_wheel(
         **copy_propagating_kwargs(kwargs)
     )
 
+    if "dist_info" not in kwargs or kwargs["dist_info"] == None:
+        dist_info_target = "{}.dist_info".format(name)
+        dist_info_kwargs = {
+            attr_name: kwargs[attr_name]
+            for attr_name in DIST_INFO_ATTR_NAMES
+            if attr_name in kwargs
+        }
+        if "toolchains" in kwargs:
+            dist_info_kwargs["toolchains"] = kwargs["toolchains"]
+        py_dist_info(
+            name = dist_info_target,
+            tags = manual_tags,
+            **copy_propagating_kwargs(kwargs, dist_info_kwargs)
+        )
+        kwargs["dist_info"] = dist_info_target
+
     _py_wheel(
         name = name,
         tags = tags,

python/private/BUILD.bazel

@@ -729,10 +729,10 @@ bzl_library(
     srcs = ["py_wheel.bzl"],
     deps = [
         ":attributes",
+        ":py_dist_info",
         ":py_info",
         ":py_package",
         ":rule_builders",
-        ":stamp_impl",
         ":toolchain_types",
         ":transition_labels",
         ":version",
@@ -995,6 +995,18 @@ bzl_library(
     srcs = ["py_console_script_gen.bzl"],
 )
 
+bzl_library(
+    name = "py_dist_info",
+    srcs = ["py_dist_info.bzl"],
+    deps = [
+        ":attributes",
+        ":rule_builders",
+        ":stamp_impl",
+        ":toolchain_types",
+        ":transition_labels",
+    ],
+)
+
 bzl_library(
     name = "py_exec_tools_info",
     srcs = ["py_exec_tools_info.bzl"],