選択できるのは25トピックまでです。 トピックは、先頭が英数字で、英数字とダッシュ('-')を使用した35文字以内のものにしてください。

admin.py 2.0 KiB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. from api.models import PasswordInfo, Entry, LessPassUser
  2. from django import forms
  3. from django.contrib import admin
  4. from django.contrib.auth.models import Group
  5. from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
  6. from django.contrib.auth.forms import ReadOnlyPasswordHashField
  7. class UserCreationForm(forms.ModelForm):
  8. password1 = forms.CharField(label='Password', widget=forms.PasswordInput)
  9. password2 = forms.CharField(label='Password confirmation', widget=forms.PasswordInput)
  10. class Meta:
  11. model = LessPassUser
  12. fields = ('email',)
  13. def clean_password2(self):
  14. password1 = self.cleaned_data.get("password1")
  15. password2 = self.cleaned_data.get("password2")
  16. if password1 and password2 and password1 != password2:
  17. raise forms.ValidationError("Passwords don't match")
  18. return password2
  19. def save(self, commit=True):
  20. user = super(UserCreationForm, self).save(commit=False)
  21. user.set_password(self.cleaned_data["password1"])
  22. if commit:
  23. user.save()
  24. return user
  25. class UserChangeForm(forms.ModelForm):
  26. password = ReadOnlyPasswordHashField()
  27. class Meta:
  28. model = LessPassUser
  29. fields = ('email', 'password', 'is_active', 'is_admin')
  30. def clean_password(self):
  31. return self.initial["password"]
  32. class LessPassUserAdmin(BaseUserAdmin):
  33. form = UserChangeForm
  34. add_form = UserCreationForm
  35. list_display = ('email', 'is_admin',)
  36. list_filter = ('is_admin',)
  37. fieldsets = (
  38. (None, {'fields': ('email', 'password')}),
  39. ('Permissions', {'fields': ('is_admin',)}),
  40. )
  41. add_fieldsets = (
  42. (None, {
  43. 'classes': ('wide',),
  44. 'fields': ('email', 'password1', 'password2')}
  45. ),
  46. )
  47. search_fields = ('email',)
  48. ordering = ('email',)
  49. filter_horizontal = ()
  50. admin.site.register(Entry)
  51. admin.site.register(PasswordInfo)
  52. admin.site.register(LessPassUser, LessPassUserAdmin)
  53. admin.site.unregister(Group)