django之对django内置的User模型进行自定义扩展方式-猿码集

1. 介绍

Django是一款使用Python开发的Web应用框架，提供了一套完善的开发工具和库，使开发者能够快速开发出高质量的Web应用。Django内置了一个名为User的模型，用于处理用户认证和管理用户数据。但是在实际项目中，往往需要对User模型进行自定义扩展，以满足特定的业务需求。

2. 自定义User模型

在Django中，可以通过继承AbstractUser或AbstractBaseUser来创建自定义的User模型。AbstractUser提供了一些常用的属性和方法，适用于大多数情况下的用户管理；而AbstractBaseUser提供了更大的灵活性，可以完全自定义User模型。

2.1 继承AbstractUser

继承AbstractUser是最常见的对User模型进行扩展的方式，它允许我们在现有的User模型基础上添加额外的字段。

from django.contrib.auth.models import AbstractUser
from django.db import models
class CustomUser(AbstractUser):
    age = models.IntegerField(null=True)
    phone = models.CharField(max_length=20, null=True)

在上述代码中，我们定义了一个名为CustomUser的模型，它继承自AbstractUser，并添加了两个额外的字段age和phone。现在，我们可以在项目中使用CustomUser代替默认的User模型。

2.2 继承AbstractBaseUser

如果AbstractUser无法满足我们的需求，我们还可以选择继承AbstractBaseUser，完全自定义User模型。

from django.contrib.auth.models import AbstractBaseUser, BaseUserManager
from django.db import models
class CustomUserManager(BaseUserManager):
    def create_user(self, email, password=None):
        # 创建普通用户
        ...
    def create_superuser(self, email, password=None):
        # 创建超级用户
        ...
class CustomUser(AbstractBaseUser):
    email = models.EmailField(unique=True)
    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = []
    objects = CustomUserManager()

在上述代码中，我们定义了一个名为CustomUserManager的管理器，它继承自BaseUserManager，并重写了create_user和create_superuser方法。同时，我们还定义了一个名为CustomUser的模型，它继承自AbstractBaseUser，并指定了USERNAME_FIELD为email。最后，我们将CustomUserManager赋值给CustomUser的objects属性，以便管理用户对象。

3. 数据迁移

在对User模型进行扩展之后，需要进行数据迁移，以更新数据库结构。

$ python manage.py makemigrations

$ python manage.py migrate

通过以上命令，Django会自动创建迁移文件并执行迁移操作，将User模型的改动应用到数据库中。

4. 使用自定义User模型

在项目中使用自定义的User模型与使用内置的User模型大致相同，只需在配置文件中进行相应的配置即可。

首先，需要设置AUTH_USER_MODEL属性为自定义User模型的路径。

# settings.py

AUTH_USER_MODEL = 'myapp.CustomUser'

然后，在视图函数中可以通过get_user_model()方法获取当前使用的User模型，而不需要硬编码。

from django.contrib.auth import get_user_model
User = get_user_model()

接下来，可以像使用内置的User模型一样，使用自定义的User模型进行用户认证和管理。

5. 扩展User模型的方法

除了添加额外的字段之外，我们还可以通过方法和属性对User模型进行扩展。

5.1 添加属性和方法

我们可以在自定义的User模型中添加任意的属性和方法，以满足特定的业务需求。

from django.contrib.auth.models import AbstractUser
from django.db import models
class CustomUser(AbstractUser):
    @property
    def full_name(self):
        return f'{self.first_name} {self.last_name}'

在上述代码中，我们在CustomUser模型中定义了一个名为full_name的属性，它返回用户的全名。在使用的时候，可以通过user.full_name的方式来获取用户的全名。

5.2 重写方法

除了添加额外的方法之外，我们还可以重写User模型已有的方法，以改变其行为。

from django.contrib.auth.models import AbstractUser
class CustomUser(AbstractUser):
    def save(self, *args, **kwargs):
        # 在保存用户之前的操作
        ...
        super().save(*args, **kwargs)

在上述代码中，我们重写了User模型的save方法，在保存用户之前进行了一些自定义操作。然后，调用父类的save方法，以保持原有的行为。

6. 总结

通过继承AbstractUser或AbstractBaseUser，我们可以对Django内置的User模型进行自定义扩展。无论是添加额外的字段，还是定义新的方法和属性，都可以根据具体的业务需求进行灵活的扩展。在实际项目中，定制化的User模型是非常常见的需求，能够有效提高开发效率和项目的可维护性。

django之对django内置的User模型进行自定义扩展方式