扩展Django用户模型：添加自定义字段方法

本文将介绍如何在 Django 中扩展默认的 User 模型，无需使用一对一关联，直接添加自定义字段。通过创建自定义用户模型并配置 AUTH_USER_MODEL，你可以轻松地在用户注册和管理中包含额外的 boolean 或 choice fields，并确保这些字段在 admin 后台正确显示和管理。

创建自定义用户模型

Django 允许你通过继承 AbstractUser 或 AbstractBaseUser 类来创建自定义用户模型。AbstractUser 已经包含了 Django 默认的用户模型中的大部分字段和方法，因此通常是扩展用户模型的首选。

在你的 Django 应用的 models.py 文件中，创建如下所示的自定义用户模型：

from django.contrib.auth.models import AbstractUser
from django.db import models

class CustomUser(AbstractUser):
    # 添加你的自定义字段
    is_premium = models.BooleanField(default=False, verbose_name="高级用户")
    subscription_type = models.CharField(
        max_length=20,
        choices=[
            ('basic', 'Basic'),
            ('premium', 'Premium'),
            ('enterprise', 'Enterprise'),
        ],
        default='basic',
        verbose_name="订阅类型"
    )

    def __str__(self):
        return self.username

在这个例子中，我们添加了两个自定义字段：

• is_premium: 一个布尔字段，表示用户是否为高级用户。
• subscription_type: 一个字符串字段，使用 choices 参数定义了用户订阅类型的选项。

配置 AUTH_USER_MODEL

为了让 Django 使用你的自定义用户模型，你需要修改 settings.py 文件，设置 AUTH_USER_MODEL 变量：

AUTH_USER_MODEL = 'your_app_name.CustomUser'  # 将 your_app_name 替换为你的应用名称

注意: 这一步非常重要，确保 Django 使用你的自定义用户模型而不是默认的 User 模型。

执行数据库迁移

在修改了用户模型后，你需要执行数据库迁移来更新数据库结构：

python manage.py makemigrations
python manage.py migrate

makemigrations 命令会生成迁移文件，描述了你对模型所做的更改。migrate 命令会将这些更改应用到数据库。

在 Admin 后台注册自定义用户模型

为了在 Django Admin 后台管理你的自定义用户模型，你需要将其注册到 admin.py 文件中。通常，你需要继承 UserAdmin 类并进行一些自定义：

from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from .models import CustomUser

class CustomUserAdmin(UserAdmin):
    model = CustomUser
    list_display = ['username', 'email', 'is_staff', 'is_premium', 'subscription_type'] # 自定义显示的字段
    fieldsets = (
        (None, {'fields': ('username', 'password')}),
        ('Personal info', {'fields': ('first_name', 'last_name', 'email')}),
        ('Permissions', {'fields': ('is_active', 'is_staff', 'is_superuser', 'groups', 'user_permissions')}),
        ('Important dates', {'fields': ('last_login', 'date_joined')}),
        ('Custom Fields', {'fields': ('is_premium', 'subscription_type')}), # 添加自定义字段
    )

admin.site.register(CustomUser, CustomUserAdmin)

在这个例子中，我们创建了一个 CustomUserAdmin 类，继承自 UserAdmin。 我们修改了 list_display 属性，使其在用户列表页面显示自定义字段。我们还修改了 fieldsets 属性，将自定义字段添加到用户编辑页面。

注意事项

• 在项目初期进行自定义： 建议在项目初期就决定是否需要自定义用户模型。在项目后期更改 AUTH_USER_MODEL 可能会导致数据迁移问题。
• 继承的选择： AbstractUser 提供了默认字段，而 AbstractBaseUser 则需要你完全定义用户模型。根据你的需求选择合适的基类。
• 密码处理： 如果你选择 AbstractBaseUser，请确保正确处理密码的哈希和验证。
• 迁移冲突： 在多人协作开发时，请注意数据库迁移的冲突，并及时解决。
• 测试： 务必对自定义用户模型进行充分的测试，确保其功能正常。

总结

通过创建自定义用户模型并配置 AUTH_USER_MODEL，你可以轻松地扩展 Django 的用户模型，添加自定义字段，并在 admin 后台进行管理。这种方法避免了使用一对一关联带来的复杂性，使得用户信息的管理更加灵活和方便。 记住，在项目初期就进行自定义用户模型的决策，并进行充分的测试，以确保项目的稳定性和可维护性。