one-shot attempt

Author: chrisirhc

python/private/pypi/generate_whl_library_build_bazel.bzl

@@ -109,6 +109,8 @@ def generate_whl_library_build_bazel(
         kwargs["copy_executables"] = annotation.copy_executables
         kwargs["data_exclude"] = kwargs.get("data_exclude", []) + annotation.data_exclude_glob
         kwargs["srcs_exclude"] = annotation.srcs_exclude_glob
+        if annotation.enable_implicit_namespace_pkgs != None:
+            kwargs["enable_implicit_namespace_pkgs"] = annotation.enable_implicit_namespace_pkgs
         if annotation.additive_build_content:
             additional_content.append(annotation.additive_build_content)
     if default_python_version:

python/private/pypi/package_annotation.bzl

@@ -20,7 +20,8 @@ def package_annotation(
         copy_executables = {},
         data = [],
         data_exclude_glob = [],
-        srcs_exclude_glob = []):
+        srcs_exclude_glob = [],
+        enable_implicit_namespace_pkgs = None):
     """Annotations to apply to the BUILD file content from package generated from a `pip_repository` rule.
 
     [cf]: https://github.com/bazelbuild/bazel-skylib/blob/main/docs/copy_file_doc.md
@@ -35,6 +36,8 @@ def package_annotation(
         data_exclude_glob (list, optional): A list of exclude glob patterns to add as `data` to the generated
             `py_library` target.
         srcs_exclude_glob (list, optional): A list of labels to add as `srcs` to the generated `py_library` target.
+        enable_implicit_namespace_pkgs (bool, optional): Override the global setting for generating __init__.py
+            files for namespace packages. If None, uses the repository-level setting.
 
     Returns:
         str: A json encoded string of the provided content.
@@ -46,4 +49,5 @@ def package_annotation(
         data = data,
         data_exclude_glob = data_exclude_glob,
         srcs_exclude_glob = srcs_exclude_glob,
+        enable_implicit_namespace_pkgs = enable_implicit_namespace_pkgs,
     ))

tests/pypi/generate_whl_library_build_bazel/generate_whl_library_build_bazel_tests.bzl

@@ -211,6 +211,80 @@ whl_library_targets_from_requires(
 
 _tests.append(_test_all_with_loads)
 
+def _test_enable_implicit_namespace_pkgs_annotation(env):
+    want = """\
+load("@rules_python//python/private/pypi:whl_library_targets.bzl", "whl_library_targets")
+
+whl_library_targets(
+    dep_template = "@pypi//{name}:{target}",
+    enable_implicit_namespace_pkgs = True,
+    name = "foo.whl",
+    dependencies = ["foo"],
+    dependencies_by_platform = {"baz": ["bar"]},
+    entry_points = {
+        "foo": "bar.py",
+    },
+    group_deps = ["foo", "fox", "qux"],
+    group_name = "qux",
+    tags = ["tag1"],
+)
+"""
+    actual = generate_whl_library_build_bazel(
+        dep_template = "@pypi//{name}:{target}",
+        name = "foo.whl",
+        dependencies = ["foo"],
+        dependencies_by_platform = {"baz": ["bar"]},
+        entry_points = {
+            "foo": "bar.py",
+        },
+        annotation = struct(
+            enable_implicit_namespace_pkgs = True,
+        ),
+        group_name = "qux",
+        group_deps = ["foo", "fox", "qux"],
+        tags = ["tag1"],
+    )
+    env.expect.that_str(actual.replace("@@", "@")).equals(want)
+
+_tests.append(_test_enable_implicit_namespace_pkgs_annotation)
+
+def _test_enable_implicit_namespace_pkgs_annotation_false(env):
+    want = """\
+load("@rules_python//python/private/pypi:whl_library_targets.bzl", "whl_library_targets")
+
+whl_library_targets(
+    dep_template = "@pypi//{name}:{target}",
+    enable_implicit_namespace_pkgs = False,
+    name = "foo.whl",
+    dependencies = ["foo"],
+    dependencies_by_platform = {"baz": ["bar"]},
+    entry_points = {
+        "foo": "bar.py",
+    },
+    group_deps = ["foo", "fox", "qux"],
+    group_name = "qux",
+    tags = ["tag1"],
+)
+"""
+    actual = generate_whl_library_build_bazel(
+        dep_template = "@pypi//{name}:{target}",
+        name = "foo.whl",
+        dependencies = ["foo"],
+        dependencies_by_platform = {"baz": ["bar"]},
+        entry_points = {
+            "foo": "bar.py",
+        },
+        annotation = struct(
+            enable_implicit_namespace_pkgs = False,
+        ),
+        group_name = "qux",
+        group_deps = ["foo", "fox", "qux"],
+        tags = ["tag1"],
+    )
+    env.expect.that_str(actual.replace("@@", "@")).equals(want)
+
+_tests.append(_test_enable_implicit_namespace_pkgs_annotation_false)
+
 def generate_whl_library_build_bazel_test_suite(name):
     """Create the test suite.