Add bump version script (#10)

script/bump-version.py

@@ -0,0 +1,95 @@
+#!/usr/bin/env python3
+
+import argparse
+import re
+import subprocess
+from dataclasses import dataclass
+import sys
+
+
+@dataclass
+class Version:
+    major: int
+    minor: int
+    patch: int
+    beta: int = 0
+    dev: bool = False
+
+    def __str__(self):
+        return f'{self.major}.{self.minor}.{self.full_patch}'
+
+    @property
+    def full_patch(self):
+        res = f'{self.patch}'
+        if self.beta > 0:
+            res += f'b{self.beta}'
+        if self.dev:
+            res += '-dev'
+        return res
+
+    @classmethod
+    def parse(cls, value):
+        match = re.match(r'(\d+).(\d+).(\d+)(b\d+)?(-dev)?', value)
+        assert match is not None
+        major = int(match[1])
+        minor = int(match[2])
+        patch = int(match[3])
+        beta = int(match[4][1:]) if match[4] else 0
+        dev = bool(match[5])
+        return Version(
+            major=major, minor=minor, patch=patch,
+            beta=beta, dev=dev
+        )
+
+
+def sub(path, pattern, repl, expected_count=1):
+    with open(path) as fh:
+        content = fh.read()
+    content, count = re.subn(pattern, repl, content, re.MULTILINE)
+    if expected_count is not None:
+        assert count == expected_count
+    with open(path, "wt") as fh:
+        fh.write(content)
+
+
+def write_version(target: str, version: Version):
+    #  version: '1.14.5'  # BETA
+    #  version: '1.14.5'  # STABLE
+    sub(
+        'template/config.yaml',
+        r"  version: '[^']+'  # {}".format(target.upper()),
+        f"  version: '{version}'  # {target.upper()}"
+    )
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('new_version', type=str)
+    parser.add_argument('--beta', action='store_true', help="Update the beta addon")
+    parser.add_argument('--stable', action='store_true', help="Update the stable addon")
+    parser.add_argument('--commit', action='store_true')
+    args = parser.parse_args()
+
+    if not any([args.beta, args.stable]):
+        print("At least one of beta or stable has to be bumped")
+        return 1
+
+    if args.commit and subprocess.call(["git", "diff", "--quiet"]) == 1:
+        print("Cannot use --commit because git is dirty.")
+        return 1
+
+    version = Version.parse(args.new_version)
+    print(f"Bumping to {version}")
+    if args.beta:
+        write_version('beta', version)
+    if args.stable:
+        assert not version.beta
+        write_version('stable', version)
+
+    if args.commit:
+        subprocess.check_call(["git", "commit", "-nam", f"Bump version to v{version}"])
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main() or 0)