Skip to content

Commit 2f1be4e

Browse files
committed
Rename index.py's _egg_info_matches(egg_info, canonical_name).
1 parent 32dcaa6 commit 2f1be4e

File tree

2 files changed

+29
-25
lines changed

2 files changed

+29
-25
lines changed

src/pip/_internal/index.py

Lines changed: 18 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -413,7 +413,9 @@ def evaluate_link(self, link, search):
413413
return (False, 'No sources permitted for %s' % search.supplied)
414414

415415
if not version:
416-
version = _egg_info_matches(egg_info, search.canonical)
416+
version = _extract_version_from_fragment(
417+
egg_info, search.canonical,
418+
)
417419
if not version:
418420
return (False, 'Missing project version for %s' % search.supplied)
419421

@@ -1117,45 +1119,47 @@ def _package_versions(
11171119
return result
11181120

11191121

1120-
def _find_name_version_sep(egg_info, canonical_name):
1122+
def _find_name_version_sep(fragment, canonical_name):
11211123
# type: (str, str) -> int
11221124
"""Find the separator's index based on the package's canonical name.
11231125
1124-
`egg_info` must be an egg info string for the given package, and
1125-
`canonical_name` must be the package's canonical name.
1126+
:param fragment: A <package>+<version> filename "fragment" (stem) or
1127+
egg fragment.
1128+
:param canonical_name: The package's canonical name.
11261129
11271130
This function is needed since the canonicalized name does not necessarily
11281131
have the same length as the egg info's name part. An example::
11291132
1130-
>>> egg_info = 'foo__bar-1.0'
1133+
>>> fragment = 'foo__bar-1.0'
11311134
>>> canonical_name = 'foo-bar'
1132-
>>> _find_name_version_sep(egg_info, canonical_name)
1135+
>>> _find_name_version_sep(fragment, canonical_name)
11331136
8
11341137
"""
11351138
# Project name and version must be separated by one single dash. Find all
11361139
# occurrences of dashes; if the string in front of it matches the canonical
11371140
# name, this is the one separating the name and version parts.
1138-
for i, c in enumerate(egg_info):
1141+
for i, c in enumerate(fragment):
11391142
if c != "-":
11401143
continue
1141-
if canonicalize_name(egg_info[:i]) == canonical_name:
1144+
if canonicalize_name(fragment[:i]) == canonical_name:
11421145
return i
1143-
raise ValueError("{} does not match {}".format(egg_info, canonical_name))
1146+
raise ValueError("{} does not match {}".format(fragment, canonical_name))
11441147

11451148

1146-
def _egg_info_matches(egg_info, canonical_name):
1149+
def _extract_version_from_fragment(fragment, canonical_name):
11471150
# type: (str, str) -> Optional[str]
1148-
"""Pull the version part out of a string.
1151+
"""Parse the version string from a <package>+<version> filename
1152+
"fragment" (stem) or egg fragment.
11491153
1150-
:param egg_info: The string to parse. E.g. foo-2.1
1154+
:param fragment: The string to parse. E.g. foo-2.1
11511155
:param canonical_name: The canonicalized name of the package this
11521156
belongs to.
11531157
"""
11541158
try:
1155-
version_start = _find_name_version_sep(egg_info, canonical_name) + 1
1159+
version_start = _find_name_version_sep(fragment, canonical_name) + 1
11561160
except ValueError:
11571161
return None
1158-
version = egg_info[version_start:]
1162+
version = fragment[version_start:]
11591163
if not version:
11601164
return None
11611165
return version

tests/unit/test_index.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from pip._internal.index import (
1010
CandidateEvaluator, HTMLPage, Link, PackageFinder, Search,
1111
_check_link_requires_python, _clean_link, _determine_base_url,
12-
_egg_info_matches, _find_name_version_sep, _get_html_page,
12+
_extract_version_from_fragment, _find_name_version_sep, _get_html_page,
1313
)
1414
from pip._internal.models.candidate import InstallationCandidate
1515
from pip._internal.models.search_scope import SearchScope
@@ -510,7 +510,7 @@ def test_secure_origin(location, trusted, expected):
510510

511511

512512
@pytest.mark.parametrize(
513-
("egg_info", "canonical_name", "expected"),
513+
("fragment", "canonical_name", "expected"),
514514
[
515515
# Trivial.
516516
("pip-18.0", "pip", 3),
@@ -539,13 +539,13 @@ def test_secure_origin(location, trusted, expected):
539539
("zope.interface-", "zope-interface", 14),
540540
],
541541
)
542-
def test_find_name_version_sep(egg_info, canonical_name, expected):
543-
index = _find_name_version_sep(egg_info, canonical_name)
542+
def test_find_name_version_sep(fragment, canonical_name, expected):
543+
index = _find_name_version_sep(fragment, canonical_name)
544544
assert index == expected
545545

546546

547547
@pytest.mark.parametrize(
548-
("egg_info", "canonical_name"),
548+
("fragment", "canonical_name"),
549549
[
550550
# A dash must follow the package name.
551551
("zope.interface4.5.0", "zope-interface"),
@@ -554,15 +554,15 @@ def test_find_name_version_sep(egg_info, canonical_name, expected):
554554
("zope.interface", "zope-interface"),
555555
],
556556
)
557-
def test_find_name_version_sep_failure(egg_info, canonical_name):
557+
def test_find_name_version_sep_failure(fragment, canonical_name):
558558
with pytest.raises(ValueError) as ctx:
559-
_find_name_version_sep(egg_info, canonical_name)
560-
message = "{} does not match {}".format(egg_info, canonical_name)
559+
_find_name_version_sep(fragment, canonical_name)
560+
message = "{} does not match {}".format(fragment, canonical_name)
561561
assert str(ctx.value) == message
562562

563563

564564
@pytest.mark.parametrize(
565-
("egg_info", "canonical_name", "expected"),
565+
("fragment", "canonical_name", "expected"),
566566
[
567567
# Trivial.
568568
("pip-18.0", "pip", "18.0"),
@@ -594,8 +594,8 @@ def test_find_name_version_sep_failure(egg_info, canonical_name):
594594
("zope.interface", "zope-interface", None),
595595
],
596596
)
597-
def test_egg_info_matches(egg_info, canonical_name, expected):
598-
version = _egg_info_matches(egg_info, canonical_name)
597+
def test_extract_version_from_fragment(fragment, canonical_name, expected):
598+
version = _extract_version_from_fragment(fragment, canonical_name)
599599
assert version == expected
600600

601601

0 commit comments

Comments
 (0)