from models.streams import VideoFile from requests import get from dotenv import load_dotenv from os import getenv from datetime import datetime load_dotenv() def get_movie_details(VideoFile: VideoFile) -> VideoFile: url = "https://www.omdbapi.com/" params = { "i": VideoFile.imdb, "apikey": getenv("API_KEY"), } response = get(url, params=params) if response.status_code == 200: VideoFile.title = response.json()["Title"] VideoFile.date_released = datetime.strptime( response.json()["Released"], "%d %b %Y" ) return VideoFile