Fix error for unauthenticated user plan checks

modern-tables
Mark Veidemanis 2 years ago
parent 11b5eb50ec
commit 764a6a48f8
Signed by: m
GPG Key ID: 5ACFCEED46C0904F

@ -1,8 +1,11 @@
from django import template
from core.models import User
register = template.Library()
@register.filter
def has_plan(user, plan_name):
if not hasattr(user, "plans"):
return False
plan_list = [plan.name for plan in user.plans.all()]
return plan_name in plan_list
return plan_name in plan_list

Loading…
Cancel
Save