vb.py

import os
import re
import subprocess
from pathlib import Path

__author__ = "Elad Yaniv"

import click

VERSION_REGEX = r"version=\"(\d*[.]\d*[.]\d*)"


@click.group(help="Version Bump CLI")
def vb():
    pass


def update_readme_to_sphinx():
    import re

    regex = r"([#]\s)([A-Z]\w+)(\s?\n)"
    subst = "\\2\\n---------------\\n"
    with open("README.md", "r") as f:
        r_parsed = f.read()
    title = "# QuickWalkThrough\n============"
    r_parsed = r_parsed.replace("# SC2MapAnalysis", title)
    r_result = re.sub(regex, subst, r_parsed, 0, re.MULTILINE)
    with open("README.md", "w") as f:
        f.write(r_result)


def parse_setup():
    with open("setup.py", "r") as f:
        setup_parsed = f.read()
    return setup_parsed


@vb.command(help="sphinx make for gh pages")
def makedocs():
    p = Path()
    path = p.joinpath("docs").absolute()
    click.echo(click.style(f"calling {path}//make github", fg="green"))
    subprocess.check_call(f"{path}//make github", shell=True)


@vb.command(help="print setup.py")
def printsetup():
    setup_parsed = parse_setup()
    click.echo(click.style(setup_parsed, fg="blue"))


@vb.command(help="MonkeyType apply on list-modules")
@click.option("--apply/--no-apply", default=False)
def mt(apply):
    click.echo(click.style("This could take a few seconds", fg="blue"))
    encoded_modules = subprocess.check_output("monkeytype list-modules", shell=True)
    list_modules = encoded_modules.decode().split("\r\n")
    to_exclude = {"mocksetup"}
    if apply:
        for m in list_modules:
            if [x for x in to_exclude if x in m] == []:
                click.echo(click.style(f"Applying on {m}", fg="green"))
                subprocess.check_call(f"monkeytype apply {m}", shell=True)


@vb.command(help="Get current version")
def gv():
    click.echo("Running git describe")
    subprocess.check_call("git describe")


@vb.command(help="Bump Minor")
def bumpminor():
    setup_parsed = parse_setup()
    old_version_regex = VERSION_REGEX
    old_version = re.findall(old_version_regex, setup_parsed)[0]
    minor = re.findall(r"([.]\d*)", old_version)[-1]
    minor = minor.replace(".", "")
    click.echo("Current Version: " + click.style(old_version, fg="green"))
    click.echo("Minor Found: " + click.style(minor, fg="green"))
    bump = str(int(minor) + 1)
    click.echo("Bumping to : " + click.style(bump, fg="blue"))
    new_version = str(old_version).replace(minor, bump)
    click.echo("Updated Version: " + click.style(new_version, fg="red"))
    b_minor(new_version)


def b_minor(new_version):
    setup_parsed = parse_setup()
    old_version_regex = VERSION_REGEX
    old_version = re.findall(old_version_regex, setup_parsed)[0]
    setup_updated = setup_parsed.replace(old_version, new_version)
    with open("setup.py", "w") as f:
        f.write(setup_updated)

    curdir = os.getcwd()
    click.echo(click.style(curdir + "\\standard-version", fg="blue"))
    subprocess.check_call("git fetch", shell=True)
    subprocess.check_call("git pull", shell=True)
    subprocess.check_call("git add setup.py", shell=True)
    subprocess.check_call(f'git commit -m " setup bump {new_version} " ', shell=True)
    subprocess.check_call(f"standard-version --release-as {new_version}", shell=True)
    # subprocess.check_call('git push --follow-tags origin', shell=True)


@vb.command(help="Custom git log command for last N days")
@click.argument("days")
def gh(days):
    click.echo(
        click.style("Showing last ", fg="blue")
        + click.style(days, fg="green")
        + click.style(" days summary", fg="blue")
    )
    subprocess.check_call("git fetch", shell=True)
    subprocess.check_call(
        f"git log --oneline --decorate --graph --all -{days}", shell=True
    )


if __name__ == "__main__":
    vb(prog_name="python -m vb")