Django 2.1用户关联两种用户模型的方式对比及相关实现咨询

Great question! Let's break this down into clear sections to cover all your requirements:

1. 直接导入django.contrib.auth.models.User #

• 优点: 简单直观，适合快速搭建原型或小型项目，是Django基础文档里的入门写法，新手容易上手理解。
• 缺点: 耦合性极强，是生产项目的大忌。如果后续你需要替换默认的User模型（比如添加手机号字段、改用邮箱作为登录凭证等），你得修改所有关联User模型的代码，甚至要重新迁移数据库，操作风险高且维护成本大。Django官方也明确不推荐在长期维护的项目中使用这种方式。

2. 使用settings.AUTH_USER_MODEL #

• 优点: 完全解耦核心用户模型，遵循Django最佳实践（也是《Two Scoops of Django》推荐它的原因）。不管当前用的是默认User还是自定义用户模型，这种写法都能适配。后续如果要切换自定义用户模型，只需要修改settings.py里的AUTH_USER_MODEL配置，不用改动业务模型的代码，扩展性和可维护性拉满。
• 缺点: 相比直接导入多了一步from django.conf import settings的操作，新手一开始可能会有点困惑，但习惯后会发现这是一劳永逸的写法。

要实现注册时自动创建UserMusician或UserFan，核心是在用户创建后触发对应的资料创建逻辑，推荐两种方式：

方式一：用Django信号（解耦推荐） #

信号能帮你在用户模型的post_save事件触发时自动执行逻辑，不需要在视图里硬编码：

1. 首先，确保你的注册表单包含用户类型选择（比如让用户选“粉丝”或“音乐人”），如果用的是自定义用户模型，也可以给用户模型加一个user_type字段：

# forms.py
from django.contrib.auth.forms import UserCreationForm
from django.contrib.auth import get_user_model

User = get_user_model()

class UserRegistrationForm(UserCreationForm):
    USER_TYPE_CHOICES = (
        ('fan', 'Fan'),
        ('musician', 'Musician'),
    )
    user_type = forms.ChoiceField(choices=USER_TYPE_CHOICES, required=True)

    class Meta:
        model = User
        fields = ('username', 'email', 'password1', 'password2', 'user_type')

2. 编写信号处理器：

# signals.py
from django.db.models.signals import post_save
from django.dispatch import receiver
from django.conf import settings
from .models import UserMusician, UserFan

@receiver(post_save, sender=settings.AUTH_USER_MODEL)
def create_user_profile(sender, instance, created, **kwargs):
    if created:
        # 假设用户模型已添加user_type字段存储注册时选择的类型
        if instance.user_type == 'musician':
            UserMusician.objects.create(user=instance)
        elif instance.user_type == 'fan':
            UserFan.objects.create(user=instance)

3. 在app的apps.py里注册信号，确保Django启动时加载：

# apps.py
from django.apps import AppConfig

class YourAppConfig(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'your_app_name'

    def ready(self):
        import your_app_name.signals

方式二：在注册视图中直接创建（简单直接） #

如果你的项目逻辑比较简单，也可以在注册表单验证通过后直接创建对应的用户资料：

# views.py
from django.contrib.auth import login
from django.shortcuts import render, redirect
from .forms import UserRegistrationForm
from django.contrib import messages

def register(request):
    if request.method == 'POST':
        form = UserRegistrationForm(request.POST)
        if form.is_valid():
            user = form.save()
            user_type = form.cleaned_data.get('user_type')
            # 根据用户类型创建对应资料
            if user_type == 'musician':
                UserMusician.objects.create(user=user)
            else:
                UserFan.objects.create(user=user)
            login(request, user)
            messages.success(request, 'Registration successful!')
            return redirect('home')
    else:
        form = UserRegistrationForm()
    return render(request, 'registration/register.html', {'form': form})

这个逻辑的核心是：先删除用户关联的UserFan实例（因为一个用户只能有一个OneToOne的资料），再创建UserMusician实例。同时要处理异常情况（比如用户已经是音乐人、不是粉丝等）：

# views.py
from django.shortcuts import redirect
from django.contrib.auth.decorators import login_required
from .models import UserFan, UserMusician
from django.contrib import messages

@login_required
def convert_to_musician(request):
    # 检查用户是否已经是音乐人
    if UserMusician.objects.filter(user=request.user).exists():
        messages.error(request, 'You are already registered as a musician!')
        return redirect('profile')
    
    # 尝试获取并删除粉丝资料
    try:
        fan_profile = UserFan.objects.get(user=request.user)
        fan_profile.delete()
    except UserFan.DoesNotExist:
        messages.error(request, 'You are not registered as a fan, cannot convert!')
        return redirect('profile')
    
    # 创建音乐人资料
    UserMusician.objects.create(user=request.user)
    messages.success(request, 'Successfully converted to a musician!')
    return redirect('profile')

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

# urls.py
from django.urls import path
from . import views

urlpatterns = [
    # ... 其他路由
    path('convert-to-musician/', views.convert_to_musician, name='convert_to_musician'),
]

你可以在模板里添加一个按钮，指向这个路由，让用户触发转换操作。

内容的提问来源于stack exchange，提问作者Point Clear Media