wip: add POST nominee

Author: jbriones1

src/nominees/urls.py

@@ -8,52 +8,71 @@
     NomineeInfoUpdateParams,
 )
 from nominees.tables import NomineeInfo
+from utils.shared_models import DetailModel
 from utils.urls import AdminTypeEnum, admin_or_raise
 
 router = APIRouter(
     prefix="/nominee",
     tags=["nominee"],
 )
 
+
+@router.post(
+    "",
+    description="Nominee info is always publically tied to election, so be careful!",
+    response_model=NomineeInfoModel,
+    responses={500: {"description": "failed to fetch new nominee", "model": DetailModel}},
+    operation_id="create_nominee",
+)
+async def create_nominee(request: Request, db_session: database.DBSession, body: NomineeInfoModel):
+    await admin_or_raise(request, db_session)
+    await nominees.crud.create_nominee_info(
+        db_session,
+        NomineeInfo(
+            computing_id=body.computing_id,
+            full_name=body.full_name,
+            linked_in=body.linked_in,
+            instagram=body.instagram,
+            email=body.email,
+            discord_username=body.discord_username,
+        ),
+    )
+
+    nominee_info = await nominees.crud.get_nominee_info(db_session, body.computing_id)
+    if nominee_info is None:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail="couldn't fetch newly created nominee"
+        )
+
+    return JSONResponse(nominee_info)
+
+
 @router.get(
     "/{computing_id:str}",
     description="Nominee info is always publically tied to election, so be careful!",
     response_model=NomineeInfoModel,
-    responses={
-        404: { "description": "nominee doesn't exist" }
-    },
-    operation_id="get_nominee"
+    responses={404: {"description": "nominee doesn't exist"}},
+    operation_id="get_nominee",
 )
-async def get_nominee_info(
-    request: Request,
-    db_session: database.DBSession,
-    computing_id: str
-):
+async def get_nominee_info(request: Request, db_session: database.DBSession, computing_id: str):
     # Putting this one behind the admin wall since it has contact information
     await admin_or_raise(request, db_session, AdminTypeEnum.Election)
     nominee_info = await nominees.crud.get_nominee_info(db_session, computing_id)
     if nominee_info is None:
-        raise HTTPException(
-            status_code=status.HTTP_404_NOT_FOUND,
-            detail="nominee doesn't exist"
-        )
+        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="nominee doesn't exist")
 
     return JSONResponse(nominee_info.serialize())
 
+
 @router.patch(
     "/{computing_id:str}",
     description="Will create or update nominee info. Returns an updated copy of their nominee info.",
     response_model=NomineeInfoModel,
-    responses={
-        500: { "description": "Failed to retrieve updated nominee." }
-    },
-    operation_id="update_nominee"
+    responses={500: {"description": "Failed to retrieve updated nominee."}},
+    operation_id="update_nominee",
 )
 async def provide_nominee_info(
-    request: Request,
-    db_session: database.DBSession,
-    body: NomineeInfoUpdateParams,
-    computing_id: str
+    request: Request, db_session: database.DBSession, body: NomineeInfoUpdateParams, computing_id: str
 ):
     # TODO: There needs to be a lot more validation here.
     await admin_or_raise(request, db_session, AdminTypeEnum.Election)
@@ -63,14 +82,15 @@ async def provide_nominee_info(
     if body.full_name is not None:
         updated_data["full_name"] = body.full_name
     if body.linked_in is not None:
-       updated_data["linked_in"] = body.linked_in
+        updated_data["linked_in"] = body.linked_in
     if body.instagram is not None:
         updated_data["instagram"] = body.instagram
     if body.email is not None:
         updated_data["email"] = body.email
     if body.discord_username is not None:
         updated_data["discord_username"] = body.discord_username
 
+    # TODO: Look into using something built into SQLAlchemy/Pydantic for better entry updates
     existing_info = await nominees.crud.get_nominee_info(db_session, computing_id)
     # if not already existing, create it
     if not existing_info:
@@ -97,8 +117,5 @@ async def provide_nominee_info(
 
     nominee_info = await nominees.crud.get_nominee_info(db_session, computing_id)
     if not nominee_info:
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail="failed to get updated nominee"
-        )
+        raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail="failed to get updated nominee")
     return JSONResponse(nominee_info.serialize())