Update routers/wordpress.py

routers/wordpress.py

@@ -1,6 +1,7 @@
 import os
 import httpx
 from fastapi import APIRouter, HTTPException
+from pydantic import BaseModel
 
 router = APIRouter()
 
@@ -14,17 +15,25 @@ if not WP_URL or not WP_USER or not WP_PASS:
 
 AUTH = (WP_USER, WP_PASS)
 
+# 📦 Modelo de entrada
+class WPPost(BaseModel):
+    title: str
+    content: str
+    image_url: str | None = None
+    status: str = "publish"
+
 # 🚀 Criar post no WordPress
 @router.post("/wordpress/post")
-async def create_wordpress_post(title: str, content: str, image_url: str = None, status: str = "publish"):
+async def create_wordpress_post(post: WPPost):
     async with httpx.AsyncClient(timeout=20.0) as client:
-        # Se houver imagem → baixar e enviar como mídia destacada
         featured_media_id = None
-        if image_url:
+
+        # Se houver imagem → baixar e enviar como mídia
+        if post.image_url:
             try:
-                img_resp = await client.get(image_url, timeout=15.0)
+                img_resp = await client.get(post.image_url, timeout=15.0)
                 if img_resp.status_code == 200:
-                    filename = image_url.split("/")[-1].split("?")[0]
+                    filename = post.image_url.split("/")[-1].split("?")[0]
                     media_headers = {
                         "Content-Disposition": f"attachment; filename={filename}",
                         "Content-Type": img_resp.headers.get("content-type", "image/jpeg")
@@ -45,9 +54,9 @@ async def create_wordpress_post(title: str, content: str, image_url: str = None,
 
         # Criar o post
         post_data = {
-            "title": title,
-            "content": content,
-            "status": status
+            "title": post.title,
+            "content": post.content,
+            "status": post.status
         }
         if featured_media_id:
             post_data["featured_media"] = featured_media_id