specklepy/patch_version.py

import re
import sys


def patch(tag):
    print(f"Patching version: {tag}")

    with open("pyproject.toml", "r") as f:
        lines = f.readlines()

        if "version" not in lines[2]:
            raise Exception("Invalid pyproject.toml. Could not patch version.")

        lines[2] = f'version = "{tag}"\n'
        with open("pyproject.toml", "w") as file:
            file.writelines(lines)


def main():
    if len(sys.argv) < 2:
        return

    tag = sys.argv[1]
    if not re.match(r"[0-9]+(\.[0-9]+)*$", tag):
        raise ValueError(f"Invalid tag provided: {tag}")

    patch(tag)


if __name__ == "__main__":
    main()