zulip/zerver/lib/url_preview/oembed.py

52 lines
1.5 KiB
Python
Raw Normal View History

import json
from typing import Any, Dict, Optional
from pyoembed import PyOembedException, oEmbed
def get_oembed_data(
url: str, maxwidth: int = 640, maxheight: int = 480
) -> Optional[Dict[str, Any]]:
try:
data = oEmbed(url, maxwidth=maxwidth, maxheight=maxheight)
except (PyOembedException, json.decoder.JSONDecodeError):
return None
oembed_resource_type = data.get("type", "")
image = data.get("url", data.get("image"))
thumbnail = data.get("thumbnail_url")
html = data.pop("html", "")
if oembed_resource_type == "photo" and image:
return dict(
oembed=True,
image=image,
type=oembed_resource_type,
title=data.get("title"),
description=data.get("description"),
)
if oembed_resource_type == "video" and html and thumbnail:
return dict(
oembed=True,
image=thumbnail,
type=oembed_resource_type,
html=strip_cdata(html),
title=data.get("title"),
description=data.get("description"),
)
# Otherwise, start with just the embed type.
return dict(
type=oembed_resource_type,
title=data.get("title"),
description=data.get("description"),
)
def strip_cdata(html: str) -> str:
# Work around a bug in SoundCloud's XML generation:
# <html>&lt;![CDATA[&lt;iframe ...&gt;&lt;/iframe&gt;]]&gt;</html>
if html.startswith("<![CDATA[") and html.endswith("]]>"):
html = html[9:-3]
return html