diff --git a/files/helpers/alerts.py b/files/helpers/alerts.py index 1a08b613d..3fff33cf5 100644 --- a/files/helpers/alerts.py +++ b/files/helpers/alerts.py @@ -147,7 +147,7 @@ def send_unblock_notif(vid, user, text): def send_pm(vid, user, text): - text_html = mistletoe.Document(text) + text_html = Renderer().render(mistletoe.Document(text)) text_html = sanitize(text_html, True) @@ -172,7 +172,7 @@ def send_admin(vid, text): text = text.replace("\n", "\n\n").replace("\n\n\n\n\n\n", "\n\n").replace("\n\n\n\n", "\n\n").replace("\n\n\n", "\n\n") - text_html = mistletoe.Document(text) + text_html = Renderer().render(mistletoe.Document(text)) text_html = sanitize(text_html, True) diff --git a/files/helpers/markdown.py b/files/helpers/markdown.py index 8930ac8f4..8156f7d07 100644 --- a/files/helpers/markdown.py +++ b/files/helpers/markdown.py @@ -94,6 +94,29 @@ class CustomRenderer(HTMLRenderer): target = token.target[1] return f'{space}r/{target}' + def render_redditor_mention(self, token): + space = token.target[0] + target = token.target[1] + return f'{space}u/{target}' + + +class Renderer(HTMLRenderer): + + def __init__(self, **kwargs): + super().__init__(SubMention, + RedditorMention, + SubMention2, + RedditorMention2, + ) + + for i in kwargs: + self.__dict__[i] = kwargs[i] + + def render_sub_mention(self, token): + space = token.target[0] + target = token.target[1] + return f'{space}r/{target}' + def render_redditor_mention(self, token): space = token.target[0] target = token.target[1] diff --git a/files/routes/users.py b/files/routes/users.py index 9e8999b26..d1a549cb1 100644 --- a/files/routes/users.py +++ b/files/routes/users.py @@ -256,7 +256,7 @@ def messagereply(v): if existing.parent_comment_id: return redirect(f'/notifications?messages=true#comment-{existing.parent_comment_id}') else: return redirect(f'/notifications?messages=true#comment-{existing.id}') - text_html = mistletoe.Document(message) + text_html = Renderer().render(mistletoe.Document(message)) text_html = sanitize(text_html, True) new_comment = Comment(author_id=v.id, parent_submission=None,