roypur/stream-api
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

simplify image fetcher

Browse Source
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
1 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
backend/stream.py
View File

@ -226,18 +226,10 @@ class UpstreamHandler():
if self.provider == "youtube":
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)
image_status_standard_future = None
image_status_full_future = None
if full_image != image:
image_status_full_future = session.head(full_image)
if standard_image != 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_standard = await image_status_standard_future
if hasattr(image_status_full, "status") and (image_status_full.status < 400):
data["og:image"] = full_image
elif hasattr(image_status_standard, "status") and (image_status_standard.status < 400):