cineclub-site/server/myapi/views/tmdb_views.py

from rest_framework import viewsets
from rest_framework.decorators import action
from rest_framework.response import Response

from tmdbv3api import TMDb, Search, Movie

from ulm_cine_club_api.settings.base import TMDB_API_KEY

tmdb = TMDb()
tmdb.api_key = TMDB_API_KEY
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)