feat: add template tags

shellfly · Dec 17, 2022 · b2e8eb4 · b2e8eb4
1 parent 89a6f59
commit b2e8eb4
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 5 deletions.
diff --git a/setup.py b/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"
     ],
 )
diff --git a/test/templates/test/comments.html b/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>
diff --git a/vote/templatetags/vote.py b/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)