Add Movie endpoint

This commit is contained in:
Esa Kataja
2024-11-09 18:32:28 +02:00
parent 88e1d1c54b
commit 60ff8a296d
3 changed files with 108 additions and 13 deletions
+22 -3
View File
@@ -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()