Endpoint to onboard an organization

app/models.py

@@ -7,8 +7,12 @@
 
 
 class Organization(BaseModel):
+    id: PyObjectId = Field(default_factory=PyObjectId, alias="_id")
     org_name: str
 
+    class Config:
+        json_encoders = {ObjectId: str}
+
 
 class Image(BaseModel):
     url: str

app/routers/organizations.py

@@ -1,15 +1,43 @@
 from fastapi import APIRouter, status, HTTPException
 from database import client
 from models import Organization
+from fastapi.responses import JSONResponse
+import string
+import random
 
 router = APIRouter()
 
 
+@router.post("/create-org/{org_name}")
+async def create_organization(org_name: str):
+    if org_name is not None:
+        # create an API key
+        char_set = string.ascii_letters + string.punctuation
+        urand = random.SystemRandom()
+        org_key = "".join([urand.choice(char_set) for _ in range(20)])
+
+        # check if API key exists
+        if (client.quiz.organization.find_one({"org_key": org_key})) is None:
+            org_data = {"org_name": org_name, "org_key": org_key}
+            client.quiz.organization.insert_one(org_data)
+            return JSONResponse(status_code=status.HTTP_201_CREATED, content=org_name)
+
+        raise HTTPException(
+            status_code=500,
+            detail="Duplicate API key. Please try again.",
+        )
+
+
 @router.get("/check-auth-token/{api_key}", response_model=Organization)
 async def check_auth_token(api_key: str):
 
     if (
-        org := client.quiz.organization.find_one({"org_key": api_key}, {"org_name": 1})
+        org := client.quiz.organization.find_one(
+            {"org_key": api_key},
+            {
+                "org_name": 1,
+            },
+        )
     ) is not None:
         return org