haldis/app/database/create_database.py

import add_admins
import add_fitchen
import add_oceans_garden
import add_primadonna
import add_simpizza
import add_testlocation
from app import db, create_app

entry_sets = {
    "Admins": add_admins.add,
    "Testlocation": add_testlocation.add,
    "Ocean's Garden": add_oceans_garden.add,
    "SimPizza": add_simpizza.add,
    "Primadonna": add_primadonna.add,
    "Fitchen": add_fitchen.add,
}

yes = ["yes", "y"]
no = ["no", "n"]


# Commit all the things
def commit() -> None:
    db.session.commit()
    print("Committing successful")


def check_if_overwrite() -> bool:
    answer = input("Do you want to overwrite the previous database? (y/N) ")
    return answer.lower() in yes


def add_all() -> None:
    for entry_set in entry_sets.keys():
        print("Adding {}.".format(entry_set))
        entry_sets[entry_set]()


def recreate_from_scratch() -> None:
    confirmation = "Are you very very sure? (Will delete previous entries!) (y/N) "
    if input(confirmation) in yes:
        print("Resetting the database!")
        db.drop_all()
        db.create_all()
        add_to_current()
    else:
        print("You cancelled.")


def add_to_current() -> None:
    available = [entry_set for entry_set in entry_sets]

    def add_numbers() -> str:
        return "  ".join(
            ["{}({}), ".format(loc, i) for i, loc in enumerate(available)]
        ).rstrip(", ")

    while input("Do you still want to add something? (Y/n) ").lower() not in no:
        print(
            "What do you want to add? (Use numbers, or A for all, or C for cancel)   "
        )
        answer = input("Available: {}  : ".format(add_numbers()))
        if answer.lower() == "a":
            add_all()
            available = []
        elif answer.lower() == "c":
            pass
        elif answer.isnumeric() and answer in [str(x) for x in range(len(available))]:
            answer_index = int(answer)
            print("Adding {}.".format(available[answer_index]))
            entry_sets[str(available[answer_index])]()
            del available[answer_index]
        else:
            print("Not a valid answer.")
    print("Thank you for adding, come again!")

manager = create_app()

@manager.command
def setup_database():
    print("Database modification script!")
    print("=============================\n\n")
    if check_if_overwrite():
        recreate_from_scratch()
    else:
        add_to_current()
    commit()


manager.run()