add Movie model and schema

Author: mahenzon

examples/api_for_sqlalchemy/models/__init__.py

@@ -1,5 +1,7 @@
+from examples.api_for_sqlalchemy.models.age_rating import AgeRating
 from examples.api_for_sqlalchemy.models.child import Child
 from examples.api_for_sqlalchemy.models.computer import Computer
+from examples.api_for_sqlalchemy.models.movie import Movie
 from examples.api_for_sqlalchemy.models.parent import Parent
 from examples.api_for_sqlalchemy.models.parent_to_child_association import ParentToChildAssociation
 from examples.api_for_sqlalchemy.models.post import Post
@@ -9,8 +11,10 @@
 from examples.api_for_sqlalchemy.models.workplace import Workplace
 
 __all__ = (
+    "AgeRating",
     "Child",
     "Computer",
+    "Movie",
     "Parent",
     "ParentToChildAssociation",
     "Post",

tests/fixtures/models/age_rating.py renamed to examples/api_for_sqlalchemy/models/age_rating.py

@@ -1,11 +1,13 @@
-from sqlalchemy import (
-    String,
-    Text,
-)
-from sqlalchemy.orm import Mapped, mapped_column
+from typing import TYPE_CHECKING
+
+from sqlalchemy import String, Text
+from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from examples.api_for_sqlalchemy.models.base import BaseMetadata
 
+if TYPE_CHECKING:
+    from examples.api_for_sqlalchemy.models.movie import Movie
+
 
 class AgeRating(BaseMetadata):
     __tablename__ = "age_rating"
@@ -20,5 +22,9 @@ class AgeRating(BaseMetadata):
         server_default="",
     )
 
+    movies: Mapped[list["Movie"]] = relationship(
+        back_populates="age_rating_obj",
+    )
+
     def __str__(self) -> str:
         return self.name

examples/api_for_sqlalchemy/models/movie.py

@@ -0,0 +1,54 @@
+from typing import TYPE_CHECKING
+
+from sqlalchemy import (
+    ForeignKey,
+    Identity,
+    Integer,
+    String,
+    Text,
+)
+from sqlalchemy.orm import (
+    Mapped,
+    mapped_column,
+    relationship,
+)
+
+from examples.api_for_sqlalchemy.models.base import Base
+
+if TYPE_CHECKING:
+    from examples.api_for_sqlalchemy.models.age_rating import AgeRating
+
+
+class Movie(Base):
+    __tablename__ = "movie"
+
+    id: Mapped[int] = mapped_column(
+        Integer,
+        Identity(always=True),
+        primary_key=True,
+        autoincrement=True,
+    )
+    title: Mapped[str] = mapped_column(
+        String(120),
+        index=True,
+    )
+    description: Mapped[str] = mapped_column(
+        Text,
+        default="",
+        server_default="",
+    )
+    age_rating: Mapped[str | None] = mapped_column(
+        ForeignKey(
+            "age_rating.name",
+            ondelete="SET NULL",
+        ),
+    )
+    age_rating_obj: Mapped["AgeRating"] = relationship(
+        back_populates="movies",
+    )
+
+    def __str__(self) -> str:
+        return self.title
+
+    def __repr__(self) -> str:
+        return f"Movie(id={self.id}, title={self.title!r})"

examples/api_for_sqlalchemy/schemas/__init__.py

@@ -1,3 +1,10 @@
+from .age_rating import (
+    AgeRatingAttributesSchema,
+    AgeRatingBaseSchema,
+    AgeRatingCreateSchema,
+    AgeRatingSchema,
+    AgeRatingUpdateSchema,
+)
 from .child import (
     ChildAttributesSchema,
     ChildInSchema,
@@ -10,6 +17,13 @@
     ComputerPatchSchema,
     ComputerSchema,
 )
