bump_version.py

import re
import argparse

PYPROJECT_FILES = [
    "humitifier-server/pyproject.toml",
    "humitifier-scanner/pyproject.toml",
    "humitifier-common/pyproject.toml",
]
PYTHON_FILES = [
    "humitifier-server/src/humitifier_server/settings.py",
    "humitifier-scanner/src/humitifier_scanner/constants.py",
]


def update_version_in_django_settings(file_path, new_version):
    # Yes, this is terrible. But it works.

    with open(file_path, "r") as file:
        lines = file.readlines()

    pattern = re.compile(r'^(HUMITIFIER_VERSION = ")(\d+\.\d+\.\d+)(")$')
    updated_lines = []

    for line in lines:
        match = pattern.match(line)
        if match:
            updated_line = f"{match.group(1)}{new_version}{match.group(3)}\n"
            updated_lines.append(updated_line)
        else:
            updated_lines.append(line)

    with open(file_path, "w") as file:
        file.writelines(updated_lines)


def update_version_in_project_toml(file_path, new_version):
    # Yes, this is terrible. But it works.

    with open(file_path, "r") as file:
        lines = file.readlines()

    pattern = re.compile(r'^(version = ")(\d+\.\d+\.\d+)(")$')
    updated_lines = []

    for line in lines:
        match = pattern.match(line)
        if match:
            updated_line = f"{match.group(1)}{new_version}{match.group(3)}\n"
            updated_lines.append(updated_line)
        else:
            updated_lines.append(line)

    with open(file_path, "w") as file:
        file.writelines(updated_lines)


if __name__ == "__main__":

    argparser = argparse.ArgumentParser(description="Bump version")
    argparser.add_argument("new_version", type=str, help="New version")
    args = argparser.parse_args()

    for file in PYPROJECT_FILES:
        update_version_in_project_toml(file, args.new_version)

    for file in PYTHON_FILES:
        update_version_in_django_settings(file, args.new_version)