feat: use env variables to load db

This commit is contained in:
swve 2023-04-12 22:24:17 +02:00
parent ef5fcd5870
commit 8cc6111aaf
4 changed files with 22 additions and 6 deletions

View file

@ -2,17 +2,20 @@ import logging
from fastapi import FastAPI
import motor.motor_asyncio
async def connect_to_db(app: FastAPI) :
async def connect_to_db(app: FastAPI):
logging.info("Connecting to database...")
try:
app.mongodb_client = motor.motor_asyncio.AsyncIOMotorClient("mongodb://learnhouse:learnhouse@mongo:27017/") # type: ignore
app.db = app.mongodb_client["learnhouse"] # type: ignore
app.mongodb_client = motor.motor_asyncio.AsyncIOMotorClient( # type: ignore
app.learnhouse_config.database_config.mongodb_connection_string) # type: ignore
app.db = app.mongodb_client["learnhouse"] # type: ignore
logging.info("Connected to database!")
except Exception as e:
logging.error("Failed to connect to database!")
logging.error(e)
async def close_database(app: FastAPI):
app.mongodb_client.close() # type: ignore
app.mongodb_client.close() # type: ignore
logging.info("LearnHouse has been shut down.")
return app
return app