Update

Author: yushan8

gazelle/python/resolve.go

@@ -55,7 +55,9 @@ func (*Resolver) Name() string { return languageName }
 // If nil is returned, the rule will not be indexed. If any non-nil slice is
 // returned, including an empty slice, the rule will be indexed.
 func (py *Resolver) Imports(c *config.Config, r *rule.Rule, f *rule.File) []resolve.ImportSpec {
-	if r.Kind() == "py_binary" {
+	// To avoid multiple labels indexing the same import,
+	// check if there is a corresponding py_library rule with the same srcs.
+	if r.Kind() == "py_binary" && !indexPyBinaryImport(r, f) {
 		return nil
 	}
 	cfgs := c.Exts[languageName].(pythonconfig.Configs)
@@ -320,3 +322,22 @@ func convertDependencySetToExpr(set *treeset.Set) bzl.Expr {
 	}
 	return &bzl.ListExpr{List: deps}
 }
+
+func indexPyBinaryImport(r *rule.Rule, f *rule.File) bool {
+	pyBinarySrcs := r.AttrStrings("srcs")
+	if len(pyBinarySrcs) == 0 {
+		return false
+	}
+	for _, otherRule := range f.Rules {
+		if otherRule.Kind() != "py_library" {
+			continue
+		}
+		pyLibrarySrcs := otherRule.AttrStrings("srcs")
+		for _, src := range pyLibrarySrcs {
+			if src == pyBinarySrcs[0] {
+				return false
+			}
+		}
+	}
+	return true
+}

gazelle/python/testdata/binary_without_entrypoint_per_file_generation/BUILD.out

@@ -19,7 +19,10 @@ py_library(
     name = "lib2",
     srcs = ["lib2.py"],
     visibility = ["//:__subpackages__"],
-    deps = [":lib"],
+    deps = [
+        ":lib",
+        ":lib_and_main",
+    ],
 )
 
 py_binary(

gazelle/python/testdata/binary_without_entrypoint_per_file_generation/lib2.py

@@ -1 +1,2 @@
 import lib
+import lib_and_main