python-markdown-oembed/src/python_markdown_oembed_extension/oembedextension.py

from markdown import Extension
import oembed
from python_markdown_oembed_extension.endpoints import DEFAULT_ENDPOINTS
from python_markdown_oembed_extension.inlinepatterns import OEmbedLinkPattern, OEMBED_LINK_RE


class OEmbedExtension(Extension):

    def __init__(self, **kwargs):
        self.config = {
            'allowed_endpoints': [
                DEFAULT_ENDPOINTS,
                "A list of oEmbed endpoints to allow. "
                "Defaults to endpoints.DEFAULT_ENDPOINTS",
            ],
            'wrapper_class': [
                'oembed',
                "CSS class(es) for the <figure> wrapper element. "
                "Set to empty string to disable wrapping.",
            ],
        }
        super().__init__(**kwargs)

    def extendMarkdown(self, md):
        consumer = self._prepare_oembed_consumer()
        wrapper_class = self.getConfig('wrapper_class', 'oembed')
        link_pattern = OEmbedLinkPattern(
            OEMBED_LINK_RE, md, consumer, wrapper_class=wrapper_class,
        )
        # Priority 175 — run before the default image pattern (priority 150)
        md.inlinePatterns.register(link_pattern, 'oembed_link', 175)

    def _prepare_oembed_consumer(self):
        allowed_endpoints = self.getConfig('allowed_endpoints', DEFAULT_ENDPOINTS)
        consumer = oembed.OEmbedConsumer()
        for endpoint in (allowed_endpoints or []):
            consumer.addEndpoint(endpoint)
        return consumer