在 Django 中从基本模型实例转换为派生代理模型？

Question

我想为 Django 的默认 User 类定义一个代理模型，类似于这样：

class MyUser(User):

    def pretty_username(self): 
        if self.first_name:
            return self.first_name
        return self.username 

    class Meta: 
        proxy = True 

而且，我希望能够从视图代码（理想情况下，甚至从模板）调用 Pretty_username 。是否有一种简单的方法来获取标准用户模型的实例并将其类型转换为 MyUser 的实例？

即使是一些 __init__ 魔法对我来说也没什么问题，只要我能说：

my_user = MyUser(request.user) 

在我的视图代码中。

Answer 1

我认为这个问题的实际答案应该是@fhahn 在另一个答案中的评论。通过更改类，我们可以避免额外的数据库调用。以下是示例代码：

我的代理模型，如果设置，则将表示形式从 username 更改为 email：

class MyUser(User):
    class Meta:
        proxy = True
        verbose_name = _('my user')
        verbose_name_plural = _('my users')

    def __str__(self):
        return "%s" % (self.email or self.username)

class Wallet(models.Model):
    owner = models.OneToOneField(MyUser, on_delete=models.PROTECT, related_name='wallet')

    def __str__(self):
        return "%s" % self.owner

在 shell 中进行简短测试：

>>> from django.contrib.auth.models import User
>>> from my_apps.models import MyUser
>>> user = User.objects.get(pk=4)
>>> user
<User: AbKec6rumI9H9UmAC3Bh2kXUHzj4>
>>> user.email
'[email protected]'
>>> user.wallet
Traceback (most recent call last):
  File "<console>", line 1, in <module>
AttributeError: 'User' object has no attribute 'wallet'
>>> user.__class__ = MyUser
>>> user
<MyUser: [email protected]>
>>> user.wallet
<Wallet: [email protected]>

Answer 2

如果您确实希望拥有完整的代理对象可用，那么这是一个快速而肮脏的解决方案（以额外的数据库调用为代价）

class MyUser(User):

    def pretty_username(self): 
        if self.first_name:
            return self.first_name
        return self.username 

    class Meta: 
        proxy = True


def get_myuser(self):
    try:
        return MyUser.objects.get(pk=self.pk)
    except MyUser.DoesNotExist:
        return None

User.add_to_class('get_myuser', get_myuser)

因此，要在视图中使用它，您可以说：

request.user.get_myuser().pretty_username()

或者在模板中：

{{ request.user.get_myuser.pretty_username }}

一个更好的解决方案，如果您不受代理模型想法的束缚，将是以下内容：

def pretty_username(self):
    if self.first_name:
        return self.first_name
    return self.username

User.add_to_class('pretty_username', pretty_username)

这将允许以下内容：

request.user.pretty_username()

或者

{{ request.user.pretty_username }}