diff --git a/app/db_maker.py b/app/db_maker.py index 425986e..4ea114e 100644 --- a/app/db_maker.py +++ b/app/db_maker.py @@ -13,7 +13,22 @@ Update : from peewee import * from tqdm import tqdm -from models.princes_db_tables import db, Institution, State, Place, Diplo_type, Document, Acte +from modeles.princes_db_tables import db, Institution, State, Place, Diplo_type, Document, Acte +from data.institution_data import institution +def _create_institution(data_lst: list)-> None: + for data in tqdm(data_lst, desc="Populating Institution..."): + Institution.create(**data) + + +def init(): + db.connect() + print("Dropping existing DB...") + db.drop_tables([Institution, State, Place, Diplo_type, Document, Acte]) + print("Re-creating schema...") + db.create_tables([Institution, State, Place, Diplo_type, Document, Acte]) + _create_institution(institution) + +init()