StepanovPlaton/torrent_backend
1
0
Fork 0
mirror of https://github.com/StepanovPlaton/torrent_backend.git synced 2026-04-03 20:30:38 +04:00
Code Issues Packages Projects Releases Wiki Activity

Complete genres and actors

Browse Source
This commit is contained in:
Степанов Платон
2024-07-06 20:08:30 +04:00
parent b9e22fcc4c
commit 4c0bdcbd6a
9 changed files with 33 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
database/crud/movies.py
View File

@@ -22,21 +22,26 @@ class MoviesCRUD(EntityCRUD[mdl.Movie]):
async def change_actors(db: AsyncSession, movie: mdl.Movie, info: sch.MovieCreate):
movie_actors = await MovieActorsCRUD.get_all(db)
if (info.actors):
actors_id = [actor.id for actor in info.actors]
movie.actors = [
actor for actor in movie_actors if actor.id in info.actors]
actor for actor in movie_actors if actor.id in actors_id]
@staticmethod
async def change_genres(db: AsyncSession, movie: mdl.Movie, info: sch.MovieCreate):
movie_genres = await MovieGenresCRUD.get_all(db)
if (info.genres):
genres_id = [genre.id for genre in info.genres]
movie.genres = [
genre for genre in movie_genres if genre.id in info.genres]
genre for genre in movie_genres if genre.id in genres_id]
@staticmethod
async def add(db: AsyncSession,
info: sch.MovieCreate,
owner_id: int):
movie = mdl.Movie(**info.model_dump(),
movie_data_db = \
{k: v for k, v in info.model_dump().items()
if not k in ["actors", "genres", "update_date"]}
movie = mdl.Movie(**movie_data_db,
update_date=strftime("%Y-%m-%d %H:%M:%S"),
owner_id=owner_id)
await MoviesCRUD.change_genres(db, movie, info)