from django.contrib import admin, messages from django.utils.translation import gettext_lazy as _ from orchestra.admin import ExtendedModelAdmin, ChangePasswordAdminMixin from orchestra.admin.actions import disable, enable from orchestra.contrib.accounts.actions import list_accounts from orchestra.contrib.accounts.admin import SelectAccountAdminMixin from orchestra.contrib.accounts.filters import IsActiveListFilter from .actions import set_permission, create_link from .filters import IsMainListFilter from .forms import SystemUserCreationForm, SystemUserChangeForm, WebappUserChangeForm, WebappUserCreationForm from .models import SystemUser, WebappUsers class SystemUserAdmin(ChangePasswordAdminMixin, SelectAccountAdminMixin, ExtendedModelAdmin): list_display = ( 'username', 'account_link', 'shell', 'display_home', 'display_active', 'display_main' ) list_filter = (IsActiveListFilter, 'shell', IsMainListFilter) fieldsets = ( (None, { 'fields': ('username', 'password', 'account_link', 'is_active') }), (_("System"), { 'fields': ('shell', ('home', 'directory'), 'groups'), }), ) add_fieldsets = ( (None, { 'fields': ('account_link', 'username', 'password1', 'password2') }), (_("System"), { 'fields': ('shell', ('home', 'directory'), 'groups'), }), ) search_fields = ('username', 'account__username') readonly_fields = ('account_link',) change_readonly_fields = ('username',) filter_horizontal = ('groups',) filter_by_account_fields = ('groups',) add_form = SystemUserCreationForm form = SystemUserChangeForm ordering = ('-id',) change_view_actions = (set_permission, create_link) actions = (disable, enable, list_accounts) + change_view_actions def display_main(self, user): return user.is_main display_main.short_description = _("Main") display_main.boolean = True def display_home(self, user): return user.get_home() display_home.short_description = _("Home") display_home.admin_order_field = 'home' def get_form(self, request, obj=None, **kwargs): form = super(SystemUserAdmin, self).get_form(request, obj, **kwargs) form.account = self.account if obj: # Has to be done here and not in the form because of strange phenomenon # derived from monkeypatching formfield.widget.render on AccountAdminMinxin, # don't ask. formfield = form.base_fields['groups'] formfield.queryset = formfield.queryset.exclude(id=obj.id) return form def has_delete_permission(self, request, obj=None): if obj and obj.is_main: self.message_user(request, _( "You have selected one main system user (%(account)s), which can not be deleted.", ) % {'account': obj}, messages.ERROR, ) return False return super(SystemUserAdmin, self).has_delete_permission(request, obj) class WebappUserAdmin(ChangePasswordAdminMixin, SelectAccountAdminMixin, ExtendedModelAdmin): list_display = ( 'username', 'account_link', 'home', 'target_server' ) fieldsets = ( (None, { 'fields': ('account_link', 'username', 'password', ) }), (_("System"), { 'fields': ('shell', 'home', 'target_server'), }), ) add_fieldsets = ( (None, { 'fields': ('account_link', 'username', 'password1', 'password2') }), (_("System"), { 'fields': ('shell', 'home', 'target_server'), }), ) search_fields = ('username', 'account__username') readonly_fields = ('account_link',) change_readonly_fields = ('username', 'home', 'target_server') add_form = WebappUserCreationForm form = WebappUserChangeForm ordering = ('-id',) admin.site.register(SystemUser, SystemUserAdmin) admin.site.register(WebappUsers, WebappUserAdmin)