django重写user模型

1. 概述

在Django中，用户身份验证是一个非常常见的需求。Django提供了一个默认的用户模型（User Model），但在某些情况下，我们可能需要对该用户模型进行自定义或扩展，以满足特定的需求。为了解决这个问题，Django允许我们重写用户模型。本文将介绍如何在Django中重写用户模型。

2. 创建Django项目

首先，我们需要创建一个Django项目。如果已经有一个项目，可以跳过这一部分。可以使用以下命令创建一个新的Django项目：

django-admin startproject myproject

进入项目目录：

cd myproject

3. 创建新的用户模型

要重写Django用户模型，我们需要创建一个新的用户模型，并将其配置为Django项目的默认用户模型。首先，我们需要创建一个新的应用程序：

python manage.py startapp accounts

接下来，在新的应用程序目录（accounts）下创建一个名为models.py的文件，并添加以下代码：

from django.contrib.auth.models import AbstractUser

from django.db import models
class CustomUser(AbstractUser):
    # 添加自定义字段/属性
    age = models.IntegerField(null=True, blank=True)
    location = models.CharField(max_length=255, null=True, blank=True)
    def __str__(self):
        return self.username

上述代码中，我们从django.contrib.auth.models导入AbstractUser，这是Django默认的用户模型类。然后我们创建了一个名为CustomUser的新类，继承自AbstractUser。在CustomUser中，我们添加了两个自定义字段（age和location）。

4. 修改Django项目配置

接下来，我们需要修改Django项目的配置，以使用我们新创建的用户模型。打开项目目录下的settings.py文件，在AUTH_USER_MODEL变量中添加'accounts.CustomUser'：

AUTH_USER_MODEL = 'accounts.CustomUser'

5. 迁移数据库

在修改用户模型后，我们需要迁移数据库，以便更改生效。执行以下命令：

python manage.py makemigrations

python manage.py migrate

这将创建并应用数据库迁移文件。

6. 使用新的用户模型

现在，我们可以在项目中使用我们新创建的用户模型了。在视图（views）或其他地方，可以像使用默认用户模型一样使用CustomUser。

例如，在视图中创建一个新用户：

from accounts.models import CustomUser

def create_user(request):
    user = CustomUser.objects.create_user(username='example', password='password')
    # 执行其他操作
    return HttpResponse('User created successfully.')

如上所示，我们导入了CustomUser模型，并使用create_user创建了一个新用户。

7. 继承和扩展用户模型

除了重写用户模型外，我们还可以通过继承和扩展来添加自定义功能。例如，我们可以创建一个新的模型，它继承自CustomUser，并添加一些额外的字段或方法。

class UserProfile(models.Model):

    user = models.OneToOneField(CustomUser, on_delete=models.CASCADE)
    bio = models.TextField()
    avatar = models.ImageField(upload_to='avatars/')
    def __str__(self):
        return self.user.username

上述代码中，我们创建了一个名为UserProfile的新模型，它使用OneToOneField与CustomUser模型关联，并添加了一些额外的字段，如bio和avatar。

在这个例子中，我们可以使用CustomUser和UserProfile模型的相关信息，以构建一个更详细的用户配置。

8. 总结

通过重写用户模型，我们可以自定义和扩展Django中的默认用户模型。这使我们能够根据特定需求添加额外的字段和功能，以满足项目的要求。在本文中，我们介绍了如何创建新的用户模型、修改项目配置以及如何使用新的用户模型。