angrybeanie_wagtail/env/lib/python3.12/site-packages/wagtail/embeds/finders/facebook.py

104 lines
3.6 KiB
Python
Raw Normal View History

2025-07-25 21:32:16 +10:00
import json
from urllib import request as urllib_request
from urllib.error import HTTPError, URLError
from urllib.parse import urlencode
from urllib.request import Request
from wagtail.embeds.exceptions import EmbedException, EmbedNotFoundException
from .oembed import OEmbedFinder
class AccessDeniedFacebookOEmbedException(EmbedException):
pass
FACEBOOK_PROVIDERS = [
# Videos
{
"endpoint": "https://graph.facebook.com/v11.0/oembed_video",
"urls": [
r"^https://(?:www\.)?facebook\.com/.+?/videos/.+$",
r"^https://(?:www\.)?facebook\.com/video\.php\?(?:v|id)=.+$",
r"^https://fb.watch/.+$",
],
},
# Posts
{
"endpoint": "https://graph.facebook.com/v11.0/oembed_post",
"urls": [
r"^https://(?:www\.)?facebook\.com/.+?/(?:posts|activity)/.+$",
r"^https://(?:www\.)?facebook\.com/photo\.php\?fbid=.+$",
r"^https://(?:www\.)?facebook\.com/(?:photos|questions)/.+$",
r"^https://(?:www\.)?facebook\.com/permalink\.php\?story_fbid=.+$",
r"^https://(?:www\.)?facebook\.com/media/set/?\?set=.+$",
r"^https://(?:www\.)?facebook\.com/notes/.+?/.+?/.+$",
# At the moment, not documented on https://developers.facebook.com/docs/plugins/oembed-endpoints
# Works for posts with a single photo
r"^https://(?:www\.)?facebook\.com/.+?/photos/.+$",
],
},
]
class FacebookOEmbedFinder(OEmbedFinder):
"""
An embed finder that supports the authenticated Facebook oEmbed Endpoint.
https://developers.facebook.com/docs/plugins/oembed
"""
def __init__(self, omitscript=False, app_id=None, app_secret=None):
# {settings.facebook_APP_ID}|{settings.facebook_APP_SECRET}
self.app_id = app_id
self.app_secret = app_secret
self.omitscript = omitscript
super().__init__(providers=FACEBOOK_PROVIDERS)
def find_embed(self, url, max_width=None, max_height=None):
# Find provider
endpoint = self._get_endpoint(url)
if endpoint is None:
raise EmbedNotFoundException
params = {"url": url, "format": "json"}
if max_width:
params["maxwidth"] = max_width
if max_height:
params["maxheight"] = max_height
if self.omitscript:
params["omitscript"] = "true"
# Configure request
request = Request(endpoint + "?" + urlencode(params))
request.add_header("Authorization", f"Bearer {self.app_id}|{self.app_secret}")
# Perform request
try:
r = urllib_request.urlopen(request)
except (HTTPError, URLError) as e:
if isinstance(e, HTTPError) and e.code == 404:
raise EmbedNotFoundException
elif isinstance(e, HTTPError) and e.code in [400, 401, 403]:
raise AccessDeniedFacebookOEmbedException
else:
raise EmbedNotFoundException
oembed = json.loads(r.read().decode("utf-8"))
# Return embed as a dict
return {
"title": oembed["title"] if "title" in oembed else "",
"author_name": oembed["author_name"] if "author_name" in oembed else "",
"provider_name": oembed["provider_name"]
if "provider_name" in oembed
else "Facebook",
"type": oembed["type"],
"thumbnail_url": oembed.get("thumbnail_url"),
"width": oembed.get("width"),
"height": oembed.get("height"),
"html": oembed.get("html"),
}
embed_finder_class = FacebookOEmbedFinder