Tweak media caching

This commit is contained in:
Thomas Sileo 2019-07-05 10:41:25 +02:00
parent 9a037b132a
commit abbcebb81d

View file

@ -15,8 +15,6 @@ def load(url, user_agent):
"""Initializes a `PIL.Image` from the URL.""" """Initializes a `PIL.Image` from the URL."""
with requests.get(url, stream=True, headers={"User-Agent": user_agent}) as resp: with requests.get(url, stream=True, headers={"User-Agent": user_agent}) as resp:
resp.raise_for_status() resp.raise_for_status()
if not resp.headers.get("content-type").startswith("image/"):
raise ValueError(f"bad content-type {resp.headers.get('content-type')}")
resp.raw.decode_content = True resp.raw.decode_content = True
return Image.open(BytesIO(resp.raw.read())) return Image.open(BytesIO(resp.raw.read()))