Mini Shell
Mini Shell
| Direktori : /snap/certbot/current/lib64/python3.12/site-packages/wheel/_commands/ |
| Current File : //snap/certbot/current/lib64/python3.12/site-packages/wheel/_commands/pack.py |
from __future__ import annotations
import email.policy
import os.path
import re
from email.generator import BytesGenerator
from email.parser import BytesParser
from packaging.version import InvalidVersion, Version
from ..wheelfile import WheelError, WheelFile
DIST_INFO_RE = re.compile(r"^(?P<namever>(?P<name>.+?)-(?P<ver>\d.*?))\.dist-info$")
def pack(
directory: str,
dest_dir: str,
build_number: str | None,
local_version: str | None = None,
) -> None:
"""Repack a previously unpacked wheel directory into a new wheel file.
The .dist-info/WHEEL file must contain one or more tags so that the target
wheel file name can be determined.
:param directory: The unpacked wheel directory
:param dest_dir: Destination directory (defaults to the current directory)
:param build_number: Build tag to use in the wheel name
:param local_version: Local version identifier to add or replace
"""
# Find the .dist-info directory
dist_info_dirs = [
fn
for fn in os.listdir(directory)
if os.path.isdir(os.path.join(directory, fn)) and DIST_INFO_RE.match(fn)
]
if len(dist_info_dirs) > 1:
raise WheelError(f"Multiple .dist-info directories found in {directory}")
elif not dist_info_dirs:
raise WheelError(f"No .dist-info directories found in {directory}")
# Determine the target wheel filename
dist_info_dir = dist_info_dirs[0]
match = DIST_INFO_RE.match(dist_info_dir)
name_version = match.group("namever")
name = match.group("name")
version = match.group("ver")
# Update the local version identifier if requested
if local_version is not None:
base_version = version.split("+", 1)[0]
new_version = (
base_version if local_version == "" else f"{base_version}+{local_version}"
)
try:
Version(new_version)
except InvalidVersion as e:
raise WheelError(f"Invalid version {new_version!r}: {e}") from None
if "-" in new_version:
raise WheelError(
f"Invalid local version identifier {local_version!r}: "
"wheel filenames cannot contain '-' in the version; "
"use '_' instead of '-'"
)
new_namever = f"{name}-{new_version}"
new_dist_info_dir = f"{new_namever}.dist-info"
os.rename(
os.path.join(directory, dist_info_dir),
os.path.join(directory, new_dist_info_dir),
)
dist_info_dir = new_dist_info_dir
name_version = new_namever
# Update the Version field in .dist-info/METADATA
metadata_path = os.path.join(directory, dist_info_dir, "METADATA")
try:
with open(metadata_path, "rb") as f:
metadata = BytesParser(policy=email.policy.compat32).parse(f)
except FileNotFoundError:
raise WheelError(f"Missing {dist_info_dir}/METADATA file") from None
del metadata["Version"]
metadata["Version"] = new_version
with open(metadata_path, "wb") as f:
BytesGenerator(f, maxheaderlen=0).flatten(metadata)
# Read the tags and the existing build number from .dist-info/WHEEL
wheel_file_path = os.path.join(directory, dist_info_dir, "WHEEL")
with open(wheel_file_path, "rb") as f:
info = BytesParser(policy=email.policy.compat32).parse(f)
tags: list[str] = info.get_all("Tag", [])
existing_build_number = info.get("Build")
if not tags:
raise WheelError(
f"No tags present in {dist_info_dir}/WHEEL; cannot determine target "
f"wheel filename"
)
# Set the wheel file name and add/replace/remove the Build tag in .dist-info/WHEEL
build_number = build_number if build_number is not None else existing_build_number
if build_number is not None:
del info["Build"]
if build_number:
info["Build"] = build_number
name_version += "-" + build_number
if build_number != existing_build_number:
with open(wheel_file_path, "wb") as f:
BytesGenerator(f, maxheaderlen=0).flatten(info)
# Reassemble the tags for the wheel file
tagline = compute_tagline(tags)
# Repack the wheel
wheel_path = os.path.join(dest_dir, f"{name_version}-{tagline}.whl")
with WheelFile(wheel_path, "w") as wf:
print(f"Repacking wheel as {wheel_path}...", end="", flush=True)
wf.write_files(directory)
print("OK")
def compute_tagline(tags: list[str]) -> str:
"""Compute a tagline from a list of tags.
:param tags: A list of tags
:return: A tagline
"""
impls = sorted({tag.split("-")[0] for tag in tags})
abivers = sorted({tag.split("-")[1] for tag in tags})
platforms = sorted({tag.split("-")[2] for tag in tags})
return "-".join([".".join(impls), ".".join(abivers), ".".join(platforms)])
República de Moçambique
Ministério da Economia e Finanças