本文介绍了如何在 Django 中扩展默认的 User 模型，通过创建自定义用户模型并添加额外的布尔字段和选择字段，无需使用一对一字段关联到其他模型。同时，本文还阐述了如何将自定义字段集成到 Django Admin 后台进行管理，提供完整的代码示例和操作步骤，帮助开发者更好地定制用户模型。
自定义 User 模型
Django 提供了灵活的用户认证系统，允许开发者根据实际需求扩展 User 模型。最常用的方法是创建一个继承自 AbstractUser 的自定义模型。这种方式可以保留 Django 默认的用户认证功能，并允许添加自定义字段。
首先，在你的 Django 应用的 models.py 文件中，创建自定义用户模型。例如，我们创建一个名为 CustomUser 的模型，并添加一个布尔字段 is_premium 和一个选择字段 user_type。
from django.contrib.auth.models import AbstractUser
from django.db import models
class CustomUser(AbstractUser):
is_premium = models.BooleanField(default=False, verbose_name="Premium User")
USER_TYPE_CHOICES = (
('normal', 'Normal User'),
('admin', 'Administrator'),
)
user_type = models.CharField(max_length=10, choices=USER_TYPE_CHOICES, default='normal', verbose_name="User Type")
def __str__(self):
return self.username在这个例子中，is_premium 是一个布尔字段，用于表示用户是否是高级用户。user_type 是一个选择字段，允许用户选择不同的用户类型。verbose_name 用于在 Admin 后台中显示更友好的字段名称。
配置 AUTH_USER_MODEL
为了让 Django 使用你的自定义用户模型，需要在 settings.py 文件中设置 AUTH_USER_MODEL。
AUTH_USER_MODEL = 'your_app_name.CustomUser'
将 your_app_name 替换为你的 Django 应用的名称。例如，如果你的应用名为 accounts，则应设置为 AUTH_USER_MODEL = 'accounts.CustomUser'。
注意： 在设置 AUTH_USER_MODEL 之前，请确保你还没有创建任何与默认 User 模型相关的迁移文件或数据库表。否则，可能会导致迁移错误。
创建并应用迁移
在完成模型定义和 AUTH_USER_MODEL 配置后，需要创建并应用迁移，以便将更改同步到数据库。
本系统使用的是XDcms内核，在原来基础上做来相应修改 前台修改调用数据，可以使用{loop catid=栏目ID}{/loop}方式调用 主要功能： A、内容管理模型，自定义字段，更方便扩展功能。自带模型：单页模型、新闻模型、产品模型、招聘模型 B、栏目自定义，便于内容管理 C、内容模块化，二次开发更便捷。自带模块：幻灯片、QQ客服、友情链接、自定义表单(在线留言、简历管理) D、模板管理，后台
python manage.py makemigrations python manage.py migrate
makemigrations 命令会生成迁移文件，描述数据库结构的更改。migrate 命令会将这些更改应用到数据库。
在 Django 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):
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', 'user_type')}), # Add custom fields here
)
add_fieldsets = (
(None, {
'classes': ('wide',),
'fields': ('username', 'password', 'first_name', 'last_name', 'email', 'is_premium', 'user_type')}
),
)
list_display = ('username', 'email', 'first_name', 'last_name', 'is_staff', 'is_premium', 'user_type')
admin.site.register(CustomUser, CustomUserAdmin)在这个例子中，我们创建了一个 CustomUserAdmin 类，继承自 UserAdmin。fieldsets 属性定义了在编辑用户时显示的字段分组。我们添加了一个名为 Custom Fields 的分组，并将 is_premium 和 user_type 字段添加到该分组中。 add_fieldsets定义了添加用户时的字段分组。list_display属性定义了在用户列表页面显示的字段。
注意事项：
- 务必在修改 AUTH_USER_MODEL 之前备份数据库。
- 在生产环境中，需要考虑数据迁移的兼容性，避免数据丢失。
- 自定义用户模型后，所有与用户相关的代码都需要进行相应的调整，例如权限验证、用户查询等。
总结：
通过继承 AbstractUser 并配置 AUTH_USER_MODEL，可以方便地扩展 Django 的 User 模型，添加自定义字段，并将其集成到 Django Admin 后台中进行管理。这种方法既保留了 Django 默认的用户认证功能，又允许开发者根据实际需求定制用户模型，提高开发效率和灵活性。
