diff --git a/README.md b/README.md index 55574dd..5ace1c5 100644 --- a/README.md +++ b/README.md @@ -47,3 +47,4 @@ Käynnistä ohjelma: python src/index.py ``` + diff --git a/src/db_helper.py b/src/db_helper.py index 1d951e5..e3aa9b2 100644 --- a/src/db_helper.py +++ b/src/db_helper.py @@ -20,17 +20,26 @@ def table_exists(name): result = db.session.execute(sql_table_existence) return result.fetchall()[0][0] + + + def reset_db(): clear_table(CITATIONS_TABLE) clear_table(CITATION_FIELDS_TABLE) clear_table(TAGS_TABLE) + + + def clear_table(table_name: str): print(f"Clearing contents from table {table_name}") sql = text(f"DELETE FROM {table_name}") db.session.execute(sql) db.session.commit() + + + def setup_db(): drop_table_if_exists(CITATIONS_TABLE) drop_table_if_exists(CITATION_FIELDS_TABLE) @@ -68,6 +77,9 @@ def setup_db(): db.session.execute(sql) db.session.commit() + + + def drop_table_if_exists(table_name: str): if table_exists(table_name): print(f"Table {table_name} exists, dropping") @@ -75,6 +87,9 @@ def drop_table_if_exists(table_name: str): db.session.execute(sql) db.session.commit() + + + if __name__ == "__main__": with app.app_context(): setup_db()