在Django框架中，开发人员经常需要扩展默认的`User`模型以满足特定项目的复杂需求。Django的内置`User`模型（`auth.User`）虽然提供了基础的身份验证功能，但通常不能涵盖如用户个人资料、额外信息等扩展字段。本篇文章将详细探讨如何解决在扩展Django自带`User`表时遇到的问题。
为了扩展`User`模型，我们需要自定义一个新的模型来继承Django的`AbstractUser`模型。`AbstractUser`是`User`模型的一个抽象版本，允许我们在不改动Django内部代码的情况下添加自定义字段和行为。例如，在`users`应用的`models.py`中，我们可以这样定义：
```python
from django.db import models
from django.contrib.auth.models import AbstractUser
class UserProfile(AbstractUser):
nick_name = models.CharField(max_length=20, verbose_name='昵称')
birthday = models.DateField(verbose_name='生日', null=True, blank=True)
gender = models.CharField(choices=(('male', '男'), ('female', '女')), verbose_name='性别', default='female', max_length=10)
address = models.CharField(max_length=50, verbose_name='地址', default='')
mobile = models.CharField(max_length=11, verbose_name='联系电话', null=True, blank=True)
image = models.ImageField(upload_to='image/%Y/%m', default='image/default.png', max_length=100)
class Meta:
verbose_name = '用户信息'
verbose_name_plural = verbose_name
def __str__(self):
return self.username
```
上述代码定义了一个名为`UserProfile`的新模型，它继承了`AbstractUser`并增加了如昵称、生日、性别、地址、电话和头像等字段。
接下来，为了使Django使用我们的自定义`UserProfile`模型，我们需要在项目的`settings.py`中设置`AUTH_USER_MODEL`。这告诉Django将用户模型替换为我们定义的新模型：
```python
AUTH_USER_MODEL = "users.UserProfile"
```
设置完成后，执行`makemigrations`和`migrate`命令创建和应用数据库迁移。然而，这里可能会遇到一个问题：在使用`User`的地方会报错，因为`User`已被替换为`UserProfile`。为了解决这个问题，我们需要在所有引用`User`的地方改为使用`get_user_model()`函数，这是Django提供的一种动态获取当前设置的用户模型的方法：
```python
from django.contrib.auth import get_user_model
User = get_user_model()
```
通过这种方法，无论用户模型是`User`还是`UserProfile`，代码都能正确地引用到当前设置的用户模型。
总结一下，扩展Django的`User`模型涉及以下几个关键步骤：
1. 定义一个继承自`AbstractUser`的新模型，添加所需的自定义字段。
2. 在`settings.py`中设置`AUTH_USER_MODEL`，指向新模型。
3. 使用`get_user_model()`函数替换代码中直接引用`User`的地方。
遵循这些步骤，可以确保在扩展Django的用户模型时避免常见的错误，并且保持代码的灵活性和可维护性。这在实际开发中是非常重要的，因为用户信息的定制化往往能提升项目的用户体验和功能完备性。