+from .movie import (
+    MovieAttributesSchema,
+    MovieBaseSchema,
+    MovieCreateSchema,
+    MovieSchema,
+    MovieUpdateSchema,
+)
 from .parent import (
     ParentAttributesSchema,
     ParentInSchema,
@@ -51,6 +65,11 @@
 )
 
 __all__ = (
+    "AgeRatingAttributesSchema",
+    "AgeRatingBaseSchema",
+    "AgeRatingCreateSchema",
+    "AgeRatingSchema",
+    "AgeRatingUpdateSchema",
     "ChildAttributesSchema",
     "ChildInSchema",
     "ChildPatchSchema",
@@ -60,6 +79,11 @@
     "ComputerPatchSchema",
     "ComputerSchema",
     "CustomUserAttributesSchema",
+    "MovieAttributesSchema",
+    "MovieBaseSchema",
+    "MovieCreateSchema",
+    "MovieSchema",
+    "MovieUpdateSchema",
     "ParentAttributesSchema",
     "ParentInSchema",
     "ParentPatchSchema",

examples/api_for_sqlalchemy/schemas/age_rating.py

@@ -0,0 +1,49 @@
+from typing import TYPE_CHECKING, Annotated, Optional
+
+from annotated_types import MaxLen, MinLen
+from pydantic import ConfigDict
+
+from fastapi_jsonapi.schema_base import BaseModel
+from fastapi_jsonapi.types_metadata import RelationshipInfo
+
+name_constrained = Annotated[
+    str,
+    MinLen(1),
+    MaxLen(20),
+]
+
+if TYPE_CHECKING:
+    from examples.api_for_sqlalchemy.schemas.movie import MovieSchema
+
+
+class AgeRatingAttributesSchema(BaseModel):
+    model_config = ConfigDict(
+        from_attributes=True,
+    )
+    name: str
+    description: str
+
+
+class AgeRatingBaseSchema(AgeRatingAttributesSchema):
+    movies: Annotated[
+        Optional[list["MovieSchema"]],
+        RelationshipInfo(
+            resource_type="movie",
+            many=True,
+        ),
+    ] = None
+
+
+class AgeRatingCreateSchema(AgeRatingBaseSchema):
+    name: name_constrained
+
+
+class AgeRatingUpdateSchema(AgeRatingBaseSchema):
+    name: Optional[name_constrained] = None
+    description: Optional[str] = None
+
+
+class AgeRatingSchema(AgeRatingBaseSchema):
+    """
+    Age Rating
+    """

examples/api_for_sqlalchemy/schemas/movie.py

@@ -0,0 +1,60 @@
+from datetime import date
+from typing import (
+    TYPE_CHECKING,
+    Annotated,
+    Optional,
+)
+
+from annotated_types import MaxLen, MinLen
+from pydantic import ConfigDict
+
+from fastapi_jsonapi.schema_base import BaseModel
+from fastapi_jsonapi.types_metadata import RelationshipInfo
+
+if TYPE_CHECKING:
+    from examples.api_for_sqlalchemy.schemas.age_rating import AgeRatingSchema
+
+title_constrained = Annotated[
+    str,
+    MinLen(1),
+    MaxLen(120),
+]
+
+
+class MovieAttributesSchema(BaseModel):
+    model_config = ConfigDict(
+        from_attributes=True,
+    )
+    title: str
+    description: str
+    age_rating: str | None = None
+
+
+class MovieBaseSchema(MovieAttributesSchema):
+    age_rating_obj: Annotated[
+        Optional["AgeRatingSchema"],
+        RelationshipInfo(
+            resource_type="age-rating",
+            resource_id_example="PG-13",
+            id_field_name="name",
+        ),
+    ] = None
+
+
+class MovieCreateSchema(MovieBaseSchema):
+    """
+    Create
+    """
+
+    title: title_constrained
+
+
+class MovieUpdateSchema(MovieBaseSchema):
+    title: title_constrained | None = None
+    description: str | None = None
+    release_date: date | None = None
+    duration: int | None = None
+
+
+class MovieSchema(MovieBaseSchema):
+    id: int

examples/api_for_sqlalchemy/urls.py

@@ -7,8 +7,10 @@
 
 from .api.views_base import ViewBase
 from .models import (
+    AgeRating,
     Child,
     Computer,
+    Movie,
     Parent,
     ParentToChildAssociation,
     Post,
@@ -17,12 +19,18 @@
     Workplace,
 )
 from .schemas import (
+    AgeRatingCreateSchema,
+    AgeRatingSchema,
+    AgeRatingUpdateSchema,
     ChildInSchema,
     ChildPatchSchema,
     ChildSchema,
     ComputerInSchema,
     ComputerPatchSchema,
     ComputerSchema,
+    MovieCreateSchema,
+    MovieSchema,
+    MovieUpdateSchema,
     ParentInSchema,
     ParentPatchSchema,
     ParentSchema,
@@ -131,6 +139,28 @@ def add_routes(app: FastAPI):
         schema_in_patch=WorkplacePatchSchema,
         schema_in_post=WorkplaceInSchema,
     )
+    builder.add_resource(
+        path="/age-ratings",
+        tags=["Age Ratings"],
+        resource_type="age-rating",
+        view=ViewBase,
+        model=AgeRating,
+        schema=AgeRatingSchema,
+        schema_in_post=AgeRatingCreateSchema,
+        schema_in_patch=AgeRatingUpdateSchema,
+        model_id_field_name="name",
+    )
+    builder.add_resource(
+        path="/movies",
+        tags=["Movie"],
+        resource_type="movie",
+        view=ViewBase,
+        model=Movie,
+        schema=MovieSchema,
+        schema_in_post=MovieCreateSchema,
+        schema_in_patch=MovieUpdateSchema,
+        model_id_field_name="name",
+    )
     builder.initialize()
 
     atomic = AtomicOperations()

tests/fixtures/app.py

@@ -6,8 +6,10 @@
 from pydantic import BaseModel, ConfigDict
 
 from examples.api_for_sqlalchemy.models import (
+    AgeRating,
     Child,
     Computer,
+    Movie,
     Parent,
     ParentToChildAssociation,
     Post,
@@ -16,12 +18,18 @@
     UserBio,
 )
 from examples.api_for_sqlalchemy.schemas import (
+    AgeRatingCreateSchema,
+    AgeRatingSchema,
+    AgeRatingUpdateSchema,
     ChildInSchema,
     ChildPatchSchema,
     ChildSchema,
     ComputerInSchema,
     ComputerPatchSchema,
     ComputerSchema,
+    MovieCreateSchema,
+    MovieSchema,
+    MovieUpdateSchema,
     ParentPatchSchema,
     ParentSchema,
     ParentToChildAssociationSchema,
@@ -39,11 +47,8 @@
 from fastapi_jsonapi.data_typing import TypeModel
 from fastapi_jsonapi.views.view_base import ViewBase
 
-from .models import AgeRating, Alpha, Beta, CustomUUIDItem, Delta, Gamma, Task
+from .models import Alpha, Beta, CustomUUIDItem, Delta, Gamma, Task
 from .schemas import (
-    AgeRatingCreateSchema,
-    AgeRatingSchema,
-    AgeRatingUpdateSchema,
     AlphaSchema,
     BetaSchema,
     CustomUUIDItemSchema,
@@ -176,6 +181,17 @@ def add_routers(app_plain: FastAPI):
         schema_in_patch=AgeRatingUpdateSchema,
         model_id_field_name="name",
     )
+    builder.add_resource(
+        path="/movies",
+        tags=["Movie"],
+        resource_type="movie",
+        view=ViewBaseGeneric,
+        model=Movie,
+        schema=MovieSchema,
+        schema_in_post=MovieCreateSchema,
+        schema_in_patch=MovieUpdateSchema,
+        model_id_field_name="name",
+    )
     builder.initialize()
 
     return app_plain

tests/fixtures/entities.py

@@ -6,6 +6,7 @@
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from examples.api_for_sqlalchemy.models import (
+    AgeRating,
     Child,
     Computer,
     Parent,
@@ -17,7 +18,7 @@
     Workplace,
 )
 from tests.common import is_postgres_tests
-from tests.fixtures.models import AgeRating, Task
+from tests.fixtures.models import Task
 from tests.misc.utils import fake