feat: add template tags

Author: shellfly

setup.py

@@ -38,5 +38,7 @@
         "Programming Language :: Python :: 3.7",
         "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10"
+        "Programming Language :: Python :: 3.11"
     ],
 )

test/templates/test/comments.html

@@ -1,4 +1,9 @@
 {% load vote %}
-{% for comment in comments %}
-{{comment.content}} {% vote_exists comment user %}
-{% endfor %}
+<ol>
+    {% for comment in comments %}
+    <li>
+        {{comment.content}} - up:{% vote_count comment "up" %} - down: {% vote_count comment "down" %} - exists_up:
+        {%vote_exists comment user %} - exists_down: {% vote_exists comment user "down"%}
+    </li>
+    {% endfor %}
+</ol>

vote/templatetags/vote.py

@@ -3,13 +3,26 @@
 from django import template
 from django.contrib.auth.models import AnonymousUser
 
-from vote.models import UP
+from vote.models import UP, DOWN
 
 register = template.Library()
 
 
 @register.simple_tag
-def vote_exists(model, user=AnonymousUser(), action=UP):
+def vote_exists(model, user=AnonymousUser(), action="UP"):
     if user.is_anonymous:
         return False
+    if action.lower() == "up":
+        action = UP
+    else:
+        action = DOWN
     return model.votes.exists(user.pk, action=action)
+
+
+@register.simple_tag
+def vote_count(model, action="UP"):
+    if action.lower() == "up":
+        action = UP
+    else:
+        action = DOWN
+    return model.votes.count(action=action)