markdown_wrapper.py

"""Markdown Preview wrapper."""
from __future__ import absolute_import
import traceback
from markdown import Markdown, util, version_info
from markdown.extensions import Extension


class StMarkdown(Markdown):
    """A wrapper around "markdown" lib."""

    def __init__(self, *args, **kwargs):
        """Initialize."""
        Markdown.__init__(self, *args, **kwargs)
        self.Meta = {}

    def registerExtensions(self, extensions, configs):  # noqa
        """
        Register extensions with this instance of Markdown.

        Keyword arguments:
        * `extensions`: A list of extensions, which can either
           be strings or objects.  See the docstring on Markdown.
        * `configs`: A dictionary mapping module names to config options.

        We are overriding this in order to gracefully handle bad extensions
        and to prevent old deprecated style of 'extensions(option=value)'.
        """
        md3 = version_info[0] > 2
        for ext in extensions:
            try:
                # Make sure we aren't using old form `extension(option=value)`
                if isinstance(ext, util.string_type) and ('(' not in ext):
                    ext = self.build_extension(ext, configs.get(ext, []))
                if isinstance(ext, Extension):
                    if md3:
                        ext._extendMarkdown(self)
                    else:
                        ext.extendMarkdown(self, globals())
                elif ext is not None:
                    raise TypeError(
                        'Extension "%s.%s" must be of type: "markdown.Extension"'
                        % (ext.__class__.__module__, ext.__class__.__name__))
            except Exception:
                print(str(traceback.format_exc()))

        return self