在Django中使用自定义用户模型

在Django中使用自定义用户模型需要遵循以下步骤：

1. 创建一个新的Django应用程序，并在该应用程序中创建一个models.py文件。

2. 导入Django的AbstractBaseUser和BaseUserManager类，用于创建自定义用户模型。

from django.contrib.auth.base_user import AbstractBaseUser, BaseUserManager

3. 创建一个自定义用户管理器。

```
class CustomUserManager(BaseUserManager):
def create_user(self, email, password, extra_fields):
if not email:
raise ValueError('The Email field must be set')
email = self.normalize_email(email)
user = self.model(email=email, extra_fields)
user.set_password(password)
user.save(using=self._db)
return user

   def create_superuser(self, email, password, **extra_fields):
       extra_fields.setdefault('is_staff', True)
       extra_fields.setdefault('is_superuser', True)

       if extra_fields.get('is_staff') is not True:
           raise ValueError('Superuser must have is_staff=True.')
       if extra_fields.get('is_superuser') is not True:
           raise ValueError('Superuser must have is_superuser=True.')

       return self.create_user(email, password, **extra_fields)

```

4. 创建自定义用户模型。

```
class CustomUser(AbstractBaseUser):
email = models.EmailField(max_length=255, unique=True)
first_name = models.CharField(max_length=255, null=True, blank=True)
last_name = models.CharField(max_length=255, null=True, blank=True)
is_active = models.BooleanField(default=True)
is_staff = models.BooleanField(default=False)

   objects = CustomUserManager()

   USERNAME_FIELD = 'email'
   REQUIRED_FIELDS = []

   def __str__(self):
       return self.email

   def get_full_name(self):
       if self.first_name and self.last_name:
           return f'{self.first_name} {self.last_name}'
       return self.email

   def get_short_name(self):
       return self.email

```

5. 在settings.py文件中修改AUTH_USER_MODEL选项为自定义用户模型类。

AUTH_USER_MODEL = 'my_app.CustomUser'

6. 执行数据库迁移操作，以对数据表进行更新。

python manage.py makemigrations python manage.py migrate

7. 在视图中使用自定义用户模型。

```
from django.contrib.auth import get_user_model
CustomUser = get_user_model()

# 创建用户
user = CustomUser.objects.create_user('pidancode.com', 'mypassword')
user.save()

# 创建超级用户
superuser = CustomUser.objects.create_superuser('皮蛋编程', 'mypassword')
superuser.save()
```