use new download count if available

diff --git a/panikdb/customtags/templatetags/panikdbtags.py b/panikdb/customtags/templatetags/panikdbtags.py
index 4206ccd3fa25280cef0e09859ccaf3915c0b0bcf..69608ec6142e892fa0704113857ef87fe8d99f9a 100644 (file)
--- a/panikdb/customtags/templatetags/panikdbtags.py
+++ b/panikdb/customtags/templatetags/panikdbtags.py
@@ -1,3 +1,5 @@
+import datetime
+
 from django.template import Library
 from django.conf import settings
 from django.core.urlresolvers import reverse
@@ -133,5 +135,8 @@ def is_page(obj):
 
 @register.filter
 def nb_visits(obj):
-    t = DailyStat.objects.filter(soundfile_id=obj.id).aggregate(Sum('nb_visits'))
-    return t['nb_visits__sum'] or '-'
+    if obj.creation_timestamp >= datetime.date(2021, 1, 26):
+        count = obj.download_count
+    else:
+        count = DailyStat.objects.filter(soundfile_id=obj.id).aggregate(Sum('nb_visits'))['nb_visits__sum']
+    return count or '-'