1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
"""Helper script to rebuild virtualenv_support. Downloads the wheel files using pip."""
from __future__ import annotations
import os
import shutil
import subprocess
import sys
from collections import OrderedDict, defaultdict
from pathlib import Path
from tempfile import TemporaryDirectory
from textwrap import dedent
from threading import Thread
STRICT = "UPGRADE_ADVISORY" not in os.environ
BUNDLED = ["pip", "setuptools", "wheel"]
SUPPORT = [(3, i) for i in range(8, 15)]
DEST = Path(__file__).resolve().parents[1] / "src" / "virtualenv" / "seed" / "wheels" / "embed"
def download(ver, dest, package):
subprocess.call(
[
sys.executable,
"-m",
"pip",
"--disable-pip-version-check",
"download",
"--no-cache-dir",
"--only-binary=:all:",
"--python-version",
ver,
"-d",
dest,
package,
],
)
def run(): # noqa: C901, PLR0912
old_batch = {i.name for i in DEST.iterdir() if i.suffix == ".whl"}
with TemporaryDirectory() as temp:
temp_path = Path(temp)
folders = {}
targets = []
for support in SUPPORT:
support_ver = ".".join(str(i) for i in support)
into = temp_path / support_ver
into.mkdir()
folders[into] = support_ver
for package in BUNDLED:
if package == "wheel" and support >= (3, 9):
continue
thread = Thread(target=download, args=(support_ver, str(into), package))
targets.append(thread)
thread.start()
for thread in targets:
thread.join()
new_batch = {i.name: i for f in folders for i in Path(f).iterdir()}
new_packages = new_batch.keys() - old_batch
remove_packages = old_batch - new_batch.keys()
for package in remove_packages:
(DEST / package).unlink()
for package in new_packages:
shutil.copy2(str(new_batch[package]), DEST / package)
added = collect_package_versions(new_packages)
removed = collect_package_versions(remove_packages)
outcome = (1 if STRICT else 0) if (added or removed) else 0
print(f"Outcome {outcome} added {added} removed {removed}") # noqa: T201
lines = ["Upgrade embedded wheels:", ""]
for key, versions in added.items():
text = f"* {key} to {fmt_version(versions)}"
if key in removed:
rem = ", ".join(f"``{i}``" for i in removed[key])
text += f" from {rem}"
del removed[key]
lines.append(text)
for key, versions in removed.items():
lines.append(f"Removed {key} of {fmt_version(versions)}")
lines.append("")
changelog = "\n".join(lines)
print(changelog) # noqa: T201
if len(lines) >= 4: # noqa: PLR2004
(Path(__file__).parents[1] / "docs" / "changelog" / "u.bugfix.rst").write_text(changelog, encoding="utf-8")
support_table = OrderedDict((".".join(str(j) for j in i), []) for i in SUPPORT)
for package in sorted(new_batch.keys()):
for folder, version in sorted(folders.items()):
if (folder / package).exists():
support_table[version].append(package)
support_table = {k: OrderedDict((i.split("-")[0], i) for i in v) for k, v in support_table.items()}
nl = "\n"
bundle = "".join(
f"\n {v!r}: {{{nl}{''.join(f' {p!r}: {f!r},{nl}' for p, f in line.items())} }},"
for v, line in support_table.items()
)
msg = dedent(
f"""
from pathlib import Path
from virtualenv.seed.wheels.util import Wheel
BUNDLE_FOLDER = Path(__file__).absolute().parent
BUNDLE_SUPPORT = {{ {bundle} }}
MAX = {next(iter(support_table.keys()))!r}
def get_embed_wheel(distribution, for_py_version):
mapping = BUNDLE_SUPPORT.get(for_py_version, {{}}) or BUNDLE_SUPPORT[MAX]
wheel_file = mapping.get(distribution)
if wheel_file is None:
return None
path = BUNDLE_FOLDER / wheel_file
return Wheel.from_path(path)
__all__ = [
"get_embed_wheel",
"BUNDLE_SUPPORT",
"MAX",
"BUNDLE_FOLDER",
]
""",
)
dest_target = DEST / "__init__.py"
dest_target.write_text(msg, encoding="utf-8")
subprocess.run([sys.executable, "-m", "ruff", "check", str(dest_target), "--fix", "--unsafe-fixes"])
subprocess.run([sys.executable, "-m", "ruff", "format", str(dest_target), "--preview"])
raise SystemExit(outcome)
def fmt_version(versions):
return ", ".join(f"``{v}``" for v in versions)
def collect_package_versions(new_packages):
result = defaultdict(list)
for package in new_packages:
split = package.split("-")
if len(split) < 2: # noqa: PLR2004
raise ValueError(package)
key, version = split[0:2]
result[key].append(version)
return result
if __name__ == "__main__":
run()
|
