Haldis/haldis
2
0
Fork 0
Code Issues 39 Pull requests Packages Projects Releases Wiki Activity

Ignore case in database creation script

Browse source
This commit is contained in:
midgard 2019-10-01 20:58:24 +02:00 committed by Midgard
parent f41817a5dd
commit 39f30b55c7
Signed by: midgard
GPG key ID: 511C112F1331BBB4
1 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
app/database/create_database.py
View file

@ -15,8 +15,8 @@ entry_sets = {
"Fitchen": add_fitchen.add, "Fitchen": add_fitchen.add,
} }
yes = ["yes", "y", "Y"] yes = ["yes", "y"]
no = ["no", "n", "N"] no = ["no", "n"]
# Commit all the things # Commit all the things
@ -27,7 +27,7 @@ def commit() -> None:
def check_if_overwrite() -> bool: def check_if_overwrite() -> bool:
answer = input("Do you want to overwrite the previous database? (y/N) ") answer = input("Do you want to overwrite the previous database? (y/N) ")
return answer in yes return answer.lower() in yes
def add_all() -> None: def add_all() -> None:
@ -39,11 +39,13 @@ def add_all() -> None:
def recreate_from_scratch() -> None: def recreate_from_scratch() -> None:
confirmation = "Are you very very sure? (Will delete previous entries!) (y/N) " confirmation = "Are you very very sure? (Will delete previous entries!) (y/N) "
check = "I acknowledge any repercussions!" check = "I acknowledge any repercussions!"
if input(confirmation) in yes and input("Type: '{}' ".format(check)) == check: if input(confirmation).lower() in yes and input("Type: '{}' ".format(check)).lower() == check:
print("Resetting the database!") print("Resetting the database!")
db.drop_all() db.drop_all()
db.create_all() db.create_all()
add_to_current() add_to_current()
else:
print("You cancelled.")
def add_to_current() -> None: def add_to_current() -> None:
@ -54,15 +56,15 @@ def add_to_current() -> None:
["{}({}), ".format(loc, i) for i, loc in enumerate(available)] ["{}({}), ".format(loc, i) for i, loc in enumerate(available)]
).rstrip(", ") ).rstrip(", ")
while input("Do you still want to add something? (Y/n) ") not in no: while input("Do you still want to add something? (Y/n) ").lower() not in no:
print( print(
"What do you want to add? (Use numbers, or A for all, or C for cancel) " "What do you want to add? (Use numbers, or A for all, or C for cancel) "
) )
answer = input("Available: {} : ".format(add_numbers())) answer = input("Available: {} : ".format(add_numbers()))
if answer == "A": if answer.lower() == "a":
add_all() add_all()
available = [] available = []
elif answer == "C": elif answer.lower() == "c":
pass pass
elif answer.isnumeric() and answer in [str(x) for x in range(len(available))]: elif answer.isnumeric() and answer in [str(x) for x in range(len(available))]:
answer_index = int(answer) answer_index = int(answer)