Show More
markdown_ext.py
106 lines
| 3.4 KiB
| text/x-python
|
PythonLexer
r1 | # -*- coding: utf-8 -*- | |||
r2487 | # Copyright (C) 2010-2018 RhodeCode GmbH | |||
r1 | # | |||
# This program is free software: you can redistribute it and/or modify | ||||
# it under the terms of the GNU Affero General Public License, version 3 | ||||
# (only), as published by the Free Software Foundation. | ||||
# | ||||
# This program is distributed in the hope that it will be useful, | ||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||||
# GNU General Public License for more details. | ||||
# | ||||
# You should have received a copy of the GNU Affero General Public License | ||||
# along with this program. If not, see <http://www.gnu.org/licenses/>. | ||||
# | ||||
# This program is dual-licensed. If you wish to learn more about the | ||||
# RhodeCode Enterprise Edition, including its added features, Support services, | ||||
# and proprietary license terms, please see https://rhodecode.com/licenses/ | ||||
import re | ||||
import markdown | ||||
r317 | from mdx_gfm import GithubFlavoredMarkdownExtension # noqa | |||
r1 | ||||
class FlavoredCheckboxExtension(markdown.Extension): | ||||
def extendMarkdown(self, md, md_globals): | ||||
md.preprocessors.add('checklist', | ||||
FlavoredCheckboxPreprocessor(md), '<reference') | ||||
md.postprocessors.add('checklist', | ||||
FlavoredCheckboxPostprocessor(md), '>unescape') | ||||
class FlavoredCheckboxPreprocessor(markdown.preprocessors.Preprocessor): | ||||
""" | ||||
Replaces occurrences of [ ] or [x] to checkbox input | ||||
""" | ||||
pattern = re.compile(r'^([*-]) \[([ x])\]') | ||||
def run(self, lines): | ||||
return [self._transform_line(line) for line in lines] | ||||
def _transform_line(self, line): | ||||
return self.pattern.sub(self._replacer, line) | ||||
def _replacer(self, match): | ||||
list_prefix, state = match.groups() | ||||
checked = '' if state == ' ' else ' checked="checked"' | ||||
return '%s <input type="checkbox" disabled="disabled"%s>' % (list_prefix, | ||||
checked) | ||||
class FlavoredCheckboxPostprocessor(markdown.postprocessors.Postprocessor): | ||||
""" | ||||
Adds `flavored_checkbox_list` class to list of checkboxes | ||||
""" | ||||
pattern = re.compile(r'^([*-]) \[([ x])\]') | ||||
def run(self, html): | ||||
before = '<ul>\n<li><input type="checkbox"' | ||||
after = '<ul class="flavored_checkbox_list">\n<li><input type="checkbox"' | ||||
return html.replace(before, after) | ||||
# Global Vars | ||||
URLIZE_RE = '(%s)' % '|'.join([ | ||||
r'<(?:f|ht)tps?://[^>]*>', | ||||
r'\b(?:f|ht)tps?://[^)<>\s]+[^.,)<>\s]', | ||||
r'\bwww\.[^)<>\s]+[^.,)<>\s]', | ||||
r'[^(<\s]+\.(?:com|net|org)\b', | ||||
]) | ||||
r317 | ||||
r1 | class UrlizePattern(markdown.inlinepatterns.Pattern): | |||
""" Return a link Element given an autolink (`http://example/com`). """ | ||||
def handleMatch(self, m): | ||||
url = m.group(2) | ||||
if url.startswith('<'): | ||||
url = url[1:-1] | ||||
text = url | ||||
if not url.split('://')[0] in ('http','https','ftp'): | ||||
if '@' in url and not '/' in url: | ||||
url = 'mailto:' + url | ||||
else: | ||||
url = 'http://' + url | ||||
el = markdown.util.etree.Element("a") | ||||
el.set('href', url) | ||||
el.text = markdown.util.AtomicString(text) | ||||
return el | ||||
class UrlizeExtension(markdown.Extension): | ||||
""" Urlize Extension for Python-Markdown. """ | ||||
def extendMarkdown(self, md, md_globals): | ||||
""" Replace autolink with UrlizePattern """ | ||||
md.inlinePatterns['autolink'] = UrlizePattern(URLIZE_RE, md) | ||||