diff --git a/apps/registration/tables.py b/apps/registration/tables.py index 274369fe..9b124ffa 100644 --- a/apps/registration/tables.py +++ b/apps/registration/tables.py @@ -4,6 +4,8 @@ import django_tables2 as tables from django.contrib.auth.models import User +from treasury.models import SogeCredit + class FutureUserTable(tables.Table): """ @@ -21,6 +23,7 @@ class FutureUserTable(tables.Table): fields = ('last_name', 'first_name', 'username', 'email', ) model = User row_attrs = { - 'class': 'table-row', + 'class': lambda record: 'table-row' + + (' bg-warning' if SogeCredit.objects.filter(user=record).exists() else ''), 'data-href': lambda record: record.pk } diff --git a/apps/registration/views.py b/apps/registration/views.py index 1ec069a4..3c2ea0a7 100644 --- a/apps/registration/views.py +++ b/apps/registration/views.py @@ -235,7 +235,7 @@ class FutureUserDetailView(ProtectQuerysetMixin, LoginRequiredMixin, FormMixin, fee += 8000 ctx["total_fee"] = "{:.02f}".format(fee / 100, ) - ctx["declare_soge_account"] = True + ctx["declare_soge_account"] = SogeCredit.objects.filter(user=user).exists() return ctx