from django import forms from django.utils.translation import ugettext as _ from django.contrib import admin from .models import ( AdditionalGroup, DeleteTaskResult, Group, GroupTaskResult, Shadow, User, UserTaskResult, ) PASSWORD_MISMATCH_ERROR = _("Passwords don't match") class AdditionalGroupInline(admin.TabularInline): model = AdditionalGroup class ShadowInline(admin.TabularInline): model = Shadow readonly_fields = ['passwd'] can_delete = False class TaskResultInline(admin.TabularInline): can_delete = False extra = 0 readonly_fields = ['task_uuid', 'task_name', 'is_finished', 'is_success', 'state', 'result_body'] def get_queryset(self, request): qs = super(TaskResultInline, self).get_queryset(request) for entry in qs: entry.update_taskstatus() return qs def has_add_permission(self, request, obj=None): return False class UserTaskResultInline(TaskResultInline): model = UserTaskResult class GroupTaskResultInline(TaskResultInline): model = GroupTaskResult class UserCreationForm(forms.ModelForm): """ A form for creating system users. """ password1 = forms.CharField(label=_('Password'), widget=forms.PasswordInput) password2 = forms.CharField(label=_('Password (again)'), widget=forms.PasswordInput) class Meta: model = User fields = [] def clean_password2(self): """ Check that the two password entries match. """ password1 = self.cleaned_data.get('password1') password2 = self.cleaned_data.get('password2') if password1 and password2 and password1 != password2: raise forms.ValidationError(PASSWORD_MISMATCH_ERROR) return password2 def save(self, commit=True): """ Save the provided password in hashed format. """ user = User.objects.create_user( password=self.cleaned_data['password1'], commit=commit) return user def save_m2m(self): pass class UserAdmin(admin.ModelAdmin): inlines = [AdditionalGroupInline, ShadowInline, UserTaskResultInline] readonly_fields = ['uid'] add_form = UserCreationForm add_fieldsets = ( (None, { 'classes': ('wide',), 'fields': ('password1', 'password2')}), ) def get_form(self, request, obj=None, **kwargs): """ Use special form during user creation. """ defaults = {} if obj is None: defaults.update({ 'form': self.add_form, 'fields': admin.util.flatten_fieldsets(self.add_fieldsets), }) defaults.update(kwargs) return super(UserAdmin, self).get_form(request, obj, **defaults) def get_inline_instances(self, request, obj=None): if obj is None: return [] return super(UserAdmin, self).get_inline_instances(request, obj) class GroupAdmin(admin.ModelAdmin): inlines = [GroupTaskResultInline] def get_inline_instances(self, request, obj=None): if obj is None: return [] return super(GroupAdmin, self).get_inline_instances(request, obj) class DeleteTaskResultAdmin(admin.ModelAdmin): readonly_fields = ['task_uuid', 'task_name', 'modeltype', 'modelname', 'is_finished', 'is_success', 'state', 'result_body'] list_display = ('task_uuid', 'task_name', 'modeltype', 'modelname', 'is_finished', 'state') def has_add_permission(self, request, obj=None): return False def get_queryset(self, request): qs = super(DeleteTaskResultAdmin, self).get_queryset(request) for entry in qs: entry.update_taskstatus() return qs admin.site.register(Group, GroupAdmin) admin.site.register(User, UserAdmin) admin.site.register(DeleteTaskResult, DeleteTaskResultAdmin)