Spaces:
Running
Running
# Python Markdown | |
# A Python implementation of John Gruber's Markdown. | |
# Documentation: https://python-markdown.github.io/ | |
# GitHub: https://github.com/Python-Markdown/markdown/ | |
# PyPI: https://pypi.org/project/Markdown/ | |
# Started by Manfred Stienstra (http://www.dwerg.net/). | |
# Maintained for a few years by Yuri Takhteyev (http://www.freewisdom.org). | |
# Currently maintained by Waylan Limberg (https://github.com/waylan), | |
# Dmitry Shachnev (https://github.com/mitya57) and Isaac Muse (https://github.com/facelessuser). | |
# Copyright 2007-2023 The Python Markdown Project (v. 1.7 and later) | |
# Copyright 2004, 2005, 2006 Yuri Takhteyev (v. 0.2-1.6b) | |
# Copyright 2004 Manfred Stienstra (the original version) | |
# License: BSD (see LICENSE.md for details). | |
""" | |
Post-processors run on the text of the entire document after is has been serialized into a string. | |
Postprocessors should be used to work with the text just before output. Usually, they are used add | |
back sections that were extracted in a preprocessor, fix up outgoing encodings, or wrap the whole | |
document. | |
""" | |
from __future__ import annotations | |
from collections import OrderedDict | |
from typing import TYPE_CHECKING, Any | |
from . import util | |
import re | |
if TYPE_CHECKING: # pragma: no cover | |
from markdown import Markdown | |
def build_postprocessors(md: Markdown, **kwargs: Any) -> util.Registry[Postprocessor]: | |
""" Build the default postprocessors for Markdown. """ | |
postprocessors = util.Registry() | |
postprocessors.register(RawHtmlPostprocessor(md), 'raw_html', 30) | |
postprocessors.register(AndSubstitutePostprocessor(), 'amp_substitute', 20) | |
return postprocessors | |
class Postprocessor(util.Processor): | |
""" | |
Postprocessors are run after the ElementTree it converted back into text. | |
Each Postprocessor implements a `run` method that takes a pointer to a | |
text string, modifies it as necessary and returns a text string. | |
Postprocessors must extend `Postprocessor`. | |
""" | |
def run(self, text: str) -> str: | |
""" | |
Subclasses of `Postprocessor` should implement a `run` method, which | |
takes the html document as a single text string and returns a | |
(possibly modified) string. | |
""" | |
pass # pragma: no cover | |
class RawHtmlPostprocessor(Postprocessor): | |
""" Restore raw html to the document. """ | |
BLOCK_LEVEL_REGEX = re.compile(r'^\<\/?([^ >]+)') | |
def run(self, text: str) -> str: | |
""" Iterate over html stash and restore html. """ | |
replacements = OrderedDict() | |
for i in range(self.md.htmlStash.html_counter): | |
html = self.stash_to_string(self.md.htmlStash.rawHtmlBlocks[i]) | |
if self.isblocklevel(html): | |
replacements["<p>{}</p>".format( | |
self.md.htmlStash.get_placeholder(i))] = html | |
replacements[self.md.htmlStash.get_placeholder(i)] = html | |
def substitute_match(m: re.Match[str]) -> str: | |
key = m.group(0) | |
if key not in replacements: | |
if key[3:-4] in replacements: | |
return f'<p>{ replacements[key[3:-4]] }</p>' | |
else: | |
return key | |
return replacements[key] | |
if replacements: | |
base_placeholder = util.HTML_PLACEHOLDER % r'([0-9]+)' | |
pattern = re.compile(f'<p>{ base_placeholder }</p>|{ base_placeholder }') | |
processed_text = pattern.sub(substitute_match, text) | |
else: | |
return text | |
if processed_text == text: | |
return processed_text | |
else: | |
return self.run(processed_text) | |
def isblocklevel(self, html: str) -> bool: | |
""" Check is block of HTML is block-level. """ | |
m = self.BLOCK_LEVEL_REGEX.match(html) | |
if m: | |
if m.group(1)[0] in ('!', '?', '@', '%'): | |
# Comment, PHP etc... | |
return True | |
return self.md.is_block_level(m.group(1)) | |
return False | |
def stash_to_string(self, text: str) -> str: | |
""" Convert a stashed object to a string. """ | |
return str(text) | |
class AndSubstitutePostprocessor(Postprocessor): | |
""" Restore valid entities """ | |
def run(self, text: str) -> str: | |
text = text.replace(util.AMP_SUBSTITUTE, "&") | |
return text | |
class UnescapePostprocessor(Postprocessor): | |
""" Restore escaped chars. """ | |
RE = re.compile(r'{}(\d+){}'.format(util.STX, util.ETX)) | |
def unescape(self, m: re.Match[str]) -> str: | |
return chr(int(m.group(1))) | |
def run(self, text: str) -> str: | |
return self.RE.sub(self.unescape, text) | |