2020-03-08 18:48:14 +01:00
|
|
|
from sys import argv
|
2020-03-07 10:45:39 +01:00
|
|
|
|
2020-03-08 18:48:14 +01:00
|
|
|
from models import db, Phrase, Episode, Person, Line
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def confirm(message: str) -> None:
|
|
|
|
|
if "y" not in input(message):
|
|
|
|
|
raise ValueError("abort")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if len(argv) < 2:
|
|
|
|
|
raise ValueError("select mode")
|
|
|
|
|
mode = argv[1]
|
|
|
|
|
|
|
|
|
|
if mode == "all":
|
|
|
|
|
confirm("Delete all Data? ")
|
|
|
|
|
db.drop_tables([Episode, Person, Line, Phrase])
|
|
|
|
|
db.create_tables([Episode, Person, Line, Phrase])
|
|
|
|
|
elif mode == "phrases":
|
|
|
|
|
confirm("Delete all Phrases? ")
|
|
|
|
|
db.drop_tables([Phrase])
|
|
|
|
|
db.create_tables([Phrase])
|
