Repository navigation
Expand file tree
/
Copy pathpatchlevel.py
More file actions
77 lines (62 loc) · 2.37 KB
/
Copy pathpatchlevel.py
File metadata and controls
77 lines (62 loc) · 2.37 KB
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
"""Extract version information from Include/patchlevel.h."""
import re
import sys
from pathlib import Path
from typing import Literal, NamedTuple
CPYTHON_ROOT = Path(
__file__, # cpython/Doc/tools/extensions/patchlevel.py
"..", # cpython/Doc/tools/extensions
"..", # cpython/Doc/tools
"..", # cpython/Doc
"..", # cpython
).resolve()
PATCHLEVEL_H = CPYTHON_ROOT / "Include" / "patchlevel.h"
RELEASE_LEVELS = {
"PY_RELEASE_LEVEL_ALPHA": "alpha",
"PY_RELEASE_LEVEL_BETA": "beta",
"PY_RELEASE_LEVEL_GAMMA": "candidate",
"PY_RELEASE_LEVEL_FINAL": "final",
}
class version_info(NamedTuple): # noqa: N801
major: int #: Major release number
minor: int #: Minor release number
micro: int #: Patch release number
releaselevel: Literal["alpha", "beta", "candidate", "final"]
serial: int #: Serial release number
def get_header_version_info() -> version_info:
# Capture PY_ prefixed #defines.
pat = re.compile(r"\s*#define\s+(PY_\w*)\s+(\w+)", re.ASCII)
defines = {}
patchlevel_h = PATCHLEVEL_H.read_text(encoding="utf-8")
for line in patchlevel_h.splitlines():
if (m := pat.match(line)) is not None:
name, value = m.groups()
defines[name] = value
return version_info(
major=int(defines["PY_MAJOR_VERSION"]),
minor=int(defines["PY_MINOR_VERSION"]),
micro=int(defines["PY_MICRO_VERSION"]),
releaselevel=RELEASE_LEVELS[defines["PY_RELEASE_LEVEL"]],
serial=int(defines["PY_RELEASE_SERIAL"]),
)
def format_version_info(info: version_info) -> tuple[str, str]:
version = f"{info.major}.{info.minor}"
release = f"{info.major}.{info.minor}.{info.micro}"
if info.releaselevel != "final":
suffix = {"alpha": "a", "beta": "b", "candidate": "rc"}
release += f"{suffix[info.releaselevel]}{info.serial}"
return version, release
def get_version_info():
try:
info = get_header_version_info()
return format_version_info(info)
except OSError:
version, release = format_version_info(sys.version_info)
print(
f"Failed to get version info from Include/patchlevel.h, "
f"using version of this interpreter ({release}).",
file=sys.stderr,
)
return version, release
if __name__ == "__main__":
print(format_version_info(get_header_version_info())[1])