27_扩展User模型-猿码集

1. 扩展User模型的目的

扩展User模型是在Django中定制用户模型的一种常见需求。Django的内置User模型包含了常见的用户属性，如用户名、密码和电子邮件等。然而，有时我们需要在User模型上添加额外的自定义字段，以满足特定的业务需求。扩展User模型可以使我们更好地定制用户属性，提供更丰富的用户信息。

2. 扩展User模型的方法

2.1 创建扩展模型

要扩展User模型，首先需要在Django中创建一个扩展模型。这个模型将包含我们要添加的自定义字段。具体操作如下：

from django.contrib.auth.models import AbstractUser
from django.db import models
class MyUser(AbstractUser):
    # 添加自定义字段
    phone_number = models.CharField(max_length=20)
    address = models.CharField(max_length=100)

在上述代码中，我们通过继承Django内置的AbstractUser类来创建了一个名为MyUser的模型。这个模型包含了两个自定义字段：phone_number和address。我们可以根据实际需求添加更多的字段。

2.2 配置自定义模型

创建了自定义的扩展模型后，需要进行相应的配置，使Django使用我们的自定义模型作为默认的用户模型。具体操作如下：

首先，在项目的settings.py文件中，找到AUTH_USER_MODEL配置项，将其指定为我们创建的自定义模型：

# settings.py AUTH_USER_MODEL = 'myapp.MyUser'

这里的myapp可以替换为实际使用该模型的应用名称。

2.3 进行数据库迁移

配置好自定义模型后，需要进行数据库迁移，以更新数据库结构以适应新的模型。具体操作如下：

python manage.py makemigrations python manage.py migrate

以上命令将会生成新的数据库迁移文件，并将新的模型应用到数据库中。

3. 在代码中使用扩展的User模型

一旦完成了User模型的扩展，我们可以在代码中使用新增的自定义字段。以下是一些常见的用法：

3.1 创建用户对象

创建用户对象时，可以通过设置自定义字段的值来定制用户信息：

# 创建用户对象
user = MyUser(username='john', password='password', phone_number='1234567890', address='New York')
user.save()

上述代码中，我们通过设置phone_number和address字段的值来定制用户信息，并将对象保存到数据库中。

3.2 访问自定义字段

在获取用户对象后，我们可以通过访问自定义字段来获取用户的额外信息：

# 获取自定义字段的值 phone_number = user.phone_number address = user.address

上述代码中，我们通过访问phone_number和address字段来获取相应的值。

3.3 扩展User模型的管理器

除了自定义字段外，我们还可以通过扩展User模型的管理器来添加额外的功能。例如，我们可以创建一个自定义的管理器来过滤用户对象：

from django.contrib.auth.models import BaseUserManager
class MyUserManager(BaseUserManager):
    def get_staff_users(self):
        # 获取员工用户列表
        return self.filter(is_staff=True)
class MyUser(AbstractUser):
    # 添加自定义字段
    phone_number = models.CharField(max_length=20)
    address = models.CharField(max_length=100)
    objects = MyUserManager()

在上述代码中，我们通过继承BaseUserManager类并重写get_staff_users方法来创建一个自定义的管理器。然后，将该管理器赋值给MyUser模型的objects属性。这样我们就可以通过objects属性来访问自定义的管理器方法。

4. 总结

通过扩展User模型，我们可以在Django中定制用户模型，添加额外的自定义字段和功能。首先，我们创建一个扩展模型，包含我们要添加的自定义字段。然后，通过配置自定义模型，并进行数据库迁移，使得Django使用我们的自定义模型作为默认的用户模型。最后，我们可以在代码中使用新增的自定义字段，并且通过扩展User模型的管理器来添加额外的功能。以上是扩展User模型的基本步骤和用法，希望对你有所帮助。

27_扩展User模型