Django重写默认表单

1 投票
2 回答
509 浏览
提问于 2025-04-18 01:00

我的目的是要重写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中检查用户名是否存在

撰写回答