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
|
#!/usr/bin/env python3
import json
import logging
import sys
from urllib.request import urlopen
import ruamel.yaml
PYPI_INDEX = "https://pypi.org/pypi"
YAML = ruamel.yaml.YAML()
YAML.indent(mapping=2, sequence=4, offset=2)
YAML.preserve_quotes = True
YAML.width = 180
logging.basicConfig(level="INFO", format="%(levelname)s: %(message)s")
log = logging.getLogger(__name__)
def get_package_type(module: ruamel.yaml.comments.CommentedMap) -> str:
url = module["sources"][0]["url"]
if ".whl" in url:
return "bdist_wheel"
return "sdist"
def get_current_version(
module: ruamel.yaml.comments.CommentedMap,
) -> tuple[str | None, str | None]:
name = module["name"].replace("python3-", "")
url = module["sources"][0]["url"]
if url.endswith(".git"):
return None, None
file_name = url.split("/")[-1]
file_name = file_name.replace(name, "")
parts = file_name.split("-")
pkg_version = parts[1].replace(".tar.gz", "")
py_version = parts[2] if ".tar.gz" not in parts[1] else None
return pkg_version, py_version
def get_latest_version(
package_name: str, package_type: str, py_version: str | None
) -> tuple[str, str, str, str]:
with urlopen(f"{PYPI_INDEX}/{package_name}/json") as f:
data = f.read()
d = json.loads(data)
version = d["info"]["version"]
sha = None
for entry in d["releases"][version]:
if entry["packagetype"] != package_type:
continue
if py_version is not None and entry["python_version"] != py_version:
continue
if sha is not None:
raise ValueError("No sha found")
sha = entry["digests"]["sha256"]
url = entry["url"]
filename = entry["filename"]
return version, sha, url, filename
raise ValueError("Package not found")
def update_module(module: ruamel.yaml.comments.CommentedMap) -> None:
if not module["name"].startswith("python3-"):
log.warning("Check %s manually", module["name"])
return
if "only-arches" in module:
log.warning("Update %s manually", module["name"])
return
package_type = get_package_type(module)
current_version, py_version = get_current_version(module)
if current_version is None:
print("Current version None")
return
name = module["name"].replace("python3-", "")
_latest_version, sha, url, filename = get_latest_version(
name, package_type, py_version
)
log.info("Update %s", name)
"".rsplit(" ", maxsplit=1) # noqa: SIM905
if module["sources"][0]["type"] == "file":
command = module["build-commands"][0]
current_filename = command.rsplit(" ", maxsplit=1)[1]
module["build-commands"][0] = command.replace(current_filename, filename)
module["sources"][0]["url"] = url
module["sources"][0]["sha256"] = sha
def update_modules(data) -> None:
for module in data["modules"]:
if isinstance(module, ruamel.yaml.comments.CommentedMap):
update_module(module)
if __name__ == "__main__":
if len(sys.argv) != 2:
sys.exit("Input yaml file is missing")
try:
with open(sys.argv[1]) as f:
data = YAML.load(f)
except (
FileNotFoundError,
ruamel.yaml.parser.ParserError,
ruamel.yaml.scanner.ScannerError,
) as error:
sys.exit(error)
update_modules(data)
with open(sys.argv[1], "w") as f:
YAML.dump(data, f)
|
