tasks.py

from pathlib import Path
import re
import sys
from typing import Tuple

from invoke import task


TOP_DIR = Path(__file__).parent.resolve()


def update_file(filename: str, sub_line: Tuple[str, str], strip: str = None):
    """Utility function for tasks to read, update, and write files"""
    with open(filename, "r") as handle:
        lines = [
            re.sub(sub_line[0], sub_line[1], line.rstrip(strip)) for line in handle
        ]

    with open(filename, "w") as handle:
        handle.write("\n".join(lines))
        handle.write("\n")


@task
def update_version(_, version=""):
    """Update package version using SemVer syntax."""
    if version:
        if version.startswith("v"):
            version = version[1:]
        if re.match(r"[0-9]+(\.[0-9]+){2}.*", version) is None:
            sys.exit(
                f"Error: Passed version ({version}) does to match the SemVer format: "
                "Major.Minor.Patch(+ extras)."
            )
    else:
        sys.exit(
            "Please pass --version with a value that complies to the SemVer format: "
            "Major.Minor.Patch(+ extras)."
        )

    update_file(
            TOP_DIR.joinpath("package.json"), (r'"version":.*,', f'"version": "{version}",')
    )

    print(f"Bumped version to {version} !")