Django 用户密码重置功能

写在前面

本文适用于在django开发时满足下列情况的伙伴：

1. 使用了自定义用户类，而不是django自带的User用户类；
2. 密码重置时仅仅只是判断邮箱是否正确，而不需要发送邮箱验证码，与注册时保存在数据库的信息匹配成功即可；

假设自定义了用户类CustomUser，虽然下面用不到，但还是要提及，下列为具体步骤：

步骤一：创建密码重置视图

在 views.py 文件中，可以按照以下示例编写一个简单的视图：

# views.py
from django.shortcuts import render, redirect
from django.contrib.auth import get_user_model
from django.contrib.auth import update_session_auth_hash
from django.contrib import messages

def reset_password(request):
    if request.method == 'POST':
        username = request.POST.get('username')
        email = request.POST.get('email')

        User = get_user_model()
        user_manager = User.objects

        try:
            user = user_manager.get(username=username, email=email)
        except User.DoesNotExist:
            messages.error(request, '用户名或邮箱不匹配，请检查输入。')
            return redirect('password_reset')

        # 在这里可以进行额外的验证，例如检查用户状态等

        # 处理密码重置逻辑
        new_password = request.POST.get('new_password')
        user.set_password(new_password)
        user.save()

        # 更新用户会话，确保用户在密码更改后仍然保持登录状态
        update_session_auth_hash(request, user)

        messages.success(request, '密码已成功重置。')
        return redirect('login')  # 重定向到你的登录页面

    return render(request, 'password_reset.html')

需要注意的是，用户初始化一定是采用django自带的get_user_model()，而不是自定义用户类！

笔者这里写成了自定义类，于是重置密码信息提交后出现了下列错误：

AttributeError at /password_reset/ Manager isn't accessible via CustomUser instances

步骤二：创建密码重置页面

在 templates 目录中创建 password_reset.html 文件，包含一个表单，用户可以在其中输入用户名、邮箱和密码，下列给出关键代码：

<!-- password_reset.html -->

<div class="container reset-container">
    <h2>Reset Your Password</h2>
    <div class="reset-form">
        <form method="post">
            {% csrf_token %}

            <div class="form-row">
                <label for="username">Username</label>
                <input type="text" name="username" required>
            </div>

            <div class="form-row">
                <label for="email">Email</label>
                <input type="email" name="email" required>
            </div>

            <div class="form-row">
                <label for="new_password">New Password</label>
                <input type="password" name="new_password" required>
            </div>

            <div class="form-row">
                <label for="confirm_new_password">Confirm New Password</label>
                <input type="password" name="confirm_new_password" required>
            </div>

            <button type="submit">Submit</button>
        </form>
    </div>
</div>