Update routes/approvals.py

routes/approvals.py

@@ -5,11 +5,20 @@ import asyncio
 from fastapi import APIRouter, HTTPException, Header
 from typing import Dict, Any
 from functools import lru_cache
+from pydantic import BaseModel
 import requests
 from collections import defaultdict
 
 router = APIRouter()
 
+
+class ApproveAccountRequest(BaseModel):
+    user_id: str
+
+class RejectAccountRequest(BaseModel):
+    user_id: str
+    reason: str | None = None
+
 # Configuração do Supabase
 SUPABASE_URL = "https://ussxqnifefkgkaumjann.supabase.co"
 SUPABASE_KEY = os.getenv("SUPA_KEY")
@@ -142,7 +151,7 @@ async def get_users_pending_approval() -> Dict[str, Any]:
 
 @router.post("/admin/approve-account")
 async def approve_account(
-    user_id: str,
+    body: ApproveAccountRequest,
     user_token: str = Header(None, alias="User-key")
 ):
     """
@@ -151,7 +160,7 @@ async def approve_account(
     try:
         await verify_admin_token(user_token)
 
-        update_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
+        update_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{body.user_id}"
         payload = {
             "approved_account": True
         }
@@ -172,8 +181,7 @@ async def approve_account(
 
 @router.post("/admin/reject-account")
 async def reject_account(
-    user_id: str,
-    reason: str = None,
+    body: RejectAccountRequest,
     user_token: str = Header(None, alias="User-key")
 ):
     """
@@ -182,10 +190,10 @@ async def reject_account(
     try:
         await verify_admin_token(user_token)
 
-        update_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
+        update_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{body.user_id}"
         payload = {
             "approved_account": False,
-            "approval_reason": reason or "No reason specified"
+            "approval_reason": body.reason or "No reason specified"
         }
 
         async with aiohttp.ClientSession() as session: