mirror of
https://github.com/StepanovPlaton/torrent_backend.git
synced 2026-04-03 20:30:38 +04:00
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
from time import strftime
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy.future import select
|
|
|
|
from .. import models as mdl
|
|
from .. import schemas as sch
|
|
from ..database import add_transaction
|
|
|
|
|
|
async def get_movies(db: AsyncSession):
|
|
return (await db.execute(select(mdl.Movie))).scalars().all()
|
|
|
|
|
|
async def get_movie(db: AsyncSession, movie_id: int):
|
|
return await db.get(mdl.Movie, movie_id)
|
|
|
|
|
|
async def add_movie(db: AsyncSession,
|
|
movie_info: sch.MovieCreate,
|
|
user_id: int):
|
|
movie = mdl.Movie(**movie_info.model_dump(),
|
|
update_date=strftime("%Y-%m-%d %H:%M:%S"),
|
|
upload_date=strftime("%Y-%m-%d %H:%M:%S"),
|
|
owner_id=user_id)
|
|
return await add_transaction(db, movie)
|
|
|
|
|
|
async def edit_movie(db: AsyncSession,
|
|
movie_id: int,
|
|
movie_info: sch.MovieCreate):
|
|
movie = await db.get(mdl.Movie, movie_id)
|
|
for key, value in vars(movie_info).items():
|
|
if (value and value is not None and getattr(movie, key) != value):
|
|
setattr(movie, key, value)
|
|
setattr(movie, "update_date", strftime("%Y-%m-%d %H:%M:%S"))
|
|
await db.commit()
|
|
return movie
|
|
|
|
|
|
async def delete_movie(db: AsyncSession,
|
|
movie_id: int):
|
|
movie = await get_movie(db, movie_id)
|
|
await db.delete(movie)
|
|
await db.commit()
|
|
return movie
|