feat: keep main module in py_library (#1608)

Unlike Go, the main module of Python can still be imported by other modules. We should keep them in the `py_library` target to allow such imports.
bazelbuild · Dec 14, 2023 · e7d5c8d · e7d5c8d
1 parent 2b5447b
commit e7d5c8d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/gazelle/python/generate.go b/gazelle/python/generate.go
@@ -248,7 +248,6 @@ func (py *Python) GenerateRules(args language.GenerateArgs) language.GenerateRes
 						fqTarget.String(), actualPyBinaryKind, err)
 					continue
 				}
-				srcs.Remove(filename)
 				pyBinary := newTargetBuilder(pyBinaryKind, pyBinaryTargetName, pythonProjectRoot, args.Rel, pyFileNames).
 					addVisibility(visibility).
 					addSrc(filename).

diff --git a/gazelle/python/testdata/binary_without_entrypoint/BUILD.out b/gazelle/python/testdata/binary_without_entrypoint/BUILD.out
@@ -24,6 +24,8 @@ py_library(
     srcs = [
         "__init__.py",
         "collided_main.py",
+        "main.py",
+        "main2.py",
     ],
     visibility = ["//:__subpackages__"],
 )