diff --git a/db/database.db b/db/database.db index 2fe4b57..9b0b041 100644 Binary files a/db/database.db and b/db/database.db differ diff --git a/src/config/definitions.py b/src/config/definitions.py index becca27..3068397 100644 --- a/src/config/definitions.py +++ b/src/config/definitions.py @@ -2,6 +2,7 @@ import os ROOT_PATH = project_root = os.path.dirname(os.path.dirname(os.path.dirname(__file__))) SECRETS_FILE_PATH = ROOT_PATH + '/secrets/auth.yaml' +DATABASE_FILE_PATH = ROOT_PATH + '/db/database.db' DATABASE_FILE_PROJECT_PATH = f"/db/database.db" SQLITE_URL = f"sqlite://{DATABASE_FILE_PROJECT_PATH}" diff --git a/src/main.py b/src/main.py index 18aed15..5cbd483 100644 --- a/src/main.py +++ b/src/main.py @@ -7,6 +7,7 @@ from src.modules import data_connection @asynccontextmanager async def lifespan(app: FastAPI): # Startup + data_connection.delete_database() data_connection.create_db_and_tables() yield # Shutdown diff --git a/src/modules/data_connection.py b/src/modules/data_connection.py index 4b2a700..9b17826 100644 --- a/src/modules/data_connection.py +++ b/src/modules/data_connection.py @@ -21,6 +21,8 @@ def get_user_wishlist_collection() -> list[boardgame_classes.BoardGame]: def get_plays() -> list[play_classes.Play]: return bgg_connection.get_plays() +def delete_database(): + db_connection.delete_database() def create_db_and_tables(): db_connection.create_db_and_tables() \ No newline at end of file diff --git a/src/modules/db_connection.py b/src/modules/db_connection.py index a99a90b..64fe11f 100644 --- a/src/modules/db_connection.py +++ b/src/modules/db_connection.py @@ -1,5 +1,6 @@ from sqlmodel import create_engine, SQLModel from src.config import definitions +import os sqlite_url = definitions.SQLITE_URL @@ -7,6 +8,8 @@ sqlite_url = definitions.SQLITE_URL connect_args = {"check_same_thread": False} engine = create_engine(sqlite_url, echo=True, connect_args=connect_args) +def delete_database(): + os.remove(definitions.DATABASE_FILE_PATH) def create_db_and_tables(): SQLModel.metadata.create_all(engine) \ No newline at end of file