Skip indexing py_binary rules

Update

update

Author: yushan8

gazelle/python/resolve.go

@@ -55,6 +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 !indexPyBinaryImport(r, f) {
+		return nil
+	}
 	cfgs := c.Exts[languageName].(pythonconfig.Configs)
 	cfg := cfgs[f.Pkg]
 	srcs := r.AttrStrings("srcs")
@@ -366,3 +369,29 @@ func convertDependencySetToExpr(set *treeset.Set) bzl.Expr {
 	}
 	return &bzl.ListExpr{List: deps}
 }
+
+// indexPyBinaryImport returns whether the corresponding py_binary rule need to be indexed.
+// To avoid multiple labels indexing the same import,
+// check if there is a corresponding py_library rule with the same srcs.
+func indexPyBinaryImport(r *rule.Rule, f *rule.File) bool {
+	// If the rule is not a py_binary, it should be indexed.
+	if r.Kind() != "py_binary" {
+		return true
+	}
+	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/import_with_same_srcs_library_and_binary/BUILD.in

@@ -0,0 +1,3 @@
+# gazelle:python_extension enabled
+# gazelle:python_library_naming_convention py_default_library
+# gazelle:python_generation_mode package

gazelle/python/testdata/import_with_same_srcs_library_and_binary/BUILD.out

@@ -0,0 +1,3 @@
+# gazelle:python_extension enabled
+# gazelle:python_library_naming_convention py_default_library
+# gazelle:python_generation_mode package

gazelle/python/testdata/import_with_same_srcs_library_and_binary/README.md

@@ -0,0 +1,2 @@
+# Import with same srcs for library and binary
+This test case asserts a file with py_library and py_binary rules that include the same .py file in srcs will resolve to the py_library rule correctly.

gazelle/python/testdata/import_with_same_srcs_library_and_binary/WORKSPACE

@@ -0,0 +1 @@
+# This is a Bazel workspace for the Gazelle test data.

gazelle/python/testdata/import_with_same_srcs_library_and_binary/bar/BUILD.in

@@ -0,0 +1,8 @@
+load("@rules_python//python:defs.bzl", "py_library")
+
+py_library(
+    name = "bar",
+    srcs = ["bar.py"],
+    visibility = ["//:__subpackages__"],
+    deps = ["//foo"],
+)

gazelle/python/testdata/import_with_same_srcs_library_and_binary/bar/BUILD.out

@@ -0,0 +1,8 @@
+load("@rules_python//python:defs.bzl", "py_library")
+
+py_library(
+    name = "bar",
+    srcs = ["bar.py"],
+    visibility = ["//:__subpackages__"],
+    deps = ["//foo:py_default_library"],
+)

gazelle/python/testdata/import_with_same_srcs_library_and_binary/bar/bar.py

@@ -0,0 +1 @@
+import foo.script

gazelle/python/testdata/import_with_same_srcs_library_and_binary/foo/BUILD.in

@@ -0,0 +1,13 @@
+load("@rules_python//python:defs.bzl", "py_binary", "py_library")
+
+py_binary(
+    name = "script",
+    srcs = ["script.py"],
+    visibility = ["//:__subpackages__"],
+)
+
+py_library(
+    name = "py_default_library",
+    srcs = ["script.py"],
+    visibility = ["//:__subpackages__"],
+)