Add Messages to response models

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")
+@router.patch("/movie", response_model=Message)
-async def remove_movie_endpoint(movie: str):
+async def set_movie_watched_endpoint(movie_id: str):
-    """Remove a movie from the database"""
+    """Set movie as watched"""
-    return {"movie": movie}
+    return set_movie_watched(movie_id)
 
 
-@router.post("/card")
+@router.post("/card", response_model=Message)
-async def add_card_endpoint(card: str):
+async def add_card_endpoint(card: Card):
     """Add a card to the database"""
-    return {"card": card}
+    return add_card(card)
 
 
-@router.delete("/card")
+@router.delete("/card", response_model=Message)
-async def remove_card_endpoint(card: str):
+async def remove_card_endpoint(card_id: str):
     """Remove a card from the database"""
-    return {"card": card}
+    return remove_card(card_id)