mirror of
https://github.com/Findus23/cr-search.git
synced 2024-09-18 14:13:45 +02:00
26 lines
695 B
Python
26 lines
695 B
Python
from sys import argv
|
|
|
|
from app import db
|
|
from models import Series, 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([Series, Episode, Person, Line, Phrase])
|
|
db.create_tables([Series, Episode, Person, Line, Phrase])
|
|
elif mode == "phrases":
|
|
confirm("Delete all Phrases? ")
|
|
db.drop_tables([Phrase])
|
|
db.create_tables([Phrase])
|
|
|
|
if mode in ["all", "phrases"]:
|
|
db.execute_sql("CREATE INDEX phrases_text_index ON phrase USING gin (text gin_trgm_ops)")
|