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(f"Current Version: " + click.style(old_version, fg='green'))
    click.echo(f"Minor Found: " + click.style(minor, fg='green'))
    bump = str(int(minor) + 1)
    click.echo(f"Bumping to : " + click.style(bump, fg='blue'))
    new_version = str(old_version).replace(minor, bump)
    click.echo(f"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')