Secure mongodb and rabbitmq containers

1 files changed, 8 insertions, 1 deletions

diff --git a/components/database.py b/components/database.py
index 72f63e6..f196fe1 100644
--- a/components/database.py
+++ b/components/database.py
@@ -1,12 +1,19 @@
 import atexit
+from os import getenv
 from typing import Any, Dict
+from dotenv import load_dotenv
 from pymongo import MongoClient
 from pymongo.database import Database
 from pymongo.collection import Collection
 from components.subscriptions.typing import SubsDict
 from components.users.typing import UserDict
 
-client: MongoClient[Any] = MongoClient("mongodb://localhost", tz_aware=True)
+load_dotenv('.env')
+
+client: MongoClient[Any] = MongoClient(
+    "mongodb://%s:%s@localhost/admin" % (getenv('MONGO_USER'), getenv('MONGO_PASS')),
+    tz_aware=True,
+)
 database: Database[Any] = client.get_database("youtube")
 subscriptions: Collection[SubsDict] = database.get_collection("subscriptions")
 users: Collection[UserDict] = database.get_collection("users")