# -*- coding: utf-8 -*- # Copyright (C) 2010-2018 RhodeCode GmbH # # 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 . # # 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 from mdx_gfm import GithubFlavoredMarkdownExtension # noqa class FlavoredCheckboxExtension(markdown.Extension): def extendMarkdown(self, md, md_globals): md.preprocessors.add('checklist', FlavoredCheckboxPreprocessor(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 ' % (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 = '