gva/gnuviechadmin/osusers/admin.py

96 lines
2.4 KiB
Python
Raw Normal View History

from django import forms
from django.utils.translation import ugettext as _
2014-05-24 21:28:33 +02:00
from django.contrib import admin
2014-05-24 21:53:49 +02:00
from .models import (
AdditionalGroup,
Group,
Shadow,
User,
)
PASSWORD_MISMATCH_ERROR = _("Passwords don't match")
2014-05-24 21:53:49 +02:00
class AdditionalGroupInline(admin.TabularInline):
model = AdditionalGroup
class ShadowInline(admin.TabularInline):
model = Shadow
readonly_fields = ['passwd']
can_delete = False
2014-05-24 21:53:49 +02:00
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
2014-05-24 21:53:49 +02:00
class UserAdmin(admin.ModelAdmin):
inlines = [AdditionalGroupInline, ShadowInline]
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)
2014-05-24 21:53:49 +02:00
admin.site.register(Group)
admin.site.register(User, UserAdmin)