88import elections
99import elections .crud
1010import elections .tables
11- from elections .models import ElectionModel
11+ from elections .models import ElectionModel , NomineeInfoModel
1212from elections .tables import Election , NomineeApplication , NomineeInfo , election_types
1313from officers .constants import OfficerPosition
1414from permission .types import ElectionOfficer , WebsiteAdmin
15+ from utils .shared_models import SuccessFailModel
1516from utils .urls import is_logged_in
1617
1718router = APIRouter (
@@ -375,7 +376,7 @@ async def get_election_registrations(
375376
376377@router .post (
377378 "/registration/{election_name:str}" ,
378- description = "register for a specific position in this election, but doesn't set a speech"
379+ description = "register for a specific position in this election, but doesn't set a speech" ,
379380)
380381async def register_in_election (
381382 request : Request ,
@@ -552,7 +553,8 @@ async def delete_registration(
552553
553554@router .get (
554555 "/nominee/info" ,
555- description = "Nominee info is always publically tied to elections, so be careful!"
556+ description = "Nominee info is always publically tied to elections, so be careful!" ,
557+ response_model = NomineeInfoModel
556558)
557559async def get_nominee_info (
558560 request : Request ,
@@ -576,7 +578,8 @@ async def get_nominee_info(
576578
577579@router .put (
578580 "/nominee/info" ,
579- description = "Will create or update nominee info. Returns an updated copy of their nominee info."
581+ description = "Will create or update nominee info. Returns an updated copy of their nominee info." ,
582+ response_model = NomineeInfoModel
580583)
581584async def provide_nominee_info (
582585 request : Request ,
0 commit comments