Add audiobooks

database/crud/__init__.py

@@ -1,2 +1,4 @@
 from .games import *
+from .movies import *
+from .audiobooks import *
 from .users import *

database/crud/audiobooks.py

@@ -0,0 +1,45 @@
+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_audiobooks(db: AsyncSession):
+    return (await db.execute(select(mdl.Audiobook))).scalars().all()
+
+
+async def get_audiobook(db: AsyncSession, audiobook_id: int):
+    return await db.get(mdl.Audiobook, audiobook_id)
+
+
+async def add_audiobook(db: AsyncSession,
+                        audiobook_info: sch.AudiobookCreate,
+                        user_id: int):
+    audiobook = mdl.Audiobook(**audiobook_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, audiobook)
+
+
+async def edit_audiobook(db: AsyncSession,
+                         audiobook_id: int,
+                         audiobook_info: sch.AudiobookCreate):
+    audiobook = await db.get(mdl.Audiobook, audiobook_id)
+    for key, value in vars(audiobook_info).items():
+        if (value and value is not None and getattr(audiobook, key) != value):
+            setattr(audiobook, key, value)
+    setattr(audiobook, "update_date", strftime("%Y-%m-%d %H:%M:%S"))
+    await db.commit()
+    return audiobook
+
+
+async def delete_audiobook(db: AsyncSession,
+                           audiobook_id: int):
+    audiobook = await get_audiobook(db, audiobook_id)
+    await db.delete(audiobook)
+    await db.commit()
+    return audiobook

database/crud/games.py

@@ -32,6 +32,7 @@ async def edit_game(db: AsyncSession,
     for key, value in vars(game_info).items():
         if (value and value is not None and getattr(game, key) != value):
             setattr(game, key, value)
+    setattr(game, "update_date", strftime("%Y-%m-%d %H:%M:%S"))
     await db.commit()
     return game
 

database/crud/movies.py

@@ -0,0 +1,45 @@
+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