67 lines
2.6 KiB
Python
67 lines
2.6 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', 'slots', 'registrant', 'show_public')
|
||
|
|
prepopulated_fields = {'slug': ('name',)}
|
||
|
|
actions_on_top = True
|
||
|
|
actions_on_bottom = True
|
||
|
|
form = OrganizationForm
|
||
|
|
|
||
|
|
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()
|
||
|
|
|
||
|
|
|
||
|
|
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'
|