update.py

#!/usr/bin/env python3

import os
import subprocess
import sys


def y_n(q):
    while True:
        ri = input("{} (y/n): ".format(q))
        if ri.lower() in ["yes", "y"]:
            return True
        elif ri.lower() in ["no", "n"]:
            return False


def update_deps():
    print("Attempting to update dependencies...")

    try:
        subprocess.check_call(
            '"{}" -m pip install --no-warn-script-location --user -U -r requirements.txt'.format(
                sys.executable
            ),
            shell=True,
        )
    except subprocess.CalledProcessError:
        raise OSError(
            "Could not update dependencies. You will need to run '\"{0}\" -m pip install -U -r requirements.txt' yourself.".format(
                sys.executable
            )
        )


def finalize():
    try:
        from musicbot.constants import VERSION

        print("The current MusicBot version is {0}.".format(VERSION))
    except Exception:
        print(
            "There was a problem fetching your current bot version. The installation may not have completed correctly."
        )

    print("Done!")


def main():
    print("Starting...")

    # Make sure that we're in a Git repository
    if not os.path.isdir(".git"):
        raise EnvironmentError("This isn't a Git repository.")

    # Make sure that we can actually use Git on the command line
    # because some people install Git Bash without allowing access to Windows CMD
    try:
        subprocess.check_call("git --version", shell=True, stdout=subprocess.DEVNULL)
    except subprocess.CalledProcessError:
        raise EnvironmentError(
            "Couldn't use Git on the CLI. You will need to run 'git pull' yourself."
        )

    print("Passed Git checks...")

    # Check that the current working directory is clean
    sp = subprocess.check_output(
        "git status --porcelain", shell=True, universal_newlines=True
    )
    if sp:
        oshit = y_n(
            "You have modified files that are tracked by Git (e.g the bot's source files).\n"
            "Should we try resetting the repo? You will lose local modifications."
        )
        if oshit:
            try:
                subprocess.check_call("git reset --hard", shell=True)
            except subprocess.CalledProcessError:
                raise OSError("Could not reset the directory to a clean state.")
        else:
            wowee = y_n(
                "OK, skipping bot update. Do you still want to update dependencies?"
            )
            if wowee:
                update_deps()
            else:
                finalize()
            return

    print("Checking if we need to update the bot...")

    try:
        subprocess.check_call("git pull", shell=True)
    except subprocess.CalledProcessError:
        raise OSError(
            "Could not update the bot. You will need to run 'git pull' yourself."
        )

    update_deps()
    finalize()


if __name__ == "__main__":
    main()