diff --git a/files/helpers/markdown.py b/files/helpers/markdown.py index 6505bb985..e817e1c3a 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