feat: Updated endpoint endpoints/contact-us.post.py via AI with auto lint fixes

endpoints/contact-us.post.py

@@ -1,18 +1,18 @@
-from fastapi import APIRouter, HTTPException, status
-from typing import Any, Dict
-from helpers.contact_form_helpers import validate_contact_form_data, create_contact_form
+from fastapi import APIRouter, status
+from pydantic import BaseModel
+
+from schemas.contact_form import ContactFormCreate
+from helpers.contact_form_helpers import create_contact_form
 
 router = APIRouter()
 
-@router.post("/contact-us", status_code=status.HTTP_201_CREATED, response_model=Dict[str, Any])
-async def create_new_contact_form(contact_form_data: Dict[str, Any]):
-    """Create a new contact form"""
-    is_valid, error_message = validate_contact_form_data(contact_form_data)
-    if not is_valid:
-        raise HTTPException(status_code=400, detail=error_message)
+class ContactFormResponse(BaseModel):
+    id: int
+    name: str
+    email: str
+    message: str
 
-    new_contact_form = create_contact_form(contact_form_data)
-    if not new_contact_form:
-        raise HTTPException(status_code=500, detail="Failed to create contact form")
-
-    return new_contact_form
+@router.post("/contact-us", status_code=status.HTTP_201_CREATED, response_model=ContactFormResponse)
+async def create_contact_form_endpoint(form_data: ContactFormCreate):
+    new_form = create_contact_form(form_data)
+    return new_form