Migrate to Beanie v2.0.0 and Replace Motor with PyMongo (Fixes #72) (#82)

* chore: upgrade Beanie to 2.0.0 and switch to pymongo's AsyncMongoClient

* Update main.py

---------

Co-authored-by: Nivedit Jain <nivedit@aikin.club>

Author: bhaveshAswani112

api-server/app/main.py

@@ -6,7 +6,7 @@
 from fastapi import FastAPI
 from contextlib import asynccontextmanager
 from dotenv import load_dotenv
-from motor.motor_asyncio import AsyncIOMotorClient
+from pymongo import AsyncMongoClient
 from fastapi.security import OAuth2PasswordBearer
 
 # injecting singletons
@@ -39,8 +39,13 @@ async def lifespan(app: FastAPI):
     logger.info("server starting")
 
     # initializing beanie
-    client = AsyncIOMotorClient(os.getenv("MONGO_URI"))
-    db = client[os.getenv("MONGO_DATABASE_NAME")]
+    client = AsyncMongoClient(os.getenv("MONGO_URI"))
+    
+    # Get the database name from environment variables.
+    # If not set, use "default" to avoid errors when accessing client[db_name]
+    # os.getenv can return None, which is not valid for dictionary-style access
+    db_name = os.getenv("MONGO_DATABASE_NAME", "exosphere-api-server")
+    db = client[db_name]
     await init_beanie(db, document_models=[User, Project, Satellite])
     logger.info("beanie dbs initialized")
 

api-server/pyproject.toml

@@ -6,7 +6,7 @@ readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
     "bcrypt>=4.3.0",
-    "beanie>=1.30.0",
+    "beanie>=2.0.0",
     "docker-image-py>=0.1.13",
     "email-validator>=2.2.0",
     "fastapi>=0.115.14",

api-server/run.py

@@ -9,7 +9,6 @@
 args = parser.parse_args()
 
 load_dotenv()
-
 def serve():
     mode = args.mode
     if mode == "development":