chendelian
ea25fc2638
All checks were successful
Publish Python Package to Gitea / publish (push) Successful in 1m2s
80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
"""Patch pyproject.toml and __init__.py version for CI builds (not committed)."""
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import re
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parent.parent
|
|
|
|
|
|
def _read_base_version() -> str:
|
|
text = (ROOT / "pyproject.toml").read_text(encoding="utf-8")
|
|
match = re.search(r'^version\s*=\s*"([^"]+)"', text, re.MULTILINE)
|
|
if not match:
|
|
raise SystemExit("ERROR: version not found in pyproject.toml")
|
|
base = match.group(1)
|
|
return re.sub(r"\.post\d+$", "", base)
|
|
|
|
|
|
def resolve_version() -> str:
|
|
ref = os.environ.get("GITHUB_REF", "")
|
|
if ref.startswith("refs/tags/v"):
|
|
return ref.removeprefix("refs/tags/v")
|
|
if ref.startswith("refs/tags/"):
|
|
return ref.removeprefix("refs/tags/")
|
|
|
|
run_number = (
|
|
os.environ.get("GITHUB_RUN_NUMBER")
|
|
or os.environ.get("GITHUB_RUN_ID")
|
|
or "0"
|
|
)
|
|
base = _read_base_version()
|
|
return f"{base}.post{run_number}"
|
|
|
|
|
|
def patch_version(version: str) -> None:
|
|
pyproject = ROOT / "pyproject.toml"
|
|
py_text = pyproject.read_text(encoding="utf-8")
|
|
py_text, n = re.subn(
|
|
r'^version\s*=\s*"[^"]*"',
|
|
f'version = "{version}"',
|
|
py_text,
|
|
count=1,
|
|
flags=re.MULTILINE,
|
|
)
|
|
if n != 1:
|
|
raise SystemExit("ERROR: failed to patch version in pyproject.toml")
|
|
pyproject.write_text(py_text, encoding="utf-8")
|
|
|
|
init_py = ROOT / "src" / "jiangchang_desktop_sdk" / "__init__.py"
|
|
init_text = init_py.read_text(encoding="utf-8")
|
|
init_text, n = re.subn(
|
|
r'^__version__\s*=\s*"[^"]*"',
|
|
f'__version__ = "{version}"',
|
|
init_text,
|
|
count=1,
|
|
flags=re.MULTILINE,
|
|
)
|
|
if n != 1:
|
|
raise SystemExit("ERROR: failed to patch __version__ in __init__.py")
|
|
init_py.write_text(init_text, encoding="utf-8")
|
|
|
|
|
|
def main() -> None:
|
|
version = resolve_version()
|
|
print(f"Resolved package version: {version}")
|
|
patch_version(version)
|
|
if os.environ.get("GITHUB_OUTPUT"):
|
|
with open(os.environ["GITHUB_OUTPUT"], "a", encoding="utf-8") as fh:
|
|
fh.write(f"package_version={version}\n")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
main()
|
|
except SystemExit as exc:
|
|
print(exc, file=sys.stderr)
|
|
raise
|