From 191168335aaab9f9a963a315fdb791afbdb05c73 Mon Sep 17 00:00:00 2001 From: jableader Date: Sat, 21 Sep 2024 11:22:30 +1000 Subject: [PATCH] Get db name from environment variable --- main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index 4a28064..410ee7b 100644 --- a/main.py +++ b/main.py @@ -10,10 +10,12 @@ from fastapi.encoders import jsonable_encoder app = FastAPI() +import os +DATABASE_PATH = os.environ.get('DOOF_DB', './data/doof.sqlite') # Dependency to create SQLite connection async def get_db(): - sql_db = await db.connect() + sql_db = await db.connect(DATABASE_PATH) try: yield sql_db finally: @@ -354,8 +356,6 @@ async def login(data: LoginBody, conn: sqlite3.Connection = Depends(get_db)) -> async def current_user(user: persons.Person = Depends(cookie_person)) -> persons.Person: return user -import os - if os.environ.get('DOOF_PROD', False): from fastapi.staticfiles import StaticFiles app.mount("/", StaticFiles(directory="./front-dist", html=True), name="front-dist")