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()