Access mongodb with host and port

AlberLC · Jan 20, 2022 · 1844cf9 · 1844cf9
1 parent 34303e9
commit 1844cf9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/multibot/models/database.py b/multibot/models/database.py
@@ -2,6 +2,11 @@
 
 import pymongo
 
-mongo_client = pymongo.MongoClient("localhost", 27017, username=os.environ.get('MONGO_USER'), password=os.environ.get('MONGO_PASSWORD'))
+mongo_client = pymongo.MongoClient(
+    host=os.environ.get('MONGO_HOST'),
+    port=int(port) if (port := os.environ.get('MONGO_PORT')) else None,
+    username=os.environ.get('MONGO_USER'),
+    password=os.environ.get('MONGO_PASSWORD')
+)
 
 db = mongo_client.flanabot