♻️ Refactor API in utils.py

src/backend/app/app/api/api_v1/endpoints/utils.py

@@ -1,35 +1,35 @@
-from typing import Any
-
 from fastapi import APIRouter, Depends
 from pydantic.networks import EmailStr
 
-from app import models, schemas
-from app.api import deps
+from app.api.deps import get_current_active_superuser
 from app.core.celery_app import celery_app
+from app.models import Message
 from app.utils import send_test_email
 
 router = APIRouter()
 
 
-@router.post("/test-celery/", response_model=schemas.Msg, status_code=201)
-def test_celery(
-    msg: schemas.Msg,
-    current_user: models.User = Depends(deps.get_current_active_superuser),
-) -> Any:
+@router.post(
+    "/test-celery/",
+    dependencies=[Depends(get_current_active_superuser)],
+    status_code=201,
+)
+def test_celery(body: Message) -> Message:
     """
     Test Celery worker.
     """
-    celery_app.send_task("app.worker.test_celery", args=[msg.msg])
-    return {"msg": "Word received"}
+    celery_app.send_task("app.worker.test_celery", args=[body.message])
+    return Message(message="Word received")
 
 
-@router.post("/test-email/", response_model=schemas.Msg, status_code=201)
-def test_email(
-    email_to: EmailStr,
-    current_user: models.User = Depends(deps.get_current_active_superuser),
-) -> Any:
+@router.post(
+    "/test-email/",
+    dependencies=[Depends(get_current_active_superuser)],
+    status_code=201,
+)
+def test_email(email_to: EmailStr) -> Message:
     """
     Test emails.
     """
     send_test_email(email_to=email_to)
-    return {"msg": "Test email sent"}
+    return Message(message="Test email sent")