Django重写默认表单
我的目的是要重写Django默认的 PasswordResetForm
中的 clean_email()
方法。这样做的原因是我想通过用户名来检查用户是否存在,而不是通过电子邮件(这只是我应用程序的业务逻辑)。然后我就可以重置用户的密码。请问我该如何重写这个方法?更重要的是,如果我这样做了,整个密码重置系统会不会崩溃?
这是Django的默认代码:
class PasswordResetForm(forms.Form):
error_messages = {
'unknown': _("That email address doesn't have an associated "
"user account. Are you sure you've registered?"),
'unusable': _("The user account associated with this email "
"address cannot reset the password."),
}
email = forms.EmailField(label=_("Email"), max_length=254)
def clean_email(self):
"""
Validates that an active user exists with the given email address.
"""
UserModel = get_user_model()
email = self.cleaned_data["email"]
self.users_cache = UserModel._default_manager.filter(email__iexact=email)
if not len(self.users_cache):
raise forms.ValidationError(self.error_messages['unknown'])
if not any(user.is_active for user in self.users_cache):
# none of the filtered users are active
raise forms.ValidationError(self.error_messages['unknown'])
if any((user.password == UNUSABLE_PASSWORD)
for user in self.users_cache):
raise forms.ValidationError(self.error_messages['unusable'])
return email
def save(self, domain_override=None,
subject_template_name='registration/password_reset_subject.txt',
email_template_name='registration/password_reset_email.html',
use_https=False, token_generator=default_token_generator,
from_email=None, request=None):
"""
Generates a one-use only link for resetting password and sends to the
user.
"""
from django.core.mail import send_mail
for user in self.users_cache:
if not domain_override:
current_site = get_current_site(request)
site_name = current_site.name
domain = current_site.domain
else:
site_name = domain = domain_override
c = {
'email': user.email,
'domain': domain,
'site_name': site_name,
'uid': int_to_base36(user.pk),
'user': user,
'token': token_generator.make_token(user),
'protocol': use_https and 'https' or 'http',
}
subject = loader.render_to_string(subject_template_name, c)
# Email subject *must not* contain newlines
subject = ''.join(subject.splitlines())
email = loader.render_to_string(email_template_name, c)
send_mail(subject, email, from_email, [user.email])
2 个回答
-1
我之所以想这样做,是因为我想在Django注册中使用电子邮件作为用户名。我曾考虑过直接把Django的用户名字段改成电子邮件字段,但我不知道从Django 1.5开始,你可以指定自己的自定义用户模型。
0
我觉得你可以写一个叫 clean_username
的方法来实现你想要的功能。这个链接可能对你有帮助:在Django中检查用户名是否存在