Implement new fields on schema and database model

Author: Gildofj

app/routers/libraries/routes.py

@@ -36,8 +36,11 @@ 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(),
+            version=body.version,
+            release_date=body.release_date,
+            releases_doc_url=body.releases_doc_url.encoded_string(),
+            fixed_release_url=body.fixed_release_url.encoded_string(),
         )
         try:
             await insert_library(library, request.app.db_session_factory)

app/schemas.py

@@ -1,13 +1,31 @@
+from datetime import date
 from enum import Enum
 from typing import List
 
 from pydantic import BaseModel, HttpUrl
 
 
+class LibraryTagEnum(str, Enum):
+    UPDATES = "updates"
+    BUG_FIX = "bug_fix"
+    NEW_FEATURE = "new_feature"
+    SECURITY_FIX = "security_fix"
+    DEPRECATION = "deprecation"
+
+
+class LibraryNews(BaseModel):
+    tag: LibraryTagEnum
+    description: str
+
+
 class Library(BaseModel):
     library_name: str
-    releases_url: HttpUrl
+    news: List[LibraryNews]
     logo: HttpUrl
+    version: str
+    release_date: date
+    releases_doc_url: HttpUrl
+    fixed_release_url: HttpUrl
 
 
 # Community / User Class
@@ -31,15 +49,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[LibraryTagEnum]
     libraries_list: List[str]

app/services/database/models/libraries.py

@@ -1,3 +1,4 @@
+from datetime import date
 from typing import Optional
 
 from sqlmodel import Field, SQLModel
@@ -9,8 +10,11 @@ class Library(SQLModel, table=True):
     id: Optional[int] = Field(default=None, primary_key=True)
     library_name: str
     user_email: str
-    releases_url: str
     logo: str
+    version: str
+    release_date: date
+    releases_doc_url: str
+    fixed_release_url: str
     community_id: Optional[int] = Field(
         default=None, foreign_key="communities.id"
     )

app/services/database/models/subscriptions.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 
-from schemas import SubscriptionTagEnum
+from schemas import LibraryTagEnum
 from sqlalchemy import JSON, Column
 from sqlmodel import Field, SQLModel
 
@@ -10,7 +10,7 @@ class Subscription(SQLModel, table=True):
 
     id: Optional[int] = Field(default=None, primary_key=True)
     email: str
-    tags: List[SubscriptionTagEnum] = Field(sa_column=Column(JSON))
+    tags: List[LibraryTagEnum] = Field(sa_column=Column(JSON))
     community_id: Optional[int] = Field(
         default=None, foreign_key="communities.id"
     )

tests/test_libraries.py

@@ -1,3 +1,5 @@
+from datetime import date
+
 import pytest
 import pytest_asyncio
 from httpx import AsyncClient
@@ -20,8 +22,11 @@ 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",
+        logo="http://teste.com",
+        version="3.12",
+        release_date=date.today(),
+        releases_doc_url="http://teste.com",
+        fixed_release_url="http://teste.com",
         community_id=community.id,
     )
     session.add(library)
@@ -34,8 +39,11 @@ async def test_insert_libraries(session: AsyncSession, community: Community):
     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 found.logo == "http://teste.com"
+    assert found.version == "3.12"
+    assert found.release_date == date.today()
+    assert found.releases_doc_url == "http://teste.com"
+    assert found.fixed_release_url == "http://teste.com"
     assert found.community_id == community.id
 
 
@@ -45,8 +53,15 @@ async def test_post_libraries_endpoint(
 ):
     body = {
         "library_name": "Python from API",
-        "releases_url": "http://teste.com/",
+        "news": [
+            {"tag": "updates", "description": "New feature"},
+            {"tag": "bug_fix", "description": "Fixed bug"},
+        ],
         "logo": "http://teste.com/",
+        "version": "3.12",
+        "release_date": "2023-01-01",
+        "releases_doc_url": "http://teste.com/",
+        "fixed_release_url": "http://teste.com/",
     }
 
     response = await async_client.post(
@@ -65,5 +80,10 @@ 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"]

tests/test_subscriptions.py

@@ -1,7 +1,7 @@
 import pytest
 import pytest_asyncio
 from httpx import AsyncClient
-from schemas import SubscriptionTagEnum
+from schemas import LibraryTagEnum
 from services.database.models import Community, Subscription
 from sqlmodel import select
 from sqlmodel.ext.asyncio.session import AsyncSession
@@ -20,7 +20,7 @@ async def community(session: AsyncSession):
 async def test_insert_subscription(session: AsyncSession, community: Community):
     subscription = Subscription(
         email="[email protected]",
-        tags=[SubscriptionTagEnum.BUG_FIX, SubscriptionTagEnum.UPDATE],
+        tags=[LibraryTagEnum.BUG_FIX, LibraryTagEnum.UPDATES],
         community_id=community.id,
     )
     session.add(subscription)
@@ -38,8 +38,8 @@ async def test_insert_subscription(session: AsyncSession, community: Community):
     assert found is not None
     assert found.email == "[email protected]"
     assert found.tags == [
-        SubscriptionTagEnum.BUG_FIX,
-        SubscriptionTagEnum.UPDATE,
+        LibraryTagEnum.BUG_FIX,
+        LibraryTagEnum.UPDATES,
     ]
     assert found.community_id == community.id
 
@@ -79,7 +79,7 @@ async def preset_library(async_client: AsyncClient):
 async def test_post_subscribe_endpoint(async_client: AsyncClient):
     body = {
         "email": "[email protected]",
-        "tags": ["bug_fix", "update"],
+        "tags": ["bug_fix", "updates"],
         "libraries_list": ["Python", "Django"],
     }