haldis/app/database/create_database.py
2015-05-13 13:34:32 +02:00

71 lines
2 KiB
Python

from app import db
import add_oceans_garden, add_admins, add_simpizza
entry_sets = {"Admins": add_admins.add, "Ocean's Garden": add_oceans_garden.add, "SimPizza": add_simpizza.add}
yes = ["yes", "y", "Y"]
no = ["no", "n", "N"]
# Commit all the things
def commit():
db.session.commit()
print("Committing successful")
def check_if_overwrite():
answer = input("Do you want to overwrite the previous database? (y/N) ")
return answer in yes
def add_all():
for entry_set in entry_sets.keys():
print("Adding {}.".format(entry_set))
entry_sets[entry_set]()
def recreate_from_scratch():
confirmation = "Are you very very sure? (Will delete previous entry's!) (y/N) "
check = "I acknowledge any repercussions!"
if input(confirmation) in yes and input("Type: '{}' ".format(check)) == check:
print("Resetting the database!")
db.drop_all()
db.create_all()
add_to_current()
def add_to_current():
available = [entry_set for entry_set in entry_sets]
def add_numbers():
return " ".join(["{}({}), ".format(loc, i) for i, loc in enumerate(available)]).rstrip(", ")
while input("Do you still want to add something? (Y/n) ") not in no:
print("What do you want to add? (Use numbers, or A for all, or C for chancel) ")
answer = input("Available: {} : ".format(add_numbers()))
if answer == "A":
add_all()
available = []
elif answer == "C":
pass
elif answer in [str(x) for x in range(len(available))]:
answer = int(answer)
print("Adding {}.".format(available[answer]))
entry_sets[str(available[answer])]()
del available[answer]
else:
print("Not a valid answer.")
print("Thank you for adding, come again!")
def init():
print('Database modification script!')
print('=============================\n\n')
if check_if_overwrite():
recreate_from_scratch()
else:
add_to_current()
commit()
init()