AdvLabDB/advlabdb/scripts/terminal_utils.py

import subprocess
from getpass import getpass


def run(command, **kwargs):
    return subprocess.run(command, shell=True, **kwargs)


def box(message, context=None):
    text_line = "| "

    if context is not None:
        text_line += context + ": "

    text_line += message + " |"

    separator = "=" * len(text_line)

    print()
    print(separator)
    print(text_line)
    print(separator)
    print()


def step(message):
    continue_message = "-> Press ENTER to continue or Ctrl+C to interrupt the script <-"
    upper_separator = "_" * len(continue_message)

    print()
    print(upper_separator)

    box(message, "Next step")

    print(continue_message)
    getpass("")
    print()


def spaced_hl():
    print("\n\n" + "_" * 20 + "\n\n")


def validating_input(
    prompt,
    options=None,
    format_function=lambda ans: ans,
    check_constraints_function=lambda ans: True,
):
    if options is not None:
        prompt += " ["

        for opt in options[:-1]:
            prompt += opt + "/"

        prompt += options[-1] + "]: "

        lowered_options = [opt.lower() for opt in options]

        def adj_check_constraints_function(ans):
            return ans.lower() in lowered_options and check_constraints_function(ans)

    else:
        prompt += ": "
        adj_check_constraints_function = check_constraints_function

    ans = None
    first_run = True

    while ans is None or not adj_check_constraints_function(ans):
        if not first_run:
            print("Invalid input!\n")
        else:
            first_run = False

        ans = input(prompt)
        try:
            formatted_ans = format_function(ans)
        except Exception as ex:
            ans = None

    return formatted_ans


def confirm(prompt):
    ans = validating_input(
        prompt,
        options=("y", "n"),
        format_function=lambda ans: ans.lower(),
    )

    if ans == "y":
        return True
    else:
        return False