Move user_badge to forum module

oioioi/contests/templatetags/get_user_name.py

@@ -4,8 +4,6 @@
 
 from oioioi.base.utils import get_user_display_name
 from oioioi.contests.utils import can_see_personal_data, is_contest_basicadmin
-from oioioi.forum.utils import is_forum_moderator
-from django.utils.translation import gettext_lazy as _
 
 register = template.Library()
 
@@ -79,31 +77,3 @@ def public_name(parser, token):
 def full_name(parser, token):
     (user, asvar) = _get_name(parser, token, "full_name")
     return FullNameUserInfoLinkNode(user, asvar)
-
-
-class BadgeNode(Node):
-    '''
-    Returns a badge for a user if they are a moderator.
-    '''
-    def __init__(self, target_user, asvar):
-        self.target_user = Variable(target_user)
-        self.asvar = asvar
-
-    def render(self, context):
-        user = self.target_user.resolve(context)
-        if is_forum_moderator(context['request'], user):
-            badge = _(" (moderator)")
-        else:
-            badge = ""
-
-        if self.asvar:
-            context[self.asvar] = str(badge)
-            return ""
-        else:
-            return str(badge)
-
-
-@register.tag
-def user_badge(parser, token):
-    (user, asvar) = _get_name(parser, token, "user_badge")
-    return BadgeNode(user, asvar)

oioioi/forum/templates/forum/category.html

@@ -3,6 +3,7 @@
 {% load check_perm %}
 {% load get_user_name %}
 {% load pagination_tags %}
+{% load user_badge %}
 
 {% block title %}{% trans "Category" %}: {{ category.name }}{% endblock %}
 

oioioi/forum/templates/forum/thread-element-footer.html

@@ -1,6 +1,7 @@
 {% load i18n %}
 {% load check_perm %}
 {% load get_user_name %}
+{% load user_badge %}
 
 {% url 'forum_post_edit' contest_id=contest.id category_id=category.id thread_id=thread.id post_id=post.id as forum_post_edit_url %}
 {% url 'forum_post_delete' contest_id=contest.id category_id=category.id thread_id=thread.id post_id=post.id as forum_post_delete_url %}

oioioi/forum/templatetags/user_badge.py

@@ -0,0 +1,37 @@
+from django import template
+from django.template import Node, Variable
+from django.utils.translation import gettext_lazy as _
+
+from oioioi.contests.templatetags.get_user_name import _get_name
+from oioioi.forum.utils import is_forum_moderator
+
+register = template.Library()
+
+
+class BadgeNode(Node):
+    '''
+    Returns a badge for a user if they are a moderator.
+    '''
+
+    def __init__(self, target_user, asvar):
+        self.target_user = Variable(target_user)
+        self.asvar = asvar
+
+    def render(self, context):
+        user = self.target_user.resolve(context)
+        if is_forum_moderator(context['request'], user):
+            badge = _(" (moderator)")
+        else:
+            badge = ""
+
+        if self.asvar:
+            context[self.asvar] = str(badge)
+            return ""
+        else:
+            return str(badge)
+
+
+@register.tag
+def user_badge(parser, token):
+    (user, asvar) = _get_name(parser, token, "user_badge")
+    return BadgeNode(user, asvar)