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
|
#!/usr/bin/env python3
import argparse
import subprocess
import sys
from collections.abc import Iterable
from pathlib import Path
from shutil import which
class UsageError(Exception):
pass
def parse_version() -> str:
output = subprocess.check_output(["poetry", "version", "--short"], encoding="utf-8")
version = output.strip()
return version
def bump_version(path: Path, prefix_list: Iterable[str], current_version: str, new_version: str):
prefixes = tuple(prefix_list)
lines = []
for line in path.open():
if line.startswith(prefixes):
line = line.replace(current_version, new_version)
lines.append(line)
path.write_text("".join(lines))
def git(*args):
return subprocess.check_call(["git"] + list(args))
def make(*args):
return subprocess.check_call(["make"] + list(args))
def check_changelog():
old_changelog = Path("CHANGES.rst").read_text()
make("changes")
new_changelog = Path("CHANGES.rst").read_text()
if old_changelog == new_changelog:
raise UsageError("No new changelog entries")
def check_environment():
for binary in ("git", "make", "poetry"):
if not which(binary):
raise UsageError("No such binary: {}".format(binary))
def main() -> None:
parser = argparse.ArgumentParser()
parser.add_argument("new_version", help="Version to release")
args = parser.parse_args()
new_version: str = args.new_version
current_version = parse_version()
if current_version is None:
raise UsageError("Unable to parse version")
print(f"Current version: {current_version}")
if current_version == new_version:
raise UsageError("Current version is the same as new version")
check_changelog()
bump_version(Path("doc/conf.py"), ["version"], current_version, new_version)
subprocess.check_call(["poetry", "version", new_version])
git("add", "pyproject.toml", "doc/conf.py")
git("commit", "-m", "Version bump to {}".format(new_version))
git("tag", new_version)
make("changes")
git("add", "CHANGES.rst")
git("commit", "-m", "CHANGES.rst: add release notes for {}".format(new_version))
git("tag", "-f", new_version)
if __name__ == "__main__":
try:
main()
except UsageError as err:
print(f"ERROR: {err}", file=sys.stderr)
sys.exit(1)
|