diff options
author | Tzu-ping Chung <uranusjr@gmail.com> | 2020-12-03 16:39:11 +0800 |
---|---|---|
committer | Tzu-ping Chung <uranusjr@gmail.com> | 2020-12-03 17:44:23 +0800 |
commit | 4ad924a66f87002195e6815a3697c3316afc02d1 (patch) | |
tree | c0be3969e09c91172f9f8778f5f8b21a22b86d7e | |
parent | 30eeb9ceb10bf76316f091e8615a9c3d8025fff0 (diff) | |
download | pip-4ad924a66f87002195e6815a3697c3316afc02d1.tar.gz |
Resolve direct and pinned requirements first
-rw-r--r-- | news/9185.feature.rst | 2 | ||||
-rw-r--r-- | src/pip/_internal/resolution/resolvelib/provider.py | 46 |
2 files changed, 47 insertions, 1 deletions
diff --git a/news/9185.feature.rst b/news/9185.feature.rst new file mode 100644 index 000000000..a9d9ae718 --- /dev/null +++ b/news/9185.feature.rst @@ -0,0 +1,2 @@ +New resolver: Resolve direct and pinned (``==`` or ``===``) requirements first +to improve resolver performance. diff --git a/src/pip/_internal/resolution/resolvelib/provider.py b/src/pip/_internal/resolution/resolvelib/provider.py index c0e6b60d9..b1df3ca52 100644 --- a/src/pip/_internal/resolution/resolvelib/provider.py +++ b/src/pip/_internal/resolution/resolvelib/provider.py @@ -56,9 +56,53 @@ class PipProvider(AbstractProvider): information # type: Sequence[Tuple[Requirement, Candidate]] ): # type: (...) -> Any + """Produce a sort key for given requirement based on preference. + + The lower the return value is, the more preferred this group of + arguments is. + + Currently pip considers the followings in order: + + * Prefer if any of the known requirements points to an explicit URL. + * If equal, prefer if any requirements contain `===` and `==`. + * If equal, prefer user-specified (non-transitive) requirements. + * If equal, order alphabetically for consistency (helps debuggability). + """ + + def _get_restrictive_rating(requirements): + # type: (Iterable[Requirement]) -> int + """Rate how restrictive a set of requirements are. + + ``Requirement.get_candidate_lookup()`` returns a 2-tuple for + lookup. The first element is ``Optional[Candidate]`` and the + second ``Optional[InstallRequirement]``. + + * If the requirement is an explicit one, the explicitly-required + candidate is returned as the first element. + * If the requirement is based on a PEP 508 specifier, the backing + ``InstallRequirement`` is returned as the second element. + + We use the first element to check whether there is an explicit + requirement, and the second for equality operator. + """ + lookups = (r.get_candidate_lookup() for r in requirements) + cands, ireqs = zip(*lookups) + if any(cand is not None for cand in cands): + return 0 + spec_sets = (ireq.specifier for ireq in ireqs if ireq) + operators = ( + specifier.operator + for spec_set in spec_sets + for specifier in spec_set + ) + if any(op in ("==", "===") for op in operators): + return 1 + return 2 + + restrictive = _get_restrictive_rating(req for req, _ in information) transitive = all(parent is not None for _, parent in information) key = next(iter(candidates)).name if candidates else "" - return (transitive, key) + return (restrictive, transitive, key) def find_matches(self, requirements): # type: (Sequence[Requirement]) -> Iterable[Candidate] |