| 1 | import dotenv |
| 2 | import os |
| 3 | import requests |
| 4 | |
| 5 | |
| 6 | dotenv.load_dotenv() |
| 7 | |
| 8 | API_KEY = os.getenv("OMDB_API_KEY") |
| 9 | |
| 10 | BASE_URL = "http://www.omdbapi.com/?apikey=" + API_KEY |
| 11 | |
| 12 | class OMDBClient: |
| 13 | def __init__(self): |
| 14 | self.BASE_URL = BASE_URL |
| 15 | |
| 16 | def get_movie(self, title: str) -> dict | None: |
| 17 | url = self.BASE_URL + "&t=" + title |
| 18 | response = requests.get(url) |
| 19 | if response.status_code == 200 and response.json()["Response"] == "True": |
| 20 | new_movie = {"name": response.json()["Title"], "director": response.json()["Director"], |
| 21 | "year": response.json()["Year"], "poster": response.json()["Poster"], |
| 22 | "country": response.json()["Country"], "genre": response.json()["Genre"], |
| 23 | "plot": response.json()["Plot"], "rating": response.json()["imdbRating"]} |
| 24 | return new_movie |
| 25 | return None |