first commit
This commit is contained in:
33
src/services/tmdb_service.py
Normal file
33
src/services/tmdb_service.py
Normal file
@@ -0,0 +1,33 @@
|
||||
import os
|
||||
import requests
|
||||
from dotenv import load_dotenv
|
||||
|
||||
load_dotenv()
|
||||
|
||||
class TMDBService:
|
||||
def __init__(self):
|
||||
self.api_key = os.getenv('TMDB_API_KEY')
|
||||
self.access_token = os.getenv('TMDB_ACCESS_TOKEN')
|
||||
self.base_url = 'https://api.themoviedb.org/3'
|
||||
|
||||
def search_movie(self, title, year):
|
||||
url = f"{self.base_url}/search/movie"
|
||||
params = {
|
||||
'api_key': self.api_key,
|
||||
'query': title,
|
||||
'year': year
|
||||
}
|
||||
response = requests.get(url, params=params)
|
||||
response_data = response.json()
|
||||
if response_data['results']:
|
||||
return response_data['results'][0]
|
||||
return None
|
||||
|
||||
def get_movie_posters(self, movie_id):
|
||||
url = f"{self.base_url}/movie/{movie_id}/images?include_image_language=en"
|
||||
headers = {
|
||||
"accept": "application/json",
|
||||
"Authorization": f"Bearer {self.access_token}"
|
||||
}
|
||||
response = requests.get(url, headers=headers)
|
||||
return response.json().get('posters', [])
|
||||
Reference in New Issue
Block a user