diff --git a/gazelle/python/resolve.go b/gazelle/python/resolve.go index 88275e007c..0dd80841d4 100644 --- a/gazelle/python/resolve.go +++ b/gazelle/python/resolve.go @@ -124,12 +124,16 @@ func (py *Resolver) Embeds(r *rule.Rule, from label.Label) []label.Label { } // addDependency adds a dependency to either the regular deps or pyiDeps set based on -// whether the module is type-checking only. -func addDependency(dep string, mod Module, deps, pyiDeps *treeset.Set) { - if mod.TypeCheckingOnly { - pyiDeps.Add(dep) +// whether the module is type-checking only. If a module is added as both +// non-type-checking and type-checking, it should end up in deps and not pyiDeps. +func addDependency(dep string, typeCheckingOnly bool, deps, pyiDeps *treeset.Set) { + if typeCheckingOnly { + if !deps.Contains(dep) { + pyiDeps.Add(dep) + } } else { deps.Add(dep) + pyiDeps.Remove(dep) } } @@ -240,7 +244,7 @@ func (py *Resolver) Resolve( override.Repo = "" } dep := override.Rel(from.Repo, from.Pkg).String() - addDependency(dep, mod, deps, pyiDeps) + addDependency(dep, mod.TypeCheckingOnly, deps, pyiDeps) if explainDependency == dep { log.Printf("Explaining dependency (%s): "+ "in the target %q, the file %q imports %q at line %d, "+ @@ -251,7 +255,7 @@ func (py *Resolver) Resolve( } } else { if dep, distributionName, ok := cfg.FindThirdPartyDependency(moduleName); ok { - addDependency(dep, mod, deps, pyiDeps) + addDependency(dep, mod.TypeCheckingOnly, deps, pyiDeps) // Add the type and stub dependencies if they exist. modules := []string{ fmt.Sprintf("%s_stubs", strings.ToLower(distributionName)), @@ -261,8 +265,8 @@ func (py *Resolver) Resolve( } for _, module := range modules { if dep, _, ok := cfg.FindThirdPartyDependency(module); ok { - // Type stub packages always go to pyiDeps - pyiDeps.Add(dep) + // Type stub packages are added as type-checking only. + addDependency(dep, true, deps, pyiDeps) } } if explainDependency == dep { @@ -321,7 +325,7 @@ func (py *Resolver) Resolve( } matchLabel := filteredMatches[0].Label.Rel(from.Repo, from.Pkg) dep := matchLabel.String() - addDependency(dep, mod, deps, pyiDeps) + addDependency(dep, mod.TypeCheckingOnly, deps, pyiDeps) if explainDependency == dep { log.Printf("Explaining dependency (%s): "+ "in the target %q, the file %q imports %q at line %d, "+ diff --git a/gazelle/python/testdata/type_checking_imports_across_packages/BUILD.in b/gazelle/python/testdata/type_checking_imports_across_packages/BUILD.in new file mode 100644 index 0000000000..8e6c1cbabb --- /dev/null +++ b/gazelle/python/testdata/type_checking_imports_across_packages/BUILD.in @@ -0,0 +1,2 @@ +# gazelle:python_generation_mode package +# gazelle:python_generate_pyi_deps true diff --git a/gazelle/python/testdata/type_checking_imports_across_packages/BUILD.out b/gazelle/python/testdata/type_checking_imports_across_packages/BUILD.out new file mode 100644 index 0000000000..8e6c1cbabb --- /dev/null +++ b/gazelle/python/testdata/type_checking_imports_across_packages/BUILD.out @@ -0,0 +1,2 @@ +# gazelle:python_generation_mode package +# gazelle:python_generate_pyi_deps true diff --git a/gazelle/python/testdata/type_checking_imports_across_packages/README.md b/gazelle/python/testdata/type_checking_imports_across_packages/README.md new file mode 100644 index 0000000000..75fb3aae56 --- /dev/null +++ b/gazelle/python/testdata/type_checking_imports_across_packages/README.md @@ -0,0 +1,6 @@ +# Overlapping deps and pyi_deps across packages + +This test reproduces a case where a dependency may be added to both `deps` and +`pyi_deps`. Package `b` imports `a.foo` normally and imports `a.bar` as a +type-checking only import. The dependency on package `a` should appear only in +`deps` (and not `pyi_deps`) of package `b`. diff --git a/gazelle/python/testdata/type_checking_imports_across_packages/WORKSPACE b/gazelle/python/testdata/type_checking_imports_across_packages/WORKSPACE new file mode 100644 index 0000000000..3e6e74e7f4 --- /dev/null +++ b/gazelle/python/testdata/type_checking_imports_across_packages/WORKSPACE @@ -0,0 +1 @@ +workspace(name = "gazelle_python_test") diff --git a/gazelle/python/testdata/type_checking_imports_across_packages/a/BUILD.in b/gazelle/python/testdata/type_checking_imports_across_packages/a/BUILD.in new file mode 100644 index 0000000000..e69de29bb2 diff --git a/gazelle/python/testdata/type_checking_imports_across_packages/a/BUILD.out b/gazelle/python/testdata/type_checking_imports_across_packages/a/BUILD.out new file mode 100644 index 0000000000..cf9be008b1 --- /dev/null +++ b/gazelle/python/testdata/type_checking_imports_across_packages/a/BUILD.out @@ -0,0 +1,10 @@ +load("@rules_python//python:defs.bzl", "py_library") + +py_library( + name = "a", + srcs = [ + "bar.py", + "foo.py", + ], + visibility = ["//:__subpackages__"], +) diff --git a/gazelle/python/testdata/type_checking_imports_across_packages/a/bar.py b/gazelle/python/testdata/type_checking_imports_across_packages/a/bar.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/gazelle/python/testdata/type_checking_imports_across_packages/a/foo.py b/gazelle/python/testdata/type_checking_imports_across_packages/a/foo.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/gazelle/python/testdata/type_checking_imports_across_packages/b/BUILD.in b/gazelle/python/testdata/type_checking_imports_across_packages/b/BUILD.in new file mode 100644 index 0000000000..e69de29bb2 diff --git a/gazelle/python/testdata/type_checking_imports_across_packages/b/BUILD.out b/gazelle/python/testdata/type_checking_imports_across_packages/b/BUILD.out new file mode 100644 index 0000000000..15f4d343e1 --- /dev/null +++ b/gazelle/python/testdata/type_checking_imports_across_packages/b/BUILD.out @@ -0,0 +1,8 @@ +load("@rules_python//python:defs.bzl", "py_library") + +py_library( + name = "b", + srcs = ["b.py"], + visibility = ["//:__subpackages__"], + deps = ["//a"], +) diff --git a/gazelle/python/testdata/type_checking_imports_across_packages/b/b.py b/gazelle/python/testdata/type_checking_imports_across_packages/b/b.py new file mode 100644 index 0000000000..93d09c0baa --- /dev/null +++ b/gazelle/python/testdata/type_checking_imports_across_packages/b/b.py @@ -0,0 +1,6 @@ +from typing import TYPE_CHECKING + +from a import foo + +if TYPE_CHECKING: + from a import bar diff --git a/gazelle/python/testdata/type_checking_imports_across_packages/test.yaml b/gazelle/python/testdata/type_checking_imports_across_packages/test.yaml new file mode 100644 index 0000000000..ed97d539c0 --- /dev/null +++ b/gazelle/python/testdata/type_checking_imports_across_packages/test.yaml @@ -0,0 +1 @@ +---