Get db name from environment variable

This commit is contained in:
jableader 2024-09-21 11:22:30 +10:00
parent b1905cc03a
commit 191168335a

View file

@ -10,10 +10,12 @@ from fastapi.encoders import jsonable_encoder
app = FastAPI() app = FastAPI()
import os
DATABASE_PATH = os.environ.get('DOOF_DB', './data/doof.sqlite')
# Dependency to create SQLite connection # Dependency to create SQLite connection
async def get_db(): async def get_db():
sql_db = await db.connect() sql_db = await db.connect(DATABASE_PATH)
try: try:
yield sql_db yield sql_db
finally: 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: async def current_user(user: persons.Person = Depends(cookie_person)) -> persons.Person:
return user return user
import os
if os.environ.get('DOOF_PROD', False): if os.environ.get('DOOF_PROD', False):
from fastapi.staticfiles import StaticFiles from fastapi.staticfiles import StaticFiles
app.mount("/", StaticFiles(directory="./front-dist", html=True), name="front-dist") app.mount("/", StaticFiles(directory="./front-dist", html=True), name="front-dist")