Fix bug with create_index for schema_collections

Added connect_to_mongo and close_mongo to database.py

Added MONGO_URI to settings

Author: JosePizarro3

jsoned/database.py

@@ -1,16 +1,23 @@
-# from pymongo import MongoClient
+from motor.motor_asyncio import AsyncIOMotorClient
 
-# client = MongoClient("mongodb://localhost:27017/")
-# db = client["jsoned_db"]
-# schemas_collection = db["schemas"]
+from jsoned.settings import settings
 
+client: AsyncIOMotorClient | None = None
+database = None
+schemas_collection = None
 
-from motor.motor_asyncio import AsyncIOMotorClient
 
-# MongoDB Driver
-client = AsyncIOMotorClient("mongodb://localhost:27017")
-database = client.jsoned_db
-schemas_collection = database.schemas
+async def connect_to_mongo():
+    global client, database, schemas_collection
+
+    client = AsyncIOMotorClient(settings.MONGO_URI)
+    database = client.jsoned_db
+    schemas_collection = database.schemas
+
+    # create index once on startup
+    await schemas_collection.create_index("title", unique=True)
+
 
-# Ensure that the `title` field is unique
-schemas_collection.create_index([("title", 1)], unique=True)
+async def close_mongo():
+    if client is not None:
+        client.close()

jsoned/main.py

@@ -4,14 +4,19 @@
 from starlette.middleware.cors import CORSMiddleware
 
 from jsoned.api.api_v1.api import api_router
+from jsoned.database import close_mongo, connect_to_mongo
 from jsoned.settings import settings
 
 
 @asynccontextmanager
 async def app_init(app: FastAPI):
+    await connect_to_mongo()
+
     app.include_router(api_router, prefix=settings.API_V1_STR)
     yield
 
+    await close_mongo()
+
 
 app = FastAPI(
     title=settings.PROJECT_NAME,
@@ -22,7 +27,7 @@ async def app_init(app: FastAPI):
 if settings.BACKEND_CORS_ORIGINS:
     app.add_middleware(
         CORSMiddleware,
-        allow_origins=settings.BACKEND_CORS_ORIGINS,
+        allow_origins=[str(o) for o in settings.BACKEND_CORS_ORIGINS],
         allow_credentials=True,
         allow_methods=["*"],
         allow_headers=["*"],

jsoned/settings.py

@@ -9,6 +9,7 @@ class Settings(BaseSettings):
         "http://localhost",
         "http://localhost:3000",
     ]
+    MONGO_URI: AnyHttpUrl = "mongodb://localhost:27017"
 
 
 settings = Settings()