Python中基于AbstractUser()的用户模型扩展方法

在django中，我们可以使用AbstractUser类来创建用户模型。AbstractUser提供了一些默认的用户字段和方法，但有时我们需要为我们的用户模型添加一些自定义字段和方法。可以通过扩展AbstractUser类来实现这一目的。

下面是一些常用的方法来扩展AbstractUser类：

1. 创建自定义用户模型：

可以通过继承AbstractUser类来创建自定义的用户模型。首先，在项目的settings.py文件中设置AUTH_USER_MODEL为我们自定义的用户模型。

例如，我们创建一个叫做CustomUser的用户模型：

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

class CustomUser(AbstractUser):
    age = models.PositiveIntegerField(null=True, blank=True)
    address = models.CharField(max_length=255, null=True, blank=True)

在上面的例子中，我们添加了两个自定义字段age和address。

2. 创建自定义用户管理器：

可以通过创建自定义的用户管理器来为用户模型添加自定义的查询方法。

例如，我们创建一个叫做CustomUserManager的用户管理器，并添加一个查询年龄大于18岁的用户的方法：

from django.contrib.auth.models import AbstractUser, BaseUserManager

class CustomUserManager(BaseUserManager):
    def adults(self):
        return self.filter(age__gte=18)

class CustomUser(AbstractUser):
    age = models.PositiveIntegerField(null=True, blank=True)
    address = models.CharField(max_length=255, null=True, blank=True)

    objects = CustomUserManager()

在上面的例子中，我们创建了一个CustomUserManager类，并添加了一个名为adults()的方法，该方法返回年龄大于等于18岁的用户。

3. 扩展用户模型的验证逻辑：

可以通过重写用户模型的clean()方法来添加自定义的验证逻辑。

例如，我们为CustomUser模型添加一个验证年龄是否大于0的逻辑：

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

class CustomUser(AbstractUser):
    age = models.PositiveIntegerField(null=True, blank=True)
    address = models.CharField(max_length=255, null=True, blank=True)

    def clean(self):
        super().clean()
        if self.age and self.age <= 0:
            raise ValidationError('Age must be greater than 0.')

在上面的例子中，我们重写了clean()方法，并添加了一个自定义的验证逻辑，如果年龄小于等于0，则会抛出一个ValidationError异常。

4. 扩展用户模型的保存逻辑：

可以通过重写用户模型的save()方法来添加自定义的保存逻辑。

例如，我们为CustomUser模型添加一个在保存用户之前自动填充地址的逻辑：

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

class CustomUser(AbstractUser):
    age = models.PositiveIntegerField(null=True, blank=True)
    address = models.CharField(max_length=255, null=True, blank=True)

    def save(self, *args, **kwargs):
        if not self.address:
            self.address = 'Unknown'
        super().save(*args, **kwargs)

在上面的例子中，我们重写了save()方法，并添加了一个自动填充地址的逻辑，如果地址为空，则将地址设置为'Unknown'，然后保存用户。

以上是一些常用的方法来扩展AbstractUser类。当需要自定义用户模型时，可以根据具体需求选择合适的方法来进行扩展。