扩展 auth.User 模型、代理字段和 Django 管理

Question

（编辑：我知道 Django 中有一个完全独立的功能，称为“代理模型”。该功能对我没有帮助，因为我需要能够向 UserProfile 添加字段。）

所以我正在启动一个新的 Django 应用程序，我正在创建一个 UserProfile 模型，它是 django.contrib.auth.models.User 的扩展，并将失败的属性请求返回给 User，如下所示：

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

class UserProfile(models.Model):
    user = models.OneToOneField(User, related_name='profile')

    def __getattr__(self, name, *args):
        if name == 'user' or name == '_user_cache':
            raise AttributeError(name)

        try:
            return getattr(self.user, name, *args)
        except AttributeError, e:
            raise AttributeError(name)

这通常工作正常，但当我尝试使用 User 时会中断UserProfileAdmin.list_display 中的 字段。问题出在此处的管理验证代码中：

def validate(cls, model):
    """
    Does basic ModelAdmin option validation. Calls custom validation
    classmethod in the end if it is provided in cls. The signature of the
    custom validation classmethod should be: def validate(cls, model).
    """
    # Before we can introspect models, they need to be fully loaded so that
    # inter-relations are set up correctly. We force that here.
    models.get_apps()

    opts = model._meta
    validate_base(cls, model)

    # list_display
    if hasattr(cls, 'list_display'):
        check_isseq(cls, 'list_display', cls.list_display)
        for idx, field in enumerate(cls.list_display):
            if not callable(field):
                if not hasattr(cls, field):
                    if not hasattr(model, field):
                        try:
                            opts.get_field(field)
                        except models.FieldDoesNotExist:
                            raise ImproperlyConfigured("%s.list_display[%d], %r is not a callable or an attribute of %r or found in the model %r."
                                % (cls.__name__, idx, field, cls.__name__, model._meta.object_name))

问题在于，虽然 UserProfile 的实例将具有代理字段（例如电子邮件），但 UserProfile 类本身却没有。在 Django shell 中演示：

>>> hasattr(UserProfile, 'email')
False
>>> hasattr(UserProfile.objects.all()[0], 'email')
True

经过一番挖掘，看起来我想覆盖 UserProfile._meta 的 django.db.models.options.Options.get_field 。但似乎没有一种非 hacky 的方法来做到这一点（我现在有一个非常 hacky 的解决方案，其中涉及猴子修补 UserProfile._meta.[get_field, get_field_by_name]）...有什么建议吗？谢谢。

Answer 1

保持简单。这是我们使用的库中的 UserProfile 模型的示例：

class UserProfile(models.Model):
    user = models.OneToOneField(User)
    accountcode = models.PositiveIntegerField(null=True, blank=True)

就是这样。不要为 __getattr__ 覆盖而烦恼。相反，自定义管理界面：

from django.contrib.auth.admin import UserAdmin
from django.contrib.auth.models import User

class UserProfileInline(admin.StackedInline):
    model = UserProfile

class StaffAdmin(UserAdmin):
    inlines = [UserProfileInline]
    # provide further customisations here

admin.site.register(User, StaffAdmin)

这允许您对 User 对象进行 CRUD，并以内联方式访问 UserProfile。现在，您不必将属性查找从 UserProfile 代理到 User 模型。要从 User u 实例访问 UserProfile，请使用 u.get_profile()

Answer 2

这不是一个代理类，而是一种关系。有关代理模型的更多信息，这是一个原始模型的子类，带有 Meta.proxy = True

Answer 3

如果您只想将 User 中的字段显示在 UserProfileAdmin 的 list_display 中，请尝试：

class UserProfileAdmin(admin.ModelAdmin):
    list_display = ('user__email',)

如果您希望将其作为表单的一部分，请将其作为额外字段添加到 UserProfileForm 中，并在表单中验证它。