From d3f1d5b8327afd41fb7501e601ab898597d7d985 Mon Sep 17 00:00:00 2001 From: jgenero Date: Tue, 11 Oct 2022 17:55:36 +0200 Subject: [PATCH] create intution table + init function --- app/db_maker.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) 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()