Create PUT news endpoint

- Add PUT news to router
- Add update news to orm
- Add publish to news model

Author: arthur-mesquita

app/routers/news/routes.py

@@ -8,15 +8,15 @@
 
 import app.services.database.orm.news as orm_news
 from app.routers.authentication import get_current_active_community
-from app.schemas import News
+from app.schemas import News, NewsPublishStatus
 from app.services.database.models import Community as DBCommunity
 from app.services.limiter import limiter
 
 SECRET_KEY = os.getenv("SECRET_KEY", "default_fallback_key")
 ALGORITHM = os.getenv("ALGORITHM", "HS256")
 
 
-class NewsPostResponse(BaseModel):
+class NewsCreateResponse(BaseModel):
     status: str = "News Criada"
 
 
@@ -25,6 +25,10 @@ class NewsGetResponse(BaseModel):
     news_list: list = []
 
 
+class NewsUpdateResponse(BaseModel):
+    status: str = "News Atualizada"
+
+
 class NewsLikeResponse(BaseModel):
     total_likes: int | None
 
@@ -39,7 +43,7 @@ def setup():
 
     @router.post(
         "",
-        response_model=NewsPostResponse,
+        response_model=NewsCreateResponse,
         status_code=status.HTTP_200_OK,
         summary="News endpoint",
         description="Creates news and returns a confirmation message",
@@ -61,7 +65,7 @@ async def post_news(
         await orm_news.create_news(
             session=request.app.db_session_factory, news=news_dict
         )
-        return NewsPostResponse()
+        return NewsCreateResponse()
 
     @router.get(
         "",
@@ -93,6 +97,48 @@ async def get_news(
         )
         return NewsGetResponse(news_list=news_list)
 
+    @router.put(
+        "",
+        response_model=NewsGetResponse,
+        status_code=status.HTTP_200_OK,
+        summary="PUT News",
+        description="Updates news by query params and set publish value",
+    )
+    @limiter.limit("60/minute")
+    async def put_news(
+        request: Request,
+        current_community: Annotated[
+            DBCommunity, Depends(get_current_active_community)
+        ],
+        publish_status: NewsPublishStatus,
+        id: str | None = None,
+        title: str | None = None,
+        content: str | None = None,
+        category: str | None = None,
+        user_email: str = str(Header(..., alias="user-email")),
+        source_url: str | None = None,
+        tags: str | None = None,
+        social_media_url: str | None = None,
+    ):
+        """
+        Get News endpoint that retrieves news filtered by user and query params.
+        """
+        news: dict = {
+            "id": id,
+            "title": title,
+            "content": content,
+            "category": category,
+            "user_email": user_email,
+            "source_url": source_url,
+            "tags": tags,
+            "social_media_url": social_media_url,
+            "publish": publish_status.publish,
+        }
+        await orm_news.update_news(
+            session=request.app.db_session_factory, news=news
+        )
+        return NewsUpdateResponse()
+
     @router.post(
         path="/{news_id}/like",
         response_model=NewsLikeResponse,

app/schemas.py

@@ -54,6 +54,10 @@ class News(BaseModel):
     likes: int = 0
 
 
+class NewsPublishStatus(BaseModel):
+    publish: bool = False
+
+
 class Token(BaseModel):
     access_token: str
     token_type: str

app/services/database/models/news.py

@@ -49,6 +49,7 @@ class News(SQLModel, table=True):
     user_email_list: str = Field(default="[]")
     social_media_url: str
     likes: int = Field(default=0)
+    publish: bool = Field(default=False)
 
     # Chaves estrangeiras
     community_id: Optional[int] = Field(

app/services/database/orm/news.py

@@ -42,7 +42,20 @@ async def get_news_by_query_params(
 
     statement = select(News).where(*filters)
     results = await session.exec(statement)
-    return results.all()
+    return list(results.all())
+
+
+async def update_news(session: AsyncSession, news: dict) -> None:
+    statement = select(News).where(News.id == news["id"])
+    results = await session.exec(statement)
+    news_item = results.first()
+    if news_item:
+        for key, value in news.items():
+            if key != "id" and value is not None:
+                setattr(news_item, key, value)
+        session.add(news_item)
+        await session.commit()
+        await session.refresh(news_item)
 
 
 async def like_news(

tests/test_news.py

@@ -294,8 +294,8 @@ async def test_news_integration(
 async def test_put_news_endpoint(
     session: AsyncSession,
     async_client: AsyncClient,
-    community: Community,
     valid_auth_headers: Mapping[str, str],
+    news_list: list,
 ):
     session.add_all(news_list)
     await session.commit()
@@ -304,35 +304,46 @@ async def test_put_news_endpoint(
     result = await session.exec(statement)
     stored_news = result.first()
     assert stored_news is not None
+    assert stored_news.publish is False
+
+    data: dict = {
+        "title": "updated title",
+        "content": "updated content",
+        "category": "updated_category",
+        "user_email": "updated_email@test.com",
+        "source_url": "https://updated_url.com",
+        "tags": "test_tag_updated",
+        "social_media_url": "https://updated_social_media_url.com",
+    }
 
     response = await async_client.put(
         "/api/news",
         params={
-            "title": "updated title",
-            "content": "updated content",
-            "category": "updated_category",
-            "user_email": "updated_email@test.com",
-            "source_url": "https://updated_url.com",
-            "tags": "test_tag_updated",
-            "user_email_list": "updated_email@test.com",
-            "social_media_url": "https://updated_social_media_url.com",
-            "likes": 42,
+            "id": stored_news.id,
+            "title": data["title"],
+            "content": data["content"],
+            "category": data["category"],
+            "user_email": data["user_email"],
+            "source_url": data["source_url"],
+            "tags": data["tags"],
+            "social_media_url": data["social_media_url"],
         },
         headers=valid_auth_headers,
         json={"publish": True},
     )
-    data = response.json()
     assert response.status_code == status.HTTP_200_OK
-    assert data["tittle"] == "updated title"
-    assert data["content"] == "updated content"
-    assert data["category"] == "updated_category"
-    assert data["user_email"] == "updated_email@test.com"
-    assert data["source_url"] == "https://updated_url.com"
-    assert data["tags"] == "test_tag_updated"
-    assert data["user_email_list"] == "updated_email@test.com"
-    assert data["social_media_url"] == "https://updated_social_media_url.com"
-    assert data["likes"] == 42
-    assert data["publish"] is True
+
+    statement = select(News).where(News.title == data["title"])
+    result = await session.exec(statement)
+    stored_news = result.first()
+    assert stored_news is not None
+    assert stored_news.content == data["content"]
+    assert stored_news.category == data["category"]
+    assert stored_news.user_email == data["user_email"]
+    assert stored_news.source_url == data["source_url"]
+    assert stored_news.tags == data["tags"]
+    assert stored_news.social_media_url == data["social_media_url"]
+    assert stored_news.publish
 
 
 @pytest.mark.asyncio