SWAMPy

✎ CARVE FLESH

EDITING: wheel.py

"""Support functions for working with wheel files.
"""

import logging
from email.message import Message
from email.parser import Parser
from typing import Tuple
from zipfile import BadZipFile, ZipFile

from pip._vendor.packaging.utils import canonicalize_name

from pip._internal.exceptions import UnsupportedWheel

VERSION_COMPATIBLE = (1, 0)

logger = logging.getLogger(__name__)

def parse_wheel(wheel_zip: ZipFile, name: str) -> Tuple[str, Message]:
    """Extract information from the provided wheel, ensuring it meets basic
    standards.

Returns the name of the .dist-info directory and the parsed WHEEL metadata.
    """
    try:
        info_dir = wheel_dist_info_dir(wheel_zip, name)
        metadata = wheel_metadata(wheel_zip, info_dir)
        version = wheel_version(metadata)
    except UnsupportedWheel as e:
        raise UnsupportedWheel(f"{name} has an invalid wheel, {str(e)}")

check_compatibility(version, name)

return info_dir, metadata

def wheel_dist_info_dir(source: ZipFile, name: str) -> str:
    """Returns the name of the contained .dist-info directory.

Raises AssertionError or UnsupportedWheel if not found, >1 found, or
    it doesn't match the provided name.
    """
    # Zip file path separators must be /
    subdirs = {p.split("/", 1)[0] for p in source.namelist()}

info_dirs = [s for s in subdirs if s.endswith(".dist-info")]

if not info_dirs:
        raise UnsupportedWheel(".dist-info directory not found")

if len(info_dirs) > 1:
        raise UnsupportedWheel(
            "multiple .dist-info directories found: {}".format(", ".join(info_dirs))
        )

info_dir = info_dirs[0]

info_dir_name = canonicalize_name(info_dir)
    canonical_name = canonicalize_name(name)
    if not info_dir_name.startswith(canonical_name):
        raise UnsupportedWheel(
            f".dist-info directory {info_dir!r} does not start with {canonical_name!r}"
        )

return info_dir

def read_wheel_metadata_file(source: ZipFile, path: str) -> bytes:
    try:
        return source.read(path)
        # BadZipFile for general corruption, KeyError for missing entry,
        # and RuntimeError for password-protected files
    except (BadZipFile, KeyError, RuntimeError) as e:
        raise UnsupportedWheel(f"could not read {path!r} file: {e!r}")

def wheel_metadata(source: ZipFile, dist_info_dir: str) -> Message:
    """Return the WHEEL metadata of an extracted wheel, if possible.
    Otherwise, raise UnsupportedWheel.
    """
    path = f"{dist_info_dir}/WHEEL"
    # Zip file path separators must be /
    wheel_contents = read_wheel_metadata_file(source, path)

try:
        wheel_text = wheel_contents.decode()
    except UnicodeDecodeError as e:
        raise UnsupportedWheel(f"error decoding {path!r}: {e!r}")

# FeedParser (used by Parser) does not raise any exceptions. The returned
    # message may have .defects populated, but for backwards-compatibility we
    # currently ignore them.
    return Parser().parsestr(wheel_text)

def wheel_version(wheel_data: Message) -> Tuple[int, ...]:
    """Given WHEEL metadata, return the parsed Wheel-Version.
    Otherwise, raise UnsupportedWheel.
    """
    version_text = wheel_data["Wheel-Version"]
    if version_text is None:
        raise UnsupportedWheel("WHEEL is missing Wheel-Version")

version = version_text.strip()

try:
        return tuple(map(int, version.split(".")))
    except ValueError:
        raise UnsupportedWheel(f"invalid Wheel-Version: {version!r}")

def check_compatibility(version: Tuple[int, ...], name: str) -> None:
    """Raises errors or warns if called with an incompatible Wheel-Version.

pip should refuse to install a Wheel-Version that's a major series
    ahead of what it's compatible with (e.g 2.0 > 1.1); and warn when
    installing a version only minor version ahead (e.g 1.2 > 1.1).

version: a 2-tuple representing a Wheel-Version (Major, Minor)
    name: name of wheel or package to raise exception about

:raises UnsupportedWheel: when an incompatible Wheel-Version is given
    """
    if version[0] > VERSION_COMPATIBLE[0]:
        raise UnsupportedWheel(
            "{}'s Wheel-Version ({}) is not compatible with this version "
            "of pip".format(name, ".".join(map(str, version)))
        )
    elif version > VERSION_COMPATIBLE:
        logger.warning(
            "Installing from a newer Wheel-Version (%s)",
            ".".join(map(str, version)),
        )

CANCEL

Name	Type	Size	Modified	Actions
↩ ..	DIR	—	—
📄 __init__.py	PY	0 B	2025-09-22 19:18	EDIT
📄 _jaraco_text.py	PY	3.3 KB	2025-09-22 19:18	EDIT
📄 _log.py	PY	1015 B	2025-09-22 19:18	EDIT
📄 appdirs.py	PY	1.6 KB	2025-09-22 19:18	EDIT
📄 compat.py	PY	1.8 KB	2025-09-22 19:18	EDIT
📄 compatibility_tags.py	PY	5.3 KB	2025-09-22 19:18	EDIT
📄 datetime.py	PY	242 B	2025-09-22 19:18	EDIT
📄 deprecation.py	PY	3.5 KB	2025-09-22 19:18	EDIT
📄 direct_url_helpers.py	PY	3.1 KB	2025-09-22 19:18	EDIT
📄 egg_link.py	PY	2.4 KB	2025-09-22 19:18	EDIT
📄 encoding.py	PY	1.1 KB	2025-09-22 19:18	EDIT
📄 entrypoints.py	PY	3 KB	2025-09-22 19:18	EDIT
📄 filesystem.py	PY	5 KB	2025-09-22 19:18	EDIT
📄 filetypes.py	PY	716 B	2025-09-22 19:18	EDIT
📄 glibc.py	PY	3 KB	2025-09-22 19:18	EDIT
📄 hashes.py	PY	5 KB	2025-09-22 19:18	EDIT
📄 logging.py	PY	11.3 KB	2025-09-22 19:18	EDIT
📄 misc.py	PY	23.1 KB	2025-09-22 19:18	EDIT
📄 models.py	PY	1.2 KB	2025-09-22 19:18	EDIT
📄 packaging.py	PY	2.1 KB	2025-09-22 19:18	EDIT
📄 setuptools_build.py	PY	4.3 KB	2025-09-22 19:18	EDIT
📄 subprocess.py	PY	9 KB	2025-09-22 19:18	EDIT
📄 temp_dir.py	PY	9.1 KB	2025-09-22 19:18	EDIT
📄 unpacking.py	PY	8.6 KB	2025-09-22 19:18	EDIT
📄 urls.py	PY	1.7 KB	2025-09-22 19:18	EDIT
📄 virtualenv.py	PY	3.4 KB	2025-09-22 19:18	EDIT
📄 wheel.py	PY	4.4 KB	2025-09-22 19:18	EDIT