Misago/misago/markup/md/shortimgs.py

import re

import markdown
from markdown.inlinepatterns import LinkPattern
from markdown.util import etree


IMAGES_RE =  r'\!(\s?)\((<.*?>|([^\)]*))\)'


class ShortImagesExtension(markdown.Extension):
    def extendMarkdown(self, md):
        md.registerExtension(self)
        md.inlinePatterns.add('misago_short_images',
                              ShortImagePattern(IMAGES_RE, md),
                              '_end')


class ShortImagePattern(LinkPattern):
    def handleMatch(self, m):
        img_src = m.groups()[2].strip()
        if img_src:
            img = etree.Element("img")
            img.set('src', img_src)
            img.set('alt', img_src)
            return img