mirror of https://github.com/pypa/pip
Add find_requirement() wrapper in legacy resolver
This commit is contained in:
parent
daff81124a
commit
f35f37ef72
|
@ -889,11 +889,11 @@ class PackageFinder(object):
|
|||
return candidate_evaluator.compute_best_candidate(candidates)
|
||||
|
||||
def find_requirement(self, req, upgrade):
|
||||
# type: (InstallRequirement, bool) -> Optional[Link]
|
||||
# type: (InstallRequirement, bool) -> Optional[InstallationCandidate]
|
||||
"""Try to find a Link matching req
|
||||
|
||||
Expects req, an InstallRequirement and upgrade, a boolean
|
||||
Returns a Link if found,
|
||||
Returns a InstallationCandidate if found,
|
||||
Raises DistributionNotFound or BestVersionAlreadyInstalled otherwise
|
||||
"""
|
||||
hashes = req.hashes(trust_internet=False)
|
||||
|
@ -967,7 +967,7 @@ class PackageFinder(object):
|
|||
best_candidate.version,
|
||||
_format_versions(best_candidate_result.iter_applicable()),
|
||||
)
|
||||
return best_candidate.link
|
||||
return best_candidate
|
||||
|
||||
|
||||
def _find_name_version_sep(fragment, canonical_name):
|
||||
|
|
|
@ -46,6 +46,7 @@ if MYPY_CHECK_RUNNING:
|
|||
from pip._internal.cache import WheelCache
|
||||
from pip._internal.distributions import AbstractDistribution
|
||||
from pip._internal.index.package_finder import PackageFinder
|
||||
from pip._internal.models.link import Link
|
||||
from pip._internal.operations.prepare import RequirementPreparer
|
||||
from pip._internal.req.req_install import InstallRequirement
|
||||
from pip._internal.resolution.base import InstallRequirementProvider
|
||||
|
@ -260,6 +261,14 @@ class Resolver(BaseResolver):
|
|||
self._set_req_to_reinstall(req_to_install)
|
||||
return None
|
||||
|
||||
def _find_requirement_link(self, req):
|
||||
# type: (InstallRequirement) -> Optional[Link]
|
||||
upgrade = self._is_upgrade_allowed(req)
|
||||
best_candidate = self.finder.find_requirement(req, upgrade)
|
||||
if not best_candidate:
|
||||
return None
|
||||
return best_candidate.link
|
||||
|
||||
def _populate_link(self, req):
|
||||
# type: (InstallRequirement) -> None
|
||||
"""Ensure that if a link can be found for this, that it is found.
|
||||
|
@ -274,9 +283,8 @@ class Resolver(BaseResolver):
|
|||
mismatches. Furthermore, cached wheels at present have undeterministic
|
||||
contents due to file modification times.
|
||||
"""
|
||||
upgrade = self._is_upgrade_allowed(req)
|
||||
if req.link is None:
|
||||
req.link = self.finder.find_requirement(req, upgrade)
|
||||
req.link = self._find_requirement_link(req)
|
||||
|
||||
if self.wheel_cache is None or self.preparer.require_hashes:
|
||||
return
|
||||
|
|
|
@ -48,7 +48,7 @@ def test_no_mpkg(data):
|
|||
req = install_req_from_line("pkgwithmpkg")
|
||||
found = finder.find_requirement(req, False)
|
||||
|
||||
assert found.url.endswith("pkgwithmpkg-1.0.tar.gz"), found
|
||||
assert found.link.url.endswith("pkgwithmpkg-1.0.tar.gz"), found
|
||||
|
||||
|
||||
def test_no_partial_name_match(data):
|
||||
|
@ -57,7 +57,7 @@ def test_no_partial_name_match(data):
|
|||
req = install_req_from_line("gmpy")
|
||||
found = finder.find_requirement(req, False)
|
||||
|
||||
assert found.url.endswith("gmpy-1.15.tar.gz"), found
|
||||
assert found.link.url.endswith("gmpy-1.15.tar.gz"), found
|
||||
|
||||
|
||||
def test_tilde():
|
||||
|
@ -79,23 +79,23 @@ def test_duplicates_sort_ok(data):
|
|||
req = install_req_from_line("duplicate")
|
||||
found = finder.find_requirement(req, False)
|
||||
|
||||
assert found.url.endswith("duplicate-1.0.tar.gz"), found
|
||||
assert found.link.url.endswith("duplicate-1.0.tar.gz"), found
|
||||
|
||||
|
||||
def test_finder_detects_latest_find_links(data):
|
||||
"""Test PackageFinder detects latest using find-links"""
|
||||
req = install_req_from_line('simple', None)
|
||||
finder = make_test_finder(find_links=[data.find_links])
|
||||
link = finder.find_requirement(req, False)
|
||||
assert link.url.endswith("simple-3.0.tar.gz")
|
||||
found = finder.find_requirement(req, False)
|
||||
assert found.link.url.endswith("simple-3.0.tar.gz")
|
||||
|
||||
|
||||
def test_incorrect_case_file_index(data):
|
||||
"""Test PackageFinder detects latest using wrong case"""
|
||||
req = install_req_from_line('dinner', None)
|
||||
finder = make_test_finder(index_urls=[data.find_links3])
|
||||
link = finder.find_requirement(req, False)
|
||||
assert link.url.endswith("Dinner-2.0.tar.gz")
|
||||
found = finder.find_requirement(req, False)
|
||||
assert found.link.url.endswith("Dinner-2.0.tar.gz")
|
||||
|
||||
|
||||
@pytest.mark.network
|
||||
|
@ -180,7 +180,7 @@ class TestWheel:
|
|||
finder = make_test_finder(find_links=[data.find_links])
|
||||
found = finder.find_requirement(req, True)
|
||||
assert (
|
||||
found.url.endswith("simple.dist-0.1-py2.py3-none-any.whl")
|
||||
found.link.url.endswith("simple.dist-0.1-py2.py3-none-any.whl")
|
||||
), found
|
||||
|
||||
def test_wheel_over_sdist_priority(self, data):
|
||||
|
@ -191,7 +191,8 @@ class TestWheel:
|
|||
req = install_req_from_line("priority")
|
||||
finder = make_test_finder(find_links=[data.find_links])
|
||||
found = finder.find_requirement(req, True)
|
||||
assert found.url.endswith("priority-1.0-py2.py3-none-any.whl"), found
|
||||
assert found.link.url.endswith("priority-1.0-py2.py3-none-any.whl"), \
|
||||
found
|
||||
|
||||
def test_existing_over_wheel_priority(self, data):
|
||||
"""
|
||||
|
@ -292,8 +293,8 @@ def test_finder_priority_file_over_page(data):
|
|||
assert all(version.link.scheme == 'https'
|
||||
for version in all_versions[1:]), all_versions
|
||||
|
||||
link = finder.find_requirement(req, False)
|
||||
assert link.url.startswith("file://")
|
||||
found = finder.find_requirement(req, False)
|
||||
assert found.link.url.startswith("file://")
|
||||
|
||||
|
||||
def test_finder_priority_nonegg_over_eggfragments():
|
||||
|
@ -306,9 +307,9 @@ def test_finder_priority_nonegg_over_eggfragments():
|
|||
assert all_versions[0].link.url.endswith('tar.gz')
|
||||
assert all_versions[1].link.url.endswith('#egg=bar-1.0')
|
||||
|
||||
link = finder.find_requirement(req, False)
|
||||
found = finder.find_requirement(req, False)
|
||||
|
||||
assert link.url.endswith('tar.gz')
|
||||
assert found.link.url.endswith('tar.gz')
|
||||
|
||||
links.reverse()
|
||||
|
||||
|
@ -316,9 +317,9 @@ def test_finder_priority_nonegg_over_eggfragments():
|
|||
all_versions = finder.find_all_candidates(req.name)
|
||||
assert all_versions[0].link.url.endswith('tar.gz')
|
||||
assert all_versions[1].link.url.endswith('#egg=bar-1.0')
|
||||
link = finder.find_requirement(req, False)
|
||||
found = finder.find_requirement(req, False)
|
||||
|
||||
assert link.url.endswith('tar.gz')
|
||||
assert found.link.url.endswith('tar.gz')
|
||||
|
||||
|
||||
def test_finder_only_installs_stable_releases(data):
|
||||
|
@ -330,21 +331,21 @@ def test_finder_only_installs_stable_releases(data):
|
|||
|
||||
# using a local index (that has pre & dev releases)
|
||||
finder = make_test_finder(index_urls=[data.index_url("pre")])
|
||||
link = finder.find_requirement(req, False)
|
||||
assert link.url.endswith("bar-1.0.tar.gz"), link.url
|
||||
found = finder.find_requirement(req, False)
|
||||
assert found.link.url.endswith("bar-1.0.tar.gz"), found.link.url
|
||||
|
||||
# using find-links
|
||||
links = ["https://foo/bar-1.0.tar.gz", "https://foo/bar-2.0b1.tar.gz"]
|
||||
|
||||
finder = make_no_network_finder(links)
|
||||
link = finder.find_requirement(req, False)
|
||||
assert link.url == "https://foo/bar-1.0.tar.gz"
|
||||
found = finder.find_requirement(req, False)
|
||||
assert found.link.url == "https://foo/bar-1.0.tar.gz"
|
||||
|
||||
links.reverse()
|
||||
|
||||
finder = make_no_network_finder(links)
|
||||
link = finder.find_requirement(req, False)
|
||||
assert link.url == "https://foo/bar-1.0.tar.gz"
|
||||
found = finder.find_requirement(req, False)
|
||||
assert found.link.url == "https://foo/bar-1.0.tar.gz"
|
||||
|
||||
|
||||
def test_finder_only_installs_data_require(data):
|
||||
|
@ -383,21 +384,21 @@ def test_finder_installs_pre_releases(data):
|
|||
index_urls=[data.index_url("pre")],
|
||||
allow_all_prereleases=True,
|
||||
)
|
||||
link = finder.find_requirement(req, False)
|
||||
assert link.url.endswith("bar-2.0b1.tar.gz"), link.url
|
||||
found = finder.find_requirement(req, False)
|
||||
assert found.link.url.endswith("bar-2.0b1.tar.gz"), found.link.url
|
||||
|
||||
# using find-links
|
||||
links = ["https://foo/bar-1.0.tar.gz", "https://foo/bar-2.0b1.tar.gz"]
|
||||
|
||||
finder = make_no_network_finder(links, allow_all_prereleases=True)
|
||||
link = finder.find_requirement(req, False)
|
||||
assert link.url == "https://foo/bar-2.0b1.tar.gz"
|
||||
found = finder.find_requirement(req, False)
|
||||
assert found.link.url == "https://foo/bar-2.0b1.tar.gz"
|
||||
|
||||
links.reverse()
|
||||
|
||||
finder = make_no_network_finder(links, allow_all_prereleases=True)
|
||||
link = finder.find_requirement(req, False)
|
||||
assert link.url == "https://foo/bar-2.0b1.tar.gz"
|
||||
found = finder.find_requirement(req, False)
|
||||
assert found.link.url == "https://foo/bar-2.0b1.tar.gz"
|
||||
|
||||
|
||||
def test_finder_installs_dev_releases(data):
|
||||
|
@ -412,8 +413,8 @@ def test_finder_installs_dev_releases(data):
|
|||
index_urls=[data.index_url("dev")],
|
||||
allow_all_prereleases=True,
|
||||
)
|
||||
link = finder.find_requirement(req, False)
|
||||
assert link.url.endswith("bar-2.0.dev1.tar.gz"), link.url
|
||||
found = finder.find_requirement(req, False)
|
||||
assert found.link.url.endswith("bar-2.0.dev1.tar.gz"), found.link.url
|
||||
|
||||
|
||||
def test_finder_installs_pre_releases_with_version_spec():
|
||||
|
@ -424,14 +425,14 @@ def test_finder_installs_pre_releases_with_version_spec():
|
|||
links = ["https://foo/bar-1.0.tar.gz", "https://foo/bar-2.0b1.tar.gz"]
|
||||
|
||||
finder = make_no_network_finder(links)
|
||||
link = finder.find_requirement(req, False)
|
||||
assert link.url == "https://foo/bar-2.0b1.tar.gz"
|
||||
found = finder.find_requirement(req, False)
|
||||
assert found.link.url == "https://foo/bar-2.0b1.tar.gz"
|
||||
|
||||
links.reverse()
|
||||
|
||||
finder = make_no_network_finder(links)
|
||||
link = finder.find_requirement(req, False)
|
||||
assert link.url == "https://foo/bar-2.0b1.tar.gz"
|
||||
found = finder.find_requirement(req, False)
|
||||
assert found.link.url == "https://foo/bar-2.0b1.tar.gz"
|
||||
|
||||
|
||||
class TestLinkEvaluator(object):
|
||||
|
|
Loading…
Reference in New Issue