Add Movie endpoint
This commit is contained in:
+22
-3
@@ -1,4 +1,8 @@
|
||||
from fastapi import APIRouter
|
||||
from fastapi import APIRouter, Response
|
||||
|
||||
from lib.database.movie import get_all_movies, get_next_movie
|
||||
from models.movie import Movie
|
||||
from models import Message
|
||||
|
||||
router = APIRouter(
|
||||
prefix="/movie",
|
||||
@@ -7,6 +11,21 @@ router = APIRouter(
|
||||
)
|
||||
|
||||
|
||||
@router.get("/")
|
||||
@router.get("/", response_model=list[Movie])
|
||||
async def get_movies():
|
||||
return {"movies": ["movie1", "movie2", "movie3"]}
|
||||
return get_all_movies()
|
||||
|
||||
|
||||
@router.get("/next", response_model=Movie)
|
||||
async def get_next_movie_endpoint():
|
||||
"""Get the next movie to watch"""
|
||||
next_movie = get_next_movie()
|
||||
if next_movie is None:
|
||||
return Response(
|
||||
status_code=200,
|
||||
media_type="application/json",
|
||||
content=Message(
|
||||
message="No more movies to watch", message_type="info"
|
||||
).model_dump_json(),
|
||||
)
|
||||
return get_next_movie()
|
||||
|
||||
Reference in New Issue
Block a user