_render_update_version.py

# NOTE: Is called by the build script
import os
import subprocess


def git_rev(short=True):
    """Returns the git revision."""
    cmd = ["git", "rev-parse"]
    if short:
        cmd.append("--short")
    cmd.append("HEAD")
    return subprocess.check_output(cmd).decode("utf-8").strip()


def git_branch_rev_count(branch):
    """Count the number of revisions on branch 'branch'."""
    return int(
        subprocess.check_output(["git", "rev-list", branch, "--count"])
        .decode("utf-8")
        .strip()
    )


def git_current_branch():
    """Return the name of the current branch."""
    return (
        subprocess.check_output(["git", "rev-parse", "--abbrev-ref", "HEAD"])
        .decode("utf-8")
        .strip()
    )


def replace_version_placeholders(file_content: str) -> str:
    return file_content.format(
        HIP_PYTHON_VERSION_SHORT=git_branch_rev_count(git_current_branch()),
        HIP_PYTHON_VERSION=git_branch_rev_count(git_current_branch()),
        HIP_PYTHON_BRANCH=git_current_branch(),
        HIP_PYTHON_REV=git_rev(),
    )


# render read _version.py (requires git)
def render_version_py(parent_dir: str):
    with open(os.path.join(parent_dir, "_version.py.in"), "r") as infile, open(
        os.path.join(parent_dir, "_version.py"), "w"
    ) as outfile:
        rendered: str = replace_version_placeholders(infile.read())
        outfile.write(rendered)


def render_hip_python_as_cuda_requirements_txt():
    reqirements_file: str = os.path.join(
        "hip-python-as-cuda", "requirements.txt"
    )
    with open(reqirements_file + ".in", "r") as infile, open(
        reqirements_file, "w"
    ) as outfile:
        rendered: str = replace_version_placeholders(infile.read())
        outfile.write(rendered)


if __name__ == "__main__":
    render_version_py(
        os.path.join("hip-python", "hip"),
    )
    render_version_py(os.path.join("hip-python-as-cuda", "cuda"))
    render_hip_python_as_cuda_requirements_txt()