Created models for genres

library_service/models/db/__init__.py

@@ -1,7 +1,14 @@
 from .author import Author
 from .book import Book
-from .links import AuthorBookLink, AuthorWithBooks, BookWithAuthors
+from .links import (
+    AuthorBookLink, GenreBookLink,
+    AuthorWithBooks, BookWithAuthors,
+    GenreWithBooks, BookWithAuthorsAndGenres
+)
 
 __all__ = [
-    'Author', 'Book', 'AuthorBookLink', 'AuthorWithBooks', 'BookWithAuthors'
+    'Author', 'Book',
+    'AuthorBookLink', 'AuthorWithBooks',
+    'BookWithAuthors', 'GenreBookLink',
+    'GenreWithBooks', 'BookWithAuthorsAndGenres'
 ]

library_service/models/db/book.py

@@ -1,10 +1,11 @@
 from typing import List, Optional, TYPE_CHECKING
 from sqlmodel import SQLModel, Field, Relationship
 from ..dto.book import BookBase
-from .links import AuthorBookLink
+from .links import AuthorBookLink, GenreBookLink
 
 if TYPE_CHECKING:
     from .author import Author
+    from .genre import Genre
 
 class Book(BookBase, table=True):
     id: Optional[int] = Field(default=None, primary_key=True, index=True)
@@ -12,3 +13,7 @@ class Book(BookBase, table=True):
         back_populates="books",
         link_model=AuthorBookLink
     )
+    genres: List["Genre"] = Relationship(
+        back_populates="books",
+        link_model=GenreBookLink
+    )

library_service/models/db/genre.py

@@ -0,0 +1,14 @@
+from typing import List, Optional, TYPE_CHECKING
+from sqlmodel import SQLModel, Field, Relationship
+from ..dto.genre import GenreBase
+from .links import GenreBookLink
+
+if TYPE_CHECKING:
+    from .book import Book
+
+class Genre(GenreBase, table=True):
+    id: Optional[int] = Field(default=None, primary_key=True, index=True)
+    books: List["Book"] = Relationship(
+        back_populates="authors",
+        link_model=GenreBookLink
+    )

library_service/models/db/links.py

@@ -3,13 +3,28 @@ from typing import List
 
 from library_service.models.dto.author import AuthorRead
 from library_service.models.dto.book import BookRead
+from library_service.models.dto.genre import GenreRead
 
 class AuthorBookLink(SQLModel, table=True):
     author_id: int | None = Field(default=None, foreign_key="author.id", primary_key=True)
     book_id: int | None = Field(default=None, foreign_key="book.id", primary_key=True)
 
+class GenreBookLink(SQLModel, table=True):
+    genre_id: int | None = Field(default=None, foreign_key="genre.id", primary_key=True)
+    book_id: int | None = Field(default=None, foreign_key="book.id", primary_key=True)
+
 class AuthorWithBooks(AuthorRead):
     books: List[BookRead] = Field(default_factory=list)
 
 class BookWithAuthors(BookRead):
     authors: List[AuthorRead] = Field(default_factory=list)
+
+class BookWithGenres(BookRead):
+    genres: List[GenreRead] = Field(default_factory=list)
+
+class GenreWithBooks(GenreRead):
+    books: List[BookRead] = Field(default_factory=list)
+
+class BookWithAuthorsAndGenres(BookRead):
+    authors: List[AuthorRead] = Field(default_factory=list)
+    genres: List[GenreRead] = Field(default_factory=list)

library_service/models/dto/__init__.py

@@ -6,10 +6,14 @@ from .book import (
     BookBase, BookCreate, BookUpdate,
     BookRead, BookList
 )
-# from .common import PaginatedResponse
+
+from .genre import (
+    GenreBase, GenreCreate, GenreUpdate,
+    GenreRead, GenreList
+)
 
 __all__ = [
     'AuthorBase', 'AuthorCreate', 'AuthorUpdate', 'AuthorRead', 'AuthorList',
     'BookBase', 'BookCreate', 'BookUpdate', 'BookRead', 'BookList',
-    # 'PaginatedResponse'
+    'GenreBase', 'GenreCreate', 'GenreUpdate', 'GenreRead', 'GenreList',
 ]

library_service/models/dto/genre.py

@@ -0,0 +1,25 @@
+from sqlmodel import SQLModel
+from pydantic import ConfigDict
+from typing import Optional, List
+
+class GenreBase(SQLModel):
+    name: str
+
+    model_config = ConfigDict( #pyright: ignore
+        json_schema_extra={
+            "example": {"name": "genre_name"}
+        }
+    )
+
+class GenreCreate(GenreBase):
+    pass
+
+class GenreUpdate(SQLModel):
+    name: Optional[str] = None
+
+class GenreRead(GenreBase):
+    id: int
+
+class GenreList(SQLModel):
+    genres: List[GenreRead]
+    total: int