Files
MediaManager-maxdorninger-1/media_manager/metadataProvider/utils.py
2025-05-29 15:36:35 +02:00

23 lines
653 B
Python

import mimetypes
import requests
def get_year_from_first_air_date(first_air_date: str | None) -> int | None:
if first_air_date:
return int(first_air_date.split("-")[0])
else:
return None
def download_poster_image(storage_path=None, poster_url=None, show=None) -> bool:
res = requests.get(poster_url, stream=True)
content_type = res.headers["content-type"]
file_extension = mimetypes.guess_extension(content_type)
if res.status_code == 200:
with open(storage_path.joinpath(str(show.id) + file_extension), "wb") as f:
f.write(res.content)
return True
else:
return False