本文将介绍如何在 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 后台进行管理。这种方法避免了使用一对一关联带来的复杂性，使得用户信息的管理更加灵活和方便。 记住，在项目初期就进行自定义用户模型的决策，并进行充分的测试，以确保项目的稳定性和可维护性。
