/
kaleo_tags.py
37 lines (26 loc) · 945 Bytes
/
kaleo_tags.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
from django import template
from kaleo.forms import InviteForm
from kaleo.models import InvitationStat
register = template.Library()
@register.inclusion_tag("kaleo/_invites_remaining.html")
def invites_remaining(user):
try:
remaining = user.invitationstat.invites_remaining()
except InvitationStat.DoesNotExist:
remaining = 0
return {"invites_remaining": remaining}
@register.inclusion_tag("kaleo/_invite_form.html")
def invite_form(user):
return {"form": InviteForm(user=user), "user": user}
@register.inclusion_tag("kaleo/_invited.html")
def invites_sent(user):
return {"invited_list": user.invites_sent.all()}
@register.filter
def status_class(invite):
if invite.status == invite.STATUS_SENT:
return "sent"
elif invite.status == invite.STATUS_ACCEPTED:
return "accepted"
elif invite.status == invite.STATUS_JOINED_INDEPENDENTLY:
return "joined"
return ""