diff --git a/src/routes/admin.py b/src/routes/admin.py index 7f60ccd..fc493c1 100644 --- a/src/routes/admin.py +++ b/src/routes/admin.py @@ -1,13 +1,14 @@ from fastapi import APIRouter from models import Message -from models.user import UserIn, UserCredentials +from models.user import UserCredentials from models.movie import Movie +from models.card import Card from lib.database.admin import ( add_user, remove_user, add_movie, - remove_movie, + set_movie_watched, add_card, remove_card, ) @@ -19,37 +20,37 @@ router = APIRouter( ) -@router.post("/user") +@router.post("/user", response_model=Message) async def add_user_endpoint(user: UserCredentials): """Add a user to the database""" return add_user(user) -@router.delete("/user") +@router.delete("/user", response_model=Message) async def remove_user_endpoint(user_id: str): """Remove a user from the database""" return remove_user(user_id) -@router.post("/movie") +@router.post("/movie", response_model=Message) async def add_movie_endpoint(movie: Movie): """Add a movie to the database""" - return {"movie": movie} + return add_movie(movie) -@router.delete("/movie") -async def remove_movie_endpoint(movie: str): - """Remove a movie from the database""" - return {"movie": movie} +@router.patch("/movie", response_model=Message) +async def set_movie_watched_endpoint(movie_id: str): + """Set movie as watched""" + return set_movie_watched(movie_id) -@router.post("/card") -async def add_card_endpoint(card: str): +@router.post("/card", response_model=Message) +async def add_card_endpoint(card: Card): """Add a card to the database""" - return {"card": card} + return add_card(card) -@router.delete("/card") -async def remove_card_endpoint(card: str): +@router.delete("/card", response_model=Message) +async def remove_card_endpoint(card_id: str): """Remove a card from the database""" - return {"card": card} + return remove_card(card_id)