Django：自定义用户模型的返回字段

Question

Django的User模型的默认返回字段是'username'。然而，我的应用需要它是'first_name'。 目前，我只是复制了Django并对User模型进行了更改：

def __str__(self):
    return self.first_name

它工作得很好，但我想知道是否有其他简单的方法可以在不分叉Django的情况下完成这个功能？

- Randy Tang

---

编辑库不是一个合适的解决方案。你应该子类化用户模型并在子类中使用str函数。 - ruddra

Answer 1

有一种方法可以在您的User类上注入一个新的方法。

from django.contrib.auth.models import User

def get_first_name(self):
    return self.first_name

User.add_to_class("__str__", get_first_name)

如果您只需要进行单一修改，您可以将此代码放入common应用程序中（最好是在models.py中）。

这种方式不够简洁，但是可以在不覆盖整个User模型的情况下完成任务。

- mariodev

---

动态地向现有类添加方法而不继承它，Python 简直让我大开眼界。 - Jordan Hyatt

Answer 2

我认为最好的方法是使用自定义用户模型，然后定义get_full_name()和/或get_short_name()来返回first_name。

- Travis

---

有时候你不想在项目中添加自定义用户模型，但这仍然是最好的方式。 - Alex M.M.

Answer 3

首先，使用Django文档创建自定义用户。

一旦您拥有自定义用户，请转到 MyUser 类。添加像first_name、last_name或您需要的任何字段。

您将得到如下内容。

class MyUser(AbstractBaseUser):
    email = models.EmailField(
        verbose_name='email address',
        max_length=255,
        unique=True,
    )
    first_name = models.CharField(max_length=32)
    last_name = models.CharField(max_length=32)
    date_of_birth = models.DateField()
    is_active = models.BooleanField(default=True)
    is_admin = models.BooleanField(default=False)

    objects = MyUserManager()

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['date_of_birth']

    def get_full_name(self):
        # The user is identified by their email address
        return self.email

    def get_short_name(self):
        # The user is identified by their email address
        return self.email

    def __str__(self):              # __unicode__ on Python 2
        return self.email

    def has_perm(self, perm, obj=None):
        "Does the user have a specific permission?"
        # Simplest possible answer: Yes, always
        return True

    def has_module_perms(self, app_label):
        "Does the user have permissions to view the app `app_label`?"
        # Simplest possible answer: Yes, always
        return True

    @property
    def is_staff(self):
        "Is the user a member of staff?"
        # Simplest possible answer: All admins are staff
        return self.is_admin

现在，请编辑您的类中的__str__函数为以下内容。

    def __str__(self):              # __unicode__ on Python 2
        return self.first_name

记住，像date_of_birth等字段是可选的。

- kartikmaji

Answer 4

与Randy Tang的答案类似：您可以使用代理模型来避免Ramast提到的问题：

class MyUser(User):
    class Meta:
        proxy = True

    def __str__(self):
        return self.first_name

根据文档：

您可以创建、删除和更新代理模型的实例，所有数据都将被保存，就像您使用原始（非代理）模型一样。

- pianoJames

Answer 5

以下解决方案可行吗？

class MyUser(User):

    def __str__(self):
        return self.first_name

不要再使用User，请改为使用MyUser。

- Randy Tang

---

遗憾的是，这将创建另一个名为my_user的数据库表，其字段与User模型相同。无论您在MyUser对象中存储什么值，都无法从User对象中访问，反之亦然。 - Ramast

正确。为了不再使用默认的用户模式，您需要执行以下操作：从django.contrib.auth.models导入AbstractUser，然后定义class MyUser(AbstractUser)。这样，您就可以通过MyUser访问first_name，而无需定义它。您还可以继承AbstractBaseUser，但这更难实现。 - Dan Swain

你也可以在Meta类中定义db_table属性为auth_user，以强制Django使用与contrib用户模型相同的数据库表。 - Eduard Luca