python-markdown-oembed/mdx_oembed/inlinepatterns.py
Sami Turcotte 3adab335ec Add support for python-markdown safe mode
Mark the resulting OEmbed html tags as safe so they don't get stripped
away in safe mode.
2015-04-03 01:22:11 -04:00

34 lines
956 B
Python

# -*- coding: utf-8 -*-
import logging
from markdown.inlinepatterns import Pattern
import oembed
LOG = logging.getLogger(__name__)
OEMBED_LINK_RE = r'\!\[([^\]]*)\]\(((?:https?:)?//[^\)]*)' \
r'(?<!png)(?<!jpg)(?<!jpeg)(?<!gif)\)'
class OEmbedLinkPattern(Pattern):
def __init__(self, pattern, markdown_instance=None, oembed_consumer=None):
Pattern.__init__(self, pattern, markdown_instance)
self.consumer = oembed_consumer
def handleMatch(self, match):
html = self.get_oembed_html_for_match(match)
if html is None:
return None
placeholder = self.markdown.htmlStash.store(html, True)
return placeholder
def get_oembed_html_for_match(self, match):
url = match.group(3).strip()
try:
response = self.consumer.embed(url)
except oembed.OEmbedNoEndpoint:
return None
else:
return response['html']