[+] improve typing

This commit is contained in:
Siarhei Siniak 2025-05-19 17:06:01 +03:00
parent d30cdc7d70
commit a7f6bd927c

@ -25,6 +25,7 @@ if typing.TYPE_CHECKING:
import pip._internal.resolution.resolvelib.resolver import pip._internal.resolution.resolvelib.resolver
import pip._internal.operations.prepare import pip._internal.operations.prepare
import pip._internal.index.package_finder import pip._internal.index.package_finder
from pip._internal.req.req_install import InstallRequirement
from typing import ( from typing import (
Literal, Optional, Iterable, Any, Literal, Optional, Iterable, Any,
@ -65,6 +66,14 @@ class pip_resolve_t:
ignore_requires_python: Any, ignore_requires_python: Any,
) -> 'pip._internal.index.package_finder.PackageFinder': ... ) -> 'pip._internal.index.package_finder.PackageFinder': ...
class complete_partial_requirements_t(Protocol):
def __call__(
self,
_self: 'pip._internal.resolution.resolvelib.resolver.Resolver',
partially_downloaded_reqs: Iterable['InstallRequirement',],
parallel_builds: bool = False,
) -> None: ...
class kwargs_t: class kwargs_t:
class mode_t(enum.StrEnum): class mode_t(enum.StrEnum):
copy_paste = "copy_paste" copy_paste = "copy_paste"
@ -381,7 +390,13 @@ def pip_resolve(
) )
raise NotImplementedError raise NotImplementedError
_complete_partial_requirements_def = pip._internal.operations.prepare.RequirementPreparer._complete_partial_requirements _complete_partial_requirements_def = cast(
pip_resolve_t.complete_partial_requirements_t,
getattr(
pip._internal.operations.prepare.RequirementPreparer,
'_complete_partial_requirements',
)
)
def _complete_partial_requirements( def _complete_partial_requirements(
_self: pip._internal.resolution.resolvelib.resolver.Resolver, _self: pip._internal.resolution.resolvelib.resolver.Resolver,