simplify image fetcher

This commit is contained in:
Roy Olav Purser 2021-05-27 20:55:08 +02:00
parent c8a4be5897
commit 7c1a99dde8
Signed by: roypur
GPG Key ID: E14D26A036F21656

View File

@ -226,18 +226,10 @@ class UpstreamHandler():
if self.provider == "youtube": if self.provider == "youtube":
full_image = re.sub(r'\/[a-zA-Z0-9]+\.([a-zA-Z0-9]+)$', r'/maxresdefault.\1', image) full_image = re.sub(r'\/[a-zA-Z0-9]+\.([a-zA-Z0-9]+)$', r'/maxresdefault.\1', image)
standard_image = re.sub(r'\/[a-zA-Z0-9]+\.([a-zA-Z0-9]+)$', r'/sddefault.\1', image) standard_image = re.sub(r'\/[a-zA-Z0-9]+\.([a-zA-Z0-9]+)$', r'/sddefault.\1', image)
image_status_standard_future = None
image_status_full_future = None
if full_image != image:
image_status_full_future = session.head(full_image) image_status_full_future = session.head(full_image)
if standard_image != image:
image_status_standard_future = session.head(standard_image) image_status_standard_future = session.head(standard_image)
image_status_full = None
image_status_standard = None
if image_status_standard_future is not None:
image_status_standard = await image_status_standard_future
if image_status_full_future is not None:
image_status_full = await image_status_full_future image_status_full = await image_status_full_future
image_status_standard = await image_status_standard_future
if hasattr(image_status_full, "status") and (image_status_full.status < 400): if hasattr(image_status_full, "status") and (image_status_full.status < 400):
data["og:image"] = full_image data["og:image"] = full_image
elif hasattr(image_status_standard, "status") and (image_status_standard.status < 400): elif hasattr(image_status_standard, "status") and (image_status_standard.status < 400):