Сreate base classes (#15)

async_api/src/models/base.py

@@ -0,0 +1,7 @@
+from pydantic import BaseModel
+
+
+class UUIDBase(BaseModel):
+    """Модель для базового класса."""
+
+    id: str

async_api/src/models/film.py

@@ -1,10 +1,41 @@
-from pydantic import BaseModel
+from dataclasses import dataclass
 
+from models.base import UUIDBase
+from models.genre import Genre
+from models.person import Person
+
+
+@dataclass
+class FilmWork(UUIDBase):
+    """Модель для хранения информации о фильме."""
+
+    title: str
+    imdb_rating: float
+    description: str
+    genre: list[Genre]
+    actors: list[Person]
+    writers: list[Person]
+    directors: list[Person]
+
+
+class FilmMinimal(UUIDBase):
+    """Модель для хранения краткой информации о фильме."""
 
-class Film(BaseModel):
-    id: str
     title: str
+    imdb_rating: float
     description: str
+    writers: list[Person]
+    directors: list[Person]
+
+
+class FilmShort(UUIDBase):
+    """Модель для хранения краткой информации о фильме на главной странице."""
+
+    title: str
+    imdb_rating: float
+
+
+class FilmWorkMinimal(UUIDBase):
+    """Модель для хранения краткой информации о кинопроизведении."""
 
-    class Config:
-        pass
+    roles: list[str]

async_api/src/models/genre.py

@@ -0,0 +1,14 @@
+from models.base import UUIDBase
+
+
+class Genre(UUIDBase):
+    """Модель для хранения информации жанре в кинопроизведении."""
+
+    name: str
+    description: str
+
+
+class GenreMinimal(UUIDBase):
+    """Модель для хранения краткой информации о жанре."""
+
+    name: str

async_api/src/models/person.py

@@ -0,0 +1,10 @@
+from models.base import UUIDBase
+from models.film import FilmWorkMinimal
+from pydantic import Field
+
+
+class Person(UUIDBase):
+    """Модель для хранения актёра."""
+
+    full_name: str = Field(alias="name")
+    films: list[FilmWorkMinimal]