X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/4b86e623b0ff7a5a53bdb29df06eab039ebe4e1e..HEAD:/src/club/templatetags/club.py

diff --git a/src/club/templatetags/club.py b/src/club/templatetags/club.py
index 1a3416971..79dc14478 100644
--- a/src/club/templatetags/club.py
+++ b/src/club/templatetags/club.py
@@ -1,5 +1,12 @@
+# This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
+#
+from datetime import timedelta
+from django.db.models import Sum
 from django import template
+from django.utils.timezone import now
 from ..helpers import get_active_schedule
+from ..models import Schedule
 
 
 register = template.Library()
@@ -8,3 +15,71 @@ register = template.Library()
 @register.filter
 def active_schedule(user):
     return get_active_schedule(user)
+
+
+@register.simple_tag
+def club_count_recurring():
+    return Schedule.objects.exclude(monthly=False, yearly=False).filter(expires_at__gt=now()).count()
+
+@register.simple_tag
+def club_active_monthly_count():
+    return Schedule.objects.filter(
+        expires_at__gt=now(),
+        monthly=True,
+        is_cancelled=False
+    ).count()
+
+@register.simple_tag
+def club_active_monthly_sum():
+    return Schedule.objects.filter(
+        expires_at__gt=now(),
+        monthly=True,
+        is_cancelled=False
+    ).aggregate(s=Sum('amount'))['s'] or 0
+
+@register.simple_tag
+def club_active_yearly_count():
+    return Schedule.objects.filter(
+        expires_at__gt=now(),
+        yearly=True,
+        is_cancelled=False
+    ).count()
+
+@register.simple_tag
+def club_active_yearly_sum():
+    return Schedule.objects.filter(
+        expires_at__gt=now(),
+        yearly=True,
+        is_cancelled=False
+    ).aggregate(s=Sum('amount'))['s'] or 0
+
+@register.simple_tag
+def club_active_30day_count():
+    return Schedule.objects.filter(
+        yearly=False, monthly=False,
+        payed_at__gte=now() - timedelta(days=30)
+    ).count()
+
+@register.simple_tag
+def club_active_30day_sum():
+    return Schedule.objects.filter(
+        yearly=False, monthly=False,
+        payed_at__gte=now() - timedelta(days=30)
+    ).aggregate(s=Sum('amount'))['s'] or 0
+
+
+@register.simple_tag
+def club_monthly_since(start):
+    return Schedule.objects.filter(
+        monthly=True, payed_at__gte=start).count()
+
+
+@register.simple_tag
+def club_monthly_missing_since(start, target):
+    return target - Schedule.objects.filter(
+        monthly=True, payed_at__gte=start).count()
+
+
+@register.simple_tag(takes_context=True)
+def invite_payment(context, payment_method, schedule):
+    return payment_method.invite_widget(schedule, context['request'])