This commit is contained in:
Matt Warren 2014-11-11 14:17:24 +00:00
commit cedbde8718
2 changed files with 21 additions and 2 deletions

View File

@ -31,6 +31,7 @@ Alberto Sanchez / @alb3rto
Eyad Al Sibai / @eyadsibai Eyad Al Sibai / @eyadsibai
Chris Franklin Chris Franklin
Benjamin Abel Benjamin Abel
Matt Warren / @mfwarren
* Possesses commit rights * Possesses commit rights

View File

@ -6,9 +6,27 @@ from django.contrib.auth.admin import UserAdmin as AuthUserAdmin
from .models import User from .models import User
class MyUserChangeForm(UserChangeForm):
class Meta(UserChangeForm.Meta):
model = User
class MyUserCreationForm(UserCreationForm):
class Meta(UserCreationForm.Meta):
model = User
def clean_username(self):
username = self.cleaned_data["username"]
try:
User.objects.get(username=username)
except User.DoesNotExist:
return username
raise forms.ValidationError(self.error_messages['duplicate_username'])
class UserAdmin(AuthUserAdmin): class UserAdmin(AuthUserAdmin):
create_form_class = UserCreationForm form = MyUserChangeForm
update_form_class = UserChangeForm add_form = MyUserCreationForm
admin.site.register(User, UserAdmin) admin.site.register(User, UserAdmin)