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

verify youtube thumbnail size

Browse Source
This commit is contained in:
Roy Olav Purser 2021-05-25 13:00:19 +02:00
parent 1647f35df1
commit 51530c2a6b
Signed by: roypur
GPG Key ID: E14D26A036F21656
1 changed files with 20 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
backend/stream.py
View File

@ -209,11 +209,8 @@ class UpstreamHandler():
text_upstream_future = resp_upstream.text()
text_embed = await text_embed_future
text_upstream = await text_upstream_future
parser = MetaParser()
parser.feed(text_upstream)
data_upstream = parser.meta_data
data_raw = json.loads(text_embed)
if isinstance(data_raw, dict):
data_new = {}
@ -225,11 +222,12 @@ class UpstreamHandler():
data_new["og:video:width"] = data_raw.get("width")
data_new["og:image:height"] = data_raw.get("thumbnail_height")
data_new["og:image:width"] = data_raw.get("thumbnail_width")
data_filtered = data_upstream
data_filtered = {}
for key in data_new:
value = data_new.get(key)
if isinstance(value, str):
data_filtered[key] = value
data_filtered.update(parser.meta_data)
data = data_filtered
except Exception as e:
logger.info(e)
@ -367,14 +365,24 @@ class MainHandler(tornado.web.RequestHandler):
if handler.provider == "nextcloud":
upstream = handler.upstream + "/download"
else:
meta = await handler.meta()
image = meta.get("og:image")
if isinstance(image, str):
if handler.provider == "youtube" and image.endswith("hqdefault.jpg"):
image = image.removesuffix("hqdefault.jpg") + "maxresdefault.jpg"
image = await handler.proxy.proxy_url(image, None)
if isinstance(image, str):
self.set_header("Custom-Poster", image)
if not redir:
meta = await handler.meta()
image = meta.get("og:image")
if isinstance(image, str):
if handler.provider == "youtube":
full_image = None
image_type = None
if image.endswith("hqdefault.jpg"):
full_image = image.removesuffix("hqdefault.jpg") + "maxresdefault.jpg"
elif image.endswith("hqdefault.webp"):
full_image = image.removesuffix("hqdefault.webp") + "maxresdefault.webp"
if isinstance(full_image, str):
image_type = await handler.proxy.content_type(full_image)
if isinstance(image_type, str) and image_type.startswith("image"):
image = full_image
image = await handler.proxy.proxy_url(image, None)
if isinstance(image, str):
self.set_header("Custom-Poster", image)
upstream = await stream_providers.get_any(handler.upstream, handler.proxy, logger)
if upstream is None:
logger.info(f'invalid upstream ({handler.provider})')