26 lines
789 B
Python
26 lines
789 B
Python
from rest_framework import viewsets
|
|
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
|
|
from tmdbv3api import TMDb, Search
|
|
tmdb = TMDb()
|
|
tmdb.api_key = "" # TODO configure key in settings
|
|
tmdb.language = "fr"
|
|
|
|
|
|
class TmdbViewSet(viewsets.ViewSet):
|
|
@action(methods=["GET"], detail=False)
|
|
def search(self, request):
|
|
query = request.query_params["query"]
|
|
search = Search()
|
|
tmdb_results = search.movies({"query": query})
|
|
res = [
|
|
{
|
|
"id": film["id"],
|
|
"release_date": film["release_date"],
|
|
"title": film["title"],
|
|
"poster_path": film["poster_path"],
|
|
}
|
|
for film in tmdb_results
|
|
]
|
|
return Response(res)
|