From 02bce6cb7c2fbfb5eb391714146ec8e801e9d9f9 Mon Sep 17 00:00:00 2001
From: Zhongpeng Lin <zplin@uber.com>
Date: Wed, 13 Dec 2023 10:50:39 -0800
Subject: [PATCH] keep main module in py_library

---
 gazelle/python/generate.go                                  | 1 -
 gazelle/python/testdata/binary_without_entrypoint/BUILD.out | 2 ++
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/gazelle/python/generate.go b/gazelle/python/generate.go
index f812f17a63..8d147d92a0 100644
--- 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
index c88f2ff11e..9dae147761 100644
--- 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__"],
 )