Endpoint to onboard an organization

app/models.py

@@ -7,7 +7,24 @@
 
 
 class Organization(BaseModel):
-    org_name: str
+    id: PyObjectId = Field(default_factory=PyObjectId, alias="_id")
+    name: str
+
+    class Config:
+        allow_population_by_field_name = True
+        arbitrary_types_allowed = True
+        json_encoders = {ObjectId: str}
+        schema_extra = {"example": {"name": "Avanti Fellows"}}
+
+
+class OrganizationResponse(Organization):
+    name: str
+
+    class Config:
+        allow_population_by_field_name = True
+        arbitrary_types_allowed = True
+        json_encoders = {ObjectId: str}
+        schema_extra = {"example": {"name": "Avanti Fellows"}}
 
 
 class Image(BaseModel):

app/routers/organizations.py

@@ -1,15 +1,57 @@
 from fastapi import APIRouter, status, HTTPException
 from database import client
-from models import Organization
+from models import Organization, OrganizationResponse
+import secrets
+import string
+from fastapi.encoders import jsonable_encoder
 
-router = APIRouter()
+router = APIRouter(prefix="/organizations", tags=["Organizations"])
 
 
-@router.get("/check-auth-token/{api_key}", response_model=Organization)
+@router.post("/", response_model=OrganizationResponse)
+async def create_organization(org: Organization):
+    org = jsonable_encoder(org)
+    if org["name"] is not None:
+
+        # create an API key
+        key = "".join(
+            [secrets.choice(string.ascii_letters + string.digits) for _ in range(20)]
+        )
+        number_of_loops = 3
+
+        while number_of_loops > 0:
+
+            # check if API key exists
+            if (client.quiz.organization.find_one({"key": key})) is None:
+                org["key"] = key
+                new_org = client.quiz.organization.insert_one(org)
+                created_org = client.quiz.organization.find_one(
+                    {"_id": new_org.inserted_id}
+                )
+                return created_org
+
+            else:
+                key = "".join(
+                    [
+                        secrets.choice(string.ascii_letters + string.digits)
+                        for _ in range(20)
+                    ]
+                )
+                number_of_loops -= 1
+
+        raise HTTPException(
+            status_code=500,
+            detail="Duplicate API key. Please try again.",
+        )
+
+
+@router.get("/authenticate/{api_key}", response_model=OrganizationResponse)
 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(
+            {"key": api_key},
+        )
     ) is not None:
         return org