NDOJ/judge/admin/organization.py

from django.contrib import admin
from django.forms import ModelForm
from django.urls import reverse_lazy
from django.utils.html import format_html
from django.utils.translation import gettext, gettext_lazy as _
from reversion.admin import VersionAdmin

from judge.models import Organization
from judge.widgets import (
    AdminHeavySelect2MultipleWidget,
    AdminHeavySelect2Widget,
    HeavyPreviewAdminPageDownWidget,
)


class OrganizationForm(ModelForm):
    class Meta:
        widgets = {
            "admins": AdminHeavySelect2MultipleWidget(data_view="profile_select2"),
            "registrant": AdminHeavySelect2Widget(data_view="profile_select2"),
        }
        if HeavyPreviewAdminPageDownWidget is not None:
            widgets["about"] = HeavyPreviewAdminPageDownWidget(
                preview=reverse_lazy("organization_preview")
            )


class OrganizationAdmin(VersionAdmin):
    readonly_fields = ("creation_date",)
    fields = (
        "name",
        "slug",
        "short_name",
        "is_open",
        "about",
        "slots",
        "registrant",
        "creation_date",
        "admins",
    )
    list_display = (
        "name",
        "short_name",
        "is_open",
        "creation_date",
        "registrant",
        "show_public",
    )
    search_fields = ("name", "short_name", "registrant__user__username")
    prepopulated_fields = {"slug": ("name",)}
    actions_on_top = True
    actions_on_bottom = True
    form = OrganizationForm
    ordering = ["-creation_date"]

    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 get_readonly_fields(self, request, obj=None):
        fields = self.readonly_fields
        if not request.user.has_perm("judge.organization_admin"):
            return fields + ("registrant", "admins", "is_open", "slots")
        return fields

    def get_queryset(self, request):
        queryset = Organization.objects.all()
        if request.user.has_perm("judge.edit_all_organization"):
            return queryset
        else:
            return queryset.filter(admins=request.profile.id)

    def has_change_permission(self, request, obj=None):
        if not request.user.has_perm("judge.change_organization"):
            return False
        if request.user.has_perm("judge.edit_all_organization") or obj is None:
            return True
        return obj.admins.filter(id=request.profile.id).exists()

    def save_related(self, request, form, formsets, change):
        super().save_related(request, form, formsets, change)
        obj = form.instance
        obj.members.add(*obj.admins.all())


class OrganizationRequestAdmin(admin.ModelAdmin):
    list_display = ("username", "organization", "state", "time")
    readonly_fields = ("user", "organization")

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

    username.short_description = _("username")
    username.admin_order_field = "user__user__username"