register.py 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. from django import forms
  2. from django.contrib.auth import get_user_model
  3. from django.contrib.auth.password_validation import validate_password
  4. from .. import validators
  5. class RegisterForm(forms.Form):
  6. username = forms.CharField(validators=[validators.validate_username])
  7. email = forms.CharField(validators=[validators.validate_email])
  8. password = forms.CharField()
  9. # placeholder field for setting captcha errors on form
  10. captcha = forms.CharField(required=False)
  11. def __init__(self, *args, **kwargs):
  12. self.request = kwargs.pop('request')
  13. super(RegisterForm, self).__init__(*args, **kwargs)
  14. def full_clean_password(self, cleaned_data):
  15. if cleaned_data.get('password'):
  16. UserModel = get_user_model()
  17. validate_password(
  18. cleaned_data['password'],
  19. user=UserModel(
  20. username=cleaned_data.get('username'),
  21. email=cleaned_data.get('email'),
  22. )
  23. )
  24. def clean(self):
  25. cleaned_data = super(RegisterForm, self).clean()
  26. try:
  27. self.full_clean_password(cleaned_data)
  28. except forms.ValidationError as e:
  29. self.add_error('password', e)
  30. validators.validate_new_registration(self.request, self, cleaned_data)
  31. return cleaned_data