From 962a842b440f07a73fe4b0103406430ad4b6f65d Mon Sep 17 00:00:00 2001
From: Yarne Coppens <gitea@yarnecoppens.com>
Date: Thu, 8 Aug 2024 16:50:52 +0200
Subject: [PATCH] Added ability to fully remove database

---
 db/database.db                 | Bin 28672 -> 28672 bytes
 src/config/definitions.py      |   1 +
 src/main.py                    |   1 +
 src/modules/data_connection.py |   2 ++
 src/modules/db_connection.py   |   3 +++
 5 files changed, 7 insertions(+)

diff --git a/db/database.db b/db/database.db
index 2fe4b571c8d141baf276e855e861d4296ceb1952..9b0b041164b060751dde65b11c58f25b3e3342ed 100644
GIT binary patch
delta 269
zcmZp8z}WDBaY7cKI0K(D_chMl?3pafnZ!3X&S&Im<z-_RcXVXzRGloqB{rF#nRD_r
zrh1<8)XemZ5(PIOf5(u`0UTnCljB&9bK0}Ai(6YWc8N~bW1WhrX7d>~CPq$w7Itw%
zL&grz$=BK2&=qZC<Z6*%W*67jXKarL+SrV)0Bn=(<oTSsJY1SBUQF!bnwpFq{*#4x
ms<0?M!&U)R+8o8mE-o+6*v>rJjV~KrsV*Z(+2()zObGzlu}jMU

delta 186
zcmZp8z}WDBaY7c~S_VEF9u6)x_Oq-iOlvm_3Yao+wUn^2i#s|pb`(!mWSY162Ga#b
zPBm6`acgVF&g97(S*8MoI2k9`@K#KIz}CfS#=<UcXvo+hKDn5^4I(vxTO26W$<<uX
z%r364&)BXDG)ZjoZO&Yvyc^?W2Nqo(F3lE0CU$X6O~wvcprOK(KXa$U3|_(M$jB}(
TFVEN(I9Zo38)m56B7p<|urxK^

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