Django用户类型升级：如何将Buyer转为Merchant？

1. 编写表单类 #

在你的app的forms.py中创建一个确认表单，用于用户确认升级操作：

from django import forms

class BecomeMerchantForm(forms.Form):
    confirm_upgrade = forms.BooleanField(
        label="我确认升级为商家",
        required=True,
        help_text="勾选此框表示你同意升级为商家账户"
    )

2. 实现视图函数 #

在views.py中编写处理升级请求的视图，确保仅登录的买家能访问并完成升级：

from django.shortcuts import render, redirect
from django.contrib.auth.decorators import login_required
from .forms import BecomeMerchantForm
from .models import Merchant

@login_required
def become_merchant(request):
    # 已是商家则直接跳转
    if request.user.is_merchant:
        return redirect('merchant-dashboard')
    
    # 非买家用户禁止访问
    if not request.user.is_buyer:
        return redirect('home')
    
    if request.method == 'POST':
        form = BecomeMerchantForm(request.POST)
        if form.is_valid():
            # 更新用户的商家状态
            request.user.is_merchant = True
            request.user.save()
            
            # 创建关联的Merchant实例
            Merchant.objects.create(user=request.user)
            
            # 可选：若需要取消买家身份，取消以下注释
            # request.user.is_buyer = False
            # request.user.save()
            
            return redirect('merchant-dashboard')
    else:
        form = BecomeMerchantForm()
    
    return render(request, 'become-a-merchant.html', {'form': form})

3. 修改模板文件 #

更新become-a-merchant.html，用Django表单渲染并完善表单结构：

{% extends 'base.html' %}

{% block content %}
<form method="post">
    {% csrf_token %}
    {{ form.as_p }}
    <button type="submit">升级为商家</button>
</form>
{% endblock %}

4. 配置URL路由 #

在app的urls.py中添加对应路由：

from django.urls import path
from . import views

urlpatterns = [
    # 其他路由...
    path('become-merchant/', views.become_merchant, name='become-merchant'),
]

补充说明 #

• 确保用户注册时自动创建Buyer实例并设置User.is_buyer = True，可通过Django信号实现：

from django.db.models.signals import post_save
from django.contrib.auth.models import User
from django.dispatch import receiver
from .models import Buyer

@receiver(post_save, sender=User)
def create_buyer_profile(sender, instance, created, **kwargs):
    if created:
        Buyer.objects.create(user=instance)
        instance.is_buyer = True
        instance.save()

（将这段代码放在models.py末尾，或在apps.py中注册信号）

• 如果需要收集商家额外信息（如PGP密钥），可修改BecomeMerchantForm添加对应字段，在创建Merchant实例时同步保存数据。

内容的提问来源于stack exchange，提问作者static-man