mirror of
https://github.com/Hopiu/python-markdown-oembed.git
synced 2026-03-16 22:10:24 +00:00
23 lines
815 B
Python
23 lines
815 B
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
from markdown import Extension
|
||
|
|
import oembed
|
||
|
|
from mdx_oembed.endpoints import ENDPOINTS
|
||
|
|
from mdx_oembed.inlinepatterns import OEmbedLinkPattern, OEMBED_LINK_RE
|
||
|
|
|
||
|
|
|
||
|
|
class OEmbedExtension(Extension):
|
||
|
|
|
||
|
|
def extendMarkdown(self, md, md_globals):
|
||
|
|
self.oembed_consumer = self.prepare_oembed_consumer()
|
||
|
|
pattern = OEmbedLinkPattern(OEMBED_LINK_RE, md, self.oembed_consumer)
|
||
|
|
md.inlinePatterns.add('oembed_link', pattern, '<image_link')
|
||
|
|
|
||
|
|
def prepare_oembed_consumer(self):
|
||
|
|
allowed_endpoints = self.getConfig('allowed_endpoints',
|
||
|
|
ENDPOINTS.keys())
|
||
|
|
consumer = oembed.OEmbedConsumer()
|
||
|
|
[consumer.addEndpoint(v)
|
||
|
|
for k,v in ENDPOINTS.items()
|
||
|
|
if k in allowed_endpoints]
|
||
|
|
return consumer
|