NDOJ/judge/admin/profile.py

from django.contrib import admin
from django.forms import ModelForm
from django.utils.html import format_html
from django.utils.translation import gettext, gettext_lazy as _, ungettext
from reversion.admin import VersionAdmin
from django.contrib.auth.admin import UserAdmin as OldUserAdmin

from django_ace import AceWidget
from judge.models import Profile
from judge.widgets import AdminPagedownWidget, AdminSelect2Widget


class ProfileForm(ModelForm):
    def __init__(self, *args, **kwargs):
        super(ProfileForm, self).__init__(*args, **kwargs)
        if "current_contest" in self.base_fields:
            # form.fields['current_contest'] does not exist when the user has only view permission on the model.
            self.fields[
                "current_contest"
            ].queryset = self.instance.contest_history.select_related("contest").only(
                "contest__name", "user_id", "virtual"
            )
            self.fields["current_contest"].label_from_instance = (
                lambda obj: "%s v%d" % (obj.contest.name, obj.virtual)
                if obj.virtual
                else obj.contest.name
            )

    class Meta:
        widgets = {
            "timezone": AdminSelect2Widget,
            "language": AdminSelect2Widget,
            "ace_theme": AdminSelect2Widget,
            "current_contest": AdminSelect2Widget,
        }
        if AdminPagedownWidget is not None:
            widgets["about"] = AdminPagedownWidget


class TimezoneFilter(admin.SimpleListFilter):
    title = _("timezone")
    parameter_name = "timezone"

    def lookups(self, request, model_admin):
        return (
            Profile.objects.values_list("timezone", "timezone")
            .distinct()
            .order_by("timezone")
        )

    def queryset(self, request, queryset):
        if self.value() is None:
            return queryset
        return queryset.filter(timezone=self.value())


class ProfileAdmin(VersionAdmin):
    fields = (
        "user",
        "display_rank",
        "about",
        "organizations",
        "timezone",
        "language",
        "ace_theme",
        "math_engine",
        "last_access",
        "ip",
        "mute",
        "is_unlisted",
        "is_banned_problem_voting",
        "notes",
        "is_totp_enabled",
        "user_script",
        "current_contest",
    )
    readonly_fields = ("user",)
    list_display = (
        "admin_user_admin",
        "email",
        "is_totp_enabled",
        "timezone_full",
        "date_joined",
        "last_access",
        "ip",
        "show_public",
    )
    ordering = ("user__username",)
    search_fields = ("user__username", "ip", "user__email")
    list_filter = ("language", TimezoneFilter)
    actions = ("recalculate_points",)
    actions_on_top = True
    actions_on_bottom = True
    form = ProfileForm

    def get_queryset(self, request):
        return super(ProfileAdmin, self).get_queryset(request).select_related("user")

    def get_fields(self, request, obj=None):
        if request.user.has_perm("judge.totp"):
            fields = list(self.fields)
            fields.insert(fields.index("is_totp_enabled") + 1, "totp_key")
            return tuple(fields)
        else:
            return self.fields

    def get_readonly_fields(self, request, obj=None):
        fields = self.readonly_fields
        if not request.user.has_perm("judge.totp"):
            fields += ("is_totp_enabled",)
        return fields

    def show_public(self, obj):
        return format_html(
            '<a href="{0}" style="white-space:nowrap;">{1}</a>',
            obj.get_absolute_url(),
            gettext("View on site"),
        )

    show_public.short_description = ""

    def admin_user_admin(self, obj):
        return obj.username

    admin_user_admin.admin_order_field = "user__username"
    admin_user_admin.short_description = _("User")

    def email(self, obj):
        return obj.email

    email.admin_order_field = "user__email"
    email.short_description = _("Email")

    def timezone_full(self, obj):
        return obj.timezone

    timezone_full.admin_order_field = "timezone"
    timezone_full.short_description = _("Timezone")

    def date_joined(self, obj):
        return obj.user.date_joined

    date_joined.admin_order_field = "user__date_joined"
    date_joined.short_description = _("date joined")

    def recalculate_points(self, request, queryset):
        count = 0
        for profile in queryset:
            profile.calculate_points()
            count += 1
        self.message_user(
            request,
            ungettext(
                "%d user have scores recalculated.",
                "%d users have scores recalculated.",
                count,
            )
            % count,
        )

    recalculate_points.short_description = _("Recalculate scores")

    def get_form(self, request, obj=None, **kwargs):
        form = super(ProfileAdmin, self).get_form(request, obj, **kwargs)
        if "user_script" in form.base_fields:
            # form.base_fields['user_script'] does not exist when the user has only view permission on the model.
            form.base_fields["user_script"].widget = AceWidget(
                "javascript", request.profile.ace_theme
            )
        return form


class UserAdmin(OldUserAdmin):
    # Customize the fieldsets for adding and editing users
    fieldsets = (
        (None, {"fields": ("username", "password")}),
        ("Personal Info", {"fields": ("first_name", "last_name", "email")}),
        (
            "Permissions",
            {
                "fields": (
                    "is_active",
                    "is_staff",
                    "is_superuser",
                    "groups",
                    "user_permissions",
                )
            },
        ),
        ("Important dates", {"fields": ("last_login", "date_joined")}),
    )

    readonly_fields = ("last_login", "date_joined")

    def get_readonly_fields(self, request, obj=None):
        fields = self.readonly_fields
        if not request.user.is_superuser:
            fields += (
                "is_staff",
                "is_active",
                "is_superuser",
                "groups",
                "user_permissions",
            )
        return fields