|
import functools |
|
import logging |
|
from email.message import Message |
|
from email.parser import FeedParser |
|
from typing import Optional, Tuple |
|
|
|
from pip._vendor import pkg_resources |
|
from pip._vendor.packaging import specifiers, version |
|
from pip._vendor.packaging.requirements import Requirement |
|
from pip._vendor.pkg_resources import Distribution |
|
|
|
from pip._internal.exceptions import NoneMetadataError |
|
from pip._internal.utils.misc import display_path |
|
|
|
logger = logging.getLogger(__name__) |
|
|
|
|
|
def check_requires_python( |
|
requires_python: Optional[str], version_info: Tuple[int, ...] |
|
) -> bool: |
|
""" |
|
Check if the given Python version matches a "Requires-Python" specifier. |
|
|
|
:param version_info: A 3-tuple of ints representing a Python |
|
major-minor-micro version to check (e.g. `sys.version_info[:3]`). |
|
|
|
:return: `True` if the given Python version satisfies the requirement. |
|
Otherwise, return `False`. |
|
|
|
:raises InvalidSpecifier: If `requires_python` has an invalid format. |
|
""" |
|
if requires_python is None: |
|
|
|
return True |
|
requires_python_specifier = specifiers.SpecifierSet(requires_python) |
|
|
|
python_version = version.parse(".".join(map(str, version_info))) |
|
return python_version in requires_python_specifier |
|
|
|
|
|
def get_metadata(dist: Distribution) -> Message: |
|
""" |
|
:raises NoneMetadataError: if the distribution reports `has_metadata()` |
|
True but `get_metadata()` returns None. |
|
""" |
|
metadata_name = "METADATA" |
|
if isinstance(dist, pkg_resources.DistInfoDistribution) and dist.has_metadata( |
|
metadata_name |
|
): |
|
metadata = dist.get_metadata(metadata_name) |
|
elif dist.has_metadata("PKG-INFO"): |
|
metadata_name = "PKG-INFO" |
|
metadata = dist.get_metadata(metadata_name) |
|
else: |
|
logger.warning("No metadata found in %s", display_path(dist.location)) |
|
metadata = "" |
|
|
|
if metadata is None: |
|
raise NoneMetadataError(dist, metadata_name) |
|
|
|
feed_parser = FeedParser() |
|
|
|
|
|
feed_parser.feed(metadata) |
|
return feed_parser.close() |
|
|
|
|
|
def get_installer(dist: Distribution) -> str: |
|
if dist.has_metadata("INSTALLER"): |
|
for line in dist.get_metadata_lines("INSTALLER"): |
|
if line.strip(): |
|
return line.strip() |
|
return "" |
|
|
|
|
|
@functools.lru_cache(maxsize=512) |
|
def get_requirement(req_string: str) -> Requirement: |
|
"""Construct a packaging.Requirement object with caching""" |
|
|
|
|
|
|
|
|
|
|
|
return Requirement(req_string) |
|
|