jeremy
/
simplylab
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

mongodb host port as env

This commit is contained in:
Jeremy Yin 2024-03-15 21:21:42 +08:00
parent 263880d52f
commit 23a83fcc74
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.env.example
View File

@ -1,3 +1,5 @@
MONGO_HOST=
MONGO_PORT=
MONGO_USERNAME= MONGO_USERNAME=
MONGO_PASSWORD= MONGO_PASSWORD=
OPENROUTER_API_KEY= OPENROUTER_API_KEY=

4
simplylab/main.py
View File

@ -34,9 +34,11 @@ app = FastAPI()
@app.on_event("startup") @app.on_event("startup")
def startup_db_client(): def startup_db_client():
mongo_host = os.getenv("MONGO_HOST", "localhost")
mongo_port = os.getenv("MONGO_PORT", "27017")
mongo_username = os.getenv("MONGO_USERNAME") mongo_username = os.getenv("MONGO_USERNAME")
mongo_password = os.getenv("MONGO_PASSWORD") mongo_password = os.getenv("MONGO_PASSWORD")
mongo_uri = f"mongodb://{mongo_username}:{mongo_password}@mongodb:27017/" mongo_uri = f"mongodb://{mongo_username}:{mongo_password}@{mongo_host}:{mongo_port}/"
app.mongodb_client = motor.motor_asyncio.AsyncIOMotorClient(mongo_uri) app.mongodb_client = motor.motor_asyncio.AsyncIOMotorClient(mongo_uri)
app.db = Database(client=app.mongodb_client) app.db = Database(client=app.mongodb_client)
logger.info("Connected to the MongoDB database!") logger.info("Connected to the MongoDB database!")