setup_version.py

# change version:
# 	readme
# 	setup
# 	latest_version.txt
# 	version file

from copy import deepcopy

setup_file = "setup.py"
version_file = "phigaro/_version.py"
readme_file = "README.md"
latest_version_file = "test_data_gitaction/latest_version.txt"
tag_file = "tag_name"

with open(setup_file, "r") as f:
    tmp_version = f.readlines()[12].strip()
print("Temporary %s" % tmp_version)

tmp_version = tmp_version.replace("version=", "").replace('"', "").replace(",", "")
tmp_version = tmp_version.split(".")
version_variants = []
for i in list(range(len(tmp_version)))[::-1]:
    tmp_version_ = deepcopy(tmp_version)
    tmp_version_[i] = str(int(tmp_version_[i]) + 1)
    version_variants.append(".".join(tmp_version_))
    tmp_version[i] = "0"


def ask_version(version_variants):
    def confirm(version, version_variants):
        print("Do you agree? Yes/No")
        user_answer = input()
        if user_answer.upper() in {"Y", "YES"}:
            return version
        elif user_answer.upper() in {"N", "NO"}:
            return ask_version(version_variants)
        else:
            confirm(version, version_variants)

    for i, version in enumerate(version_variants):
        print("[%d] %s" % (i, version))
    print("[%d] Type your version" % (len(version_variants)))
    user_answer = input()
    try:
        version = version_variants[int(user_answer)]
    except:
        version = user_answer
    print()
    print("You choose version: %s" % version)
    return confirm(version, version_variants)


version = ask_version(version_variants)

for file_ in [setup_file, version_file]:
    with open(file_, "r") as f:
        lines = f.readlines()
    lines_new = []
    for line in lines:
        if ('version="' in line) or ("__version__" in line):
            line = "=".join(line.split("=")[:-1] + ['"%s",' % version])
            if "__version__" in line:
                line = line[:-1]
            line = line + "\n"
        lines_new.append(line)
    with open(file_, "w") as f:
        f.write("".join(lines_new))

with open(readme_file, "r") as f:
    lines = f.readlines()
lines[0] = "# Phigaro v%s\n" % version
with open(readme_file, "w") as f:
    f.write("".join(lines))

with open(latest_version_file, "w") as f:
    f.write(
        "https://github.com/bobeobibo/phigaro/raw/master/dist/phigaro-%s.tar.gz"
        % version
    )


with open(tag_file, "w") as f:
    f.write("v%s" % version)