100 lines
3.1 KiB
Python
100 lines
3.1 KiB
Python
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",
|
|
"logo_override_image",
|
|
"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"
|