Update libraries table and schema and create get libraries by language

Author: Gildofj

app/enums.py

@@ -0,0 +1,8 @@
+from enum import Enum
+
+
+class LibraryTagUpdatesEnum(str, Enum):
+    UPDATE = "updates"
+    BUG_FIX = "bug_fix"
+    NEW_FEATURE = "new_feature"
+    SECURITY_FIX = "security_fix"

app/routers/libraries/routes.py

@@ -1,10 +1,14 @@
+from typing import List
+
 from fastapi import APIRouter, HTTPException, Request, status
 from pydantic import BaseModel
 
 from app.schemas import Library as LibrarySchema
+from app.schemas import LibraryNews
 from app.schemas import Subscription as SubscriptionSchema
 from app.services.database.models import Library, Subscription
 from app.services.database.orm.library import (
+    get_libraries_by_language,
     get_library_ids_by_multiple_names,
     insert_library,
 )
@@ -22,6 +26,36 @@ class SubscribeLibraryResponse(BaseModel):
 def setup():
     router = APIRouter(prefix="/libraries", tags=["libraries"])
 
+    @router.get(
+        "",
+        response_model=List[LibrarySchema],
+        status_code=status.HTTP_200_OK,
+        summary="Get libraries by language",
+        description="Get libraries by language",
+    )
+    async def get_by_language(request: Request, language: str):
+        libraryList = await get_libraries_by_language(
+            language=language, session=request.app.db_session_factory
+        )
+        return [
+            LibrarySchema(
+                library_name=libraryDb.library_name,
+                news=[
+                    LibraryNews(
+                        tag=news["tag"], description=news["description"]
+                    )
+                    for news in libraryDb.news
+                ],
+                logo=libraryDb.logo,
+                version=libraryDb.version,
+                release_date=libraryDb.release_date,
+                releases_doc_url=libraryDb.releases_doc_url,
+                fixed_release_url=libraryDb.fixed_release_url,
+                language=libraryDb.language,
+            )
+            for libraryDb in libraryList
+        ]
+
     @router.post(
         "",
         response_model=LibraryResponse,
@@ -35,9 +69,13 @@ async def create_library(
     ):
         library = Library(
             library_name=body.library_name,
-            user_email="",  # TODO: Considerar obter o email do usuário autenticado
-            releases_url=body.releases_url.encoded_string(),
-            logo=body.logo.encoded_string(),
+            news=[news.model_dump() for news in body.news],
+            logo=body.logo,
+            version=body.version,
+            release_date=body.release_date,
+            releases_doc_url=body.releases_doc_url,
+            fixed_release_url=body.fixed_release_url,
+            language=body.language,
         )
         try:
             await insert_library(library, request.app.db_session_factory)

app/schemas.py

@@ -1,13 +1,26 @@
-from enum import Enum
+from datetime import date
 from typing import List
 
-from pydantic import BaseModel, HttpUrl
+from pydantic import BaseModel
+
+from app.enums import LibraryTagUpdatesEnum
+
+
+class LibraryNews(BaseModel):
+    tag: LibraryTagUpdatesEnum
+    description: str
 
 
 class Library(BaseModel):
+    id: int | None = None
     library_name: str
-    releases_url: HttpUrl
-    logo: HttpUrl
+    news: List[LibraryNews]
+    logo: str
+    version: str
+    release_date: date
+    releases_doc_url: str
+    fixed_release_url: str
+    language: str
 
 
 # Community / User Class
@@ -41,15 +54,7 @@ class TokenPayload(BaseModel):
     username: str
 
 
-# Subscription Class
-class SubscriptionTagEnum(str, Enum):
-    UPDATE = "update"
-    BUG_FIX = "bug_fix"
-    NEW_FEATURE = "new_feature"
-    SECURITY_FIX = "security_fix"
-
-
 class Subscription(BaseModel):
     email: str
-    tags: List[SubscriptionTagEnum]
+    tags: List[LibraryTagUpdatesEnum]
     libraries_list: List[str]

app/services/database/models/libraries.py

@@ -1,5 +1,7 @@
-from typing import Optional
+from datetime import date
+from typing import List, Optional
 
+from sqlalchemy import JSON, Column
 from sqlmodel import Field, SQLModel
 
 
@@ -8,9 +10,13 @@ class Library(SQLModel, table=True):
 
     id: Optional[int] = Field(default=None, primary_key=True)
     library_name: str
-    user_email: str
-    releases_url: str
+    news: List[dict] = Field(sa_column=Column(JSON))
     logo: str
+    version: str
+    release_date: date
+    releases_doc_url: str
+    fixed_release_url: str
+    language: str
     community_id: Optional[int] = Field(
         default=None, foreign_key="communities.id"
     )

app/services/database/models/subscriptions.py

@@ -3,15 +3,15 @@
 from sqlalchemy import JSON, Column
 from sqlmodel import Field, SQLModel
 
-from app.schemas import SubscriptionTagEnum
+from app.enums import LibraryTagUpdatesEnum
 
 
 class Subscription(SQLModel, table=True):
     __tablename__ = "subscriptions"  # type: ignore
 
     id: Optional[int] = Field(default=None, primary_key=True)
     email: str
-    tags: List[SubscriptionTagEnum] = Field(sa_column=Column(JSON))
+    tags: List[LibraryTagUpdatesEnum] = Field(sa_column=Column(JSON))
     community_id: Optional[int] = Field(
         default=None, foreign_key="communities.id"
     )

app/services/database/orm/library.py

@@ -26,3 +26,14 @@ async def get_library_ids_by_multiple_names(
     )
     result = await session.exec(statement)
     return [id for id in result.all() if id is not None]
+
+
+async def get_libraries_by_language(
+    language: str,
+    session: AsyncSession,
+) -> List[Library]:
+    statement = select(Library).where(
+        func.lower(Library.language) == language.lower()
+    )
+    result = await session.exec(statement)
+    return [library for library in result.all()]

tests/test_libraries.py

@@ -1,9 +1,13 @@
+from datetime import date
+
 import pytest
 import pytest_asyncio
 from httpx import AsyncClient
 from sqlmodel import select
 from sqlmodel.ext.asyncio.session import AsyncSession
 
+from app.enums import LibraryTagUpdatesEnum
+from app.schemas import LibraryNews
 from app.services.database.models import Community, Library
 
 
@@ -19,24 +23,38 @@ async def community(session: AsyncSession):
 @pytest.mark.asyncio
 async def test_insert_libraries(session: AsyncSession, community: Community):
     library = Library(
-        library_name="Python",
-        user_email="[email protected]",
-        releases_url="http://teste.com",
-        logo="logo",
+        library_name="Flask",
+        news=[
+            LibraryNews(
+                tag=LibraryTagUpdatesEnum.UPDATE, description="Updated"
+            ).model_dump(),
+            LibraryNews(
+                tag=LibraryTagUpdatesEnum.BUG_FIX, description="Fixed"
+            ).model_dump(),
+        ],
+        logo="http://teste.com/",
+        version="3.11",
+        release_date=date.today(),
+        releases_doc_url="http://teste.com/",
+        fixed_release_url="http://teste.com/",
+        language="Python",
         community_id=community.id,
     )
     session.add(library)
     await session.commit()
 
-    statement = select(Library).where(Library.library_name == "Python")
+    statement = select(Library).where(Library.library_name == "Flask")
     result = await session.exec(statement)
     found = result.first()
 
     assert found is not None
-    assert found.library_name == "Python"
-    assert found.user_email == "[email protected]"
-    assert found.releases_url == "http://teste.com"
-    assert found.logo == "logo"
+    assert len(found.news) == 2
+    assert found.logo == "http://teste.com/"
+    assert found.version == "3.11"
+    assert found.release_date == date.today()
+    assert found.releases_doc_url == "http://teste.com/"
+    assert found.fixed_release_url == "http://teste.com/"
+    assert found.language == "Python"
     assert found.community_id == community.id
 
 
@@ -45,9 +63,17 @@ async def test_post_libraries_endpoint(
     async_client: AsyncClient, session: AsyncSession
 ):
     body = {
-        "library_name": "Python from API",
-        "releases_url": "http://teste.com/",
+        "library_name": "FastAPI",
+        "news": [
+            {"tag": "updates", "description": "Updated"},
+            {"tag": "bug_fix", "description": "Fixed"},
+        ],
         "logo": "http://teste.com/",
+        "version": "3.11",
+        "release_date": "2025-01-01",
+        "releases_doc_url": "http://teste.com/",
+        "fixed_release_url": "http://teste.com/",
+        "language": "Python",
     }
 
     response = await async_client.post(
@@ -66,5 +92,98 @@ async def test_post_libraries_endpoint(
     created_library = result.first()
 
     assert created_library is not None
-    assert created_library.releases_url == body["releases_url"]
     assert created_library.logo == body["logo"]
+    assert created_library.version == body["version"]
+    assert created_library.release_date == date.fromisoformat(
+        body["release_date"]
+    )
+    assert created_library.releases_doc_url == body["releases_doc_url"]
+    assert created_library.fixed_release_url == body["fixed_release_url"]
+    assert created_library.language == body["language"]
+    assert len(created_library.news) == len(body["news"])
+
+
+@pytest.mark.asyncio
+async def test_get_libraries_by_language(async_client: AsyncClient):
+    body = {
+        "library_name": "FastAPI",
+        "news": [
+            {"tag": "updates", "description": "Updated"},
+            {"tag": "bug_fix", "description": "Fixed"},
+        ],
+        "logo": "http://teste.com/",
+        "version": "3.11",
+        "release_date": "2025-01-01",
+        "releases_doc_url": "http://teste.com/",
+        "fixed_release_url": "http://teste.com/",
+        "language": "Python",
+    }
+
+    responsePOST = await async_client.post(
+        "/api/libraries",
+        json=body,
+        headers={"Content-Type": "application/json"},
+    )
+
+    assert responsePOST.status_code == 200
+    assert responsePOST.json()["status"] == "Library created successfully"
+
+    response = await async_client.get(
+        "/api/libraries",
+        params={"language": "Python"},
+    )
+
+    assert response.status_code == 200
+
+    data = response.json()
+
+    assert len(data) == 1
+    assert data[0]["library_name"] == "FastAPI"
+    assert len(data[0]["news"]) == 2
+    assert data[0]["news"][0]["tag"] == "updates"
+    assert data[0]["news"][0]["description"] == "Updated"
+    assert data[0]["news"][1]["tag"] == "bug_fix"
+    assert data[0]["news"][1]["description"] == "Fixed"
+    assert data[0]["logo"] == "http://teste.com/"
+    assert data[0]["version"] == "3.11"
+    assert data[0]["release_date"] == "2025-01-01"
+    assert data[0]["releases_doc_url"] == "http://teste.com/"
+    assert data[0]["fixed_release_url"] == "http://teste.com/"
+    assert data[0]["language"] == "Python"
+
+
+@pytest.mark.asyncio
+async def test_get_libraries_by_inexistent_language(async_client: AsyncClient):
+    body = {
+        "library_name": "FastAPI",
+        "news": [
+            {"tag": "updates", "description": "Updated"},
+            {"tag": "bug_fix", "description": "Fixed"},
+        ],
+        "logo": "http://teste.com/",
+        "version": "3.11",
+        "release_date": "2025-01-01",
+        "releases_doc_url": "http://teste.com/",
+        "fixed_release_url": "http://teste.com/",
+        "language": "Python",
+    }
+
+    responsePOST = await async_client.post(
+        "/api/libraries",
+        json=body,
+        headers={"Content-Type": "application/json"},
+    )
+
+    assert responsePOST.status_code == 200
+    assert responsePOST.json()["status"] == "Library created successfully"
+
+    response = await async_client.get(
+        "/api/libraries",
+        params={"language": "NodeJS"},
+    )
+
+    assert response.status_code == 200
+
+    data = response.json()
+
+    assert len(data) == 0