Update database.py

database.py

@@ -1,46 +1,43 @@
-import motor.motor_asyncio  
-import os
-import logging
-from dotenv import load_dotenv
-
-load_dotenv()
-
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-MONGO_URI = os.getenv("MONGO_URI")
-if not MONGO_URI:
-    raise ValueError("MONGO_URI is not set")
-
-logger.info("Connecting to MongoDB")
-
-client = None
-db = None
-users_collection = None
-
-async def get_database():
-    global client, db, users_collection
-
-    if client is None:  
-        try:
-            logger.info("Connecting to MongoDB")
-            client = motor.motor_asyncio.AsyncIOMotorClient(MONGO_URI)
-            db = client["fastapi_db"]
-            users_collection = db["users"]
-            logger.info("Successfully connected to MongoDB")
-        except Exception as e:
-            logger.error(f"MongoDB connection failed: {e}")
-            raise
-
-    return db
-
-async def get_users_collection():
-    await get_database()
-    return users_collection
-
-async def close_database():
-    global client
-    if client:
-        logger.info("Closing MongoDB connection")
-        client.close()
-        client = None
+import motor.motor_asyncio  
+import os
+import logging
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+MONGO_URI = os.getenv("MONGO_URI")
+if not MONGO_URI:
+    raise ValueError("MONGO_URI is not set")
+
+logger.info("Connecting to MongoDB")
+
+client = None
+db = None
+users_collection = None
+
+async def get_database():
+    global client, db, users_collection
+
+    if client is None:  
+        try:
+            logger.info("Connecting to MongoDB")
+            client = motor.motor_asyncio.AsyncIOMotorClient(MONGO_URI)
+            db = client["fastapi_db"]
+            users_collection = db["users"]
+            logger.info("Successfully connected to MongoDB")
+        except Exception as e:
+            logger.error(f"MongoDB connection failed: {e}")
+            raise
+
+    return db
+
+async def get_users_collection():
+    await get_database()
+    return users_collection
+
+async def close_database():
+    global client
+    if client:
+        logger.info("Closing MongoDB connection")
+        client.close()
+        client = None