From a8f002a7d166fe5902f6ff62cb08716c7f39b2d2 Mon Sep 17 00:00:00 2001 From: Aevann1 Date: Fri, 24 Dec 2021 04:24:23 +0200 Subject: [PATCH] dfs --- files/helpers/markdown.py | 26 +++++++++++++++++++++++--- 1 file changed, 23 insertions(+), 3 deletions(-) diff --git a/files/helpers/markdown.py b/files/helpers/markdown.py index 6505bb9852..e817e1c3a0 100644 --- a/files/helpers/markdown.py +++ b/files/helpers/markdown.py @@ -122,10 +122,30 @@ class Renderer(HTMLRenderer): target = token.target[1] return f'{space}u/{target}' -class Renderer2(HTMLRenderer): +class Renderer(HTMLRenderer): def __init__(self, **kwargs): - super().__init__() + super().__init__(UserMention, + SubMention, + RedditorMention, + SubMention2, + RedditorMention2, + ) for i in kwargs: - self.__dict__[i] = kwargs[i] \ No newline at end of file + self.__dict__[i] = kwargs[i] + + def render_user_mention(self, token): + space = token.target[0] + target = token.target[1] + return f"{space}@{target}" + + 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] + return f"{space}u/{target}" \ No newline at end of file