def _get_imports_info(
address: Address,
owners_per_import: Iterable[PythonModuleOwners],
parsed_imports: ParsedPythonImports,
explicitly_provided_deps: ExplicitlyProvidedDependencies,
) -> tuple[set[Address], set[str]]:
inferred_deps: set[Address] = set()
unowned_imports: set[str] = set()
for owners, imp in zip(owners_per_import, parsed_imports):
inferred_deps.update(owners.unambiguous)
explicitly_provided_deps.maybe_warn_of_ambiguous_dependency_inference(
owners.ambiguous,
address,
import_reference="module",
context=f"The target {address} imports `{imp}`",
)
maybe_disambiguated = explicitly_provided_deps.disambiguated(owners.ambiguous)
if maybe_disambiguated:
inferred_deps.add(maybe_disambiguated)
if (
not owners.unambiguous
and imp.split(".")[0] not in DEFAULT_UNOWNED_DEPENDENCIES
and not parsed_imports[imp].weak
):
unowned_imports.add(imp)
return inferred_deps, unowned_imports
def get_disambiguated(
ambiguous: List[Address],
*,
ignores: Optional[List[Address]] = None,
includes: Optional[List[Address]] = None,
owners_must_be_ancestors: bool = False,
) -> Optional[Address]:
epd = ExplicitlyProvidedDependencies(
address=Address("dir", target_name="input_tgt"),
includes=FrozenOrderedSet(includes or []),
ignores=FrozenOrderedSet(ignores or []),
)
return epd.disambiguated(
tuple(ambiguous),
owners_must_be_ancestors=owners_must_be_ancestors)
def _get_inferred_asset_deps(
address: Address,
request_file_path: str,
assets_by_path: AllAssetTargetsByPath,
assets: ParsedPythonAssetPaths,
explicitly_provided_deps: ExplicitlyProvidedDependencies,
) -> Iterator[Address]:
for filepath in assets:
# NB: Resources in Python's ecosystem are loaded relative to a package, so we only try and
# query for a resource relative to requesting module's path
# (I.e. we assume the user is doing something like `pkgutil.get_data(__file__, "foo/bar")`)
# See https://docs.python.org/3/library/pkgutil.html#pkgutil.get_data
# and Pants' own docs on resources.
#
# Files in Pants are always loaded relative to the build root without any source root
# stripping, so we use the full filepath to query for files.
# (I.e. we assume the user is doing something like `open("src/python/configs/prod.json")`)
#
# In either case we could also try and query based on the others' key, however this will
# almost always lead to a false positive.
resource_path = PurePath(request_file_path).parent / filepath
file_path = PurePath(filepath)
inferred_resource_tgts = assets_by_path.resources.get(
resource_path, frozenset())
inferred_file_tgts = assets_by_path.files.get(file_path, frozenset())
inferred_tgts = inferred_resource_tgts | inferred_file_tgts
if inferred_tgts:
possible_addresses = tuple(tgt.address for tgt in inferred_tgts)
explicitly_provided_deps.maybe_warn_of_ambiguous_dependency_inference(
possible_addresses,
address,
import_reference="asset",
context=f"The target {address} uses `{filepath}`",
)
maybe_disambiguated = explicitly_provided_deps.disambiguated(
possible_addresses)
if maybe_disambiguated:
yield maybe_disambiguated