Merge pull request #45 from avantifellows/org

Endpoint to onboard an organization

app/models.py

@@ -7,7 +7,23 @@
 
 
 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
+        schema_extra = {"example": {"name": "Avanti Fellows"}}
 
 
 class Image(BaseModel):

app/routers/organizations.py

@@ -1,15 +1,48 @@
 from fastapi import APIRouter, status, HTTPException
+from settings import Settings
 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"])
+settings = Settings()
 
 
-@router.get("/check-auth-token/{api_key}", response_model=Organization)
+def generate_random_string():
+    return "".join(
+        [
+            secrets.choice(string.ascii_letters + string.digits)
+            for _ in range(settings.random_string_length)
+        ]
+    )
+
+
+@router.post("/", response_model=OrganizationResponse)
+async def create_organization(organization: Organization):
+    organization = jsonable_encoder(organization)
+
+    # create an API key
+    key = generate_random_string()
+
+    # check if API key exists
+    if (client.quiz.organization.find_one({"key": key})) is None:
+        organization["key"] = key
+        new_organization = client.quiz.organization.insert_one(organization)
+        created_organization = client.quiz.organization.find_one(
+            {"_id": new_organization.inserted_id}
+        )
+        return created_organization
+
+
+@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
 

app/settings.py

@@ -0,0 +1,5 @@
+from pydantic import BaseSettings
+
+
+class Settings(BaseSettings):
+    random_string_length: int = 20