Django项目中实现PIN验证并返回错误信息的方法

问题背景 #

已创建包含pin字段的Data模型、表单及HTML页面，需要实现登录后PIN验证逻辑：输入PIN与模型存储值不一致时返回错误提示，正确则返回成功提示。

具体修改步骤 #

1. 修正视图函数逻辑 #

原视图存在模型引用错误、未处理用户输入、无消息反馈的问题，修改后的代码如下：

from django.contrib import messages
from django.contrib.auth.decorators import login_required
from .models import Data  # 导入正确的Data模型

@login_required(login_url='loginss')
def account(request):
    if request.method == 'POST':
        # 获取并处理用户输入的PIN
        input_pin = request.POST.get('pin')
        try:
            input_pin = int(input_pin)
            # 匹配当前登录用户的Data记录（这里假设用邮箱关联，若有User外键可直接用request.user）
            user_data = Data.objects.get(email=request.user.email)
            
            # 对比PIN值
            if user_data.pin == input_pin:
                messages.success(request, 'PIN验证成功！')
            else:
                messages.error(request, 'PIN输入错误，请重试！')
        except ValueError:
            messages.error(request, '请输入有效的数字PIN！')
        except Data.DoesNotExist:
            messages.error(request, '未找到您的账户信息，请联系管理员！')
    
    return render(request, 'account.html', {})

2. 修复HTML表单错误 #

原HTML中表单标签提前闭合，导致提交逻辑失效，同时优化消息样式：

<style>
    .success-msg { color: green; }
    .error-msg { color: red; }
</style>

<!-- 展示验证消息 -->
{% for message in messages %}
    <h5 class="{% if message.tags == 'success' %}success-msg{% else %}error-msg{% endif %}">
        {{ message }}
    </h5>
{% endfor %}

<!-- 修正后的表单 -->
<form action="{% url 'account' %}" method="POST" class="row g-3">
    {% csrf_token %}
    <div class="col-md-12">
        <label for="validationDefault02" class="form-label">请输入PIN码</label>
        <input type="number" name="pin" class="form-control" id="validationDefault02" required>
    </div>
    <div class="col-12">
        <button class="btn btn-primary" type="submit">验证</button>
    </div>
</form>

3. 优化模型关联（可选但推荐） #

为了更可靠地匹配当前用户的Data记录，建议给Data模型添加与User的一对一关联：

from django.contrib.auth.models import User

class Data(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='user_data')
    # 保留原有所有字段...
    
    def __str__(self):
        return self.first_name

修改后，视图中匹配记录的代码可改为：

user_data = Data.objects.get(user=request.user)

4. 确认消息框架配置 #

确保settings.py中已启用消息相关组件：

• INSTALLED_APPS包含django.contrib.messages
• MIDDLEWARE包含django.contrib.sessions.middleware.SessionMiddleware和django.contrib.messages.middleware.MessageMiddleware

内容的提问来源于stack exchange，提问作者Neo_77cyber