Django User 模型，添加功能

Question

我想向 Django 的默认用户模型添加一个新函数，用于检索模型类型的相关列表。

这样的Foo模型：

class Foo(models.Model):
    owner = models.ForeignKey(User, related_name="owner")
    likes = models.ForeignKey(User, related_name="likes")

........

    #at some view
    user = request.user
    foos= user.get_related_foo_models()

这如何实现？

Answer 1

正如文档中所述，替换用户模型并不罕见： https://docs.djangoproject.com/es/1.9/topics/auth/customizing/#substituting-a-custom-user-model，因此，考虑到这一点，最好使用以下代码获取用户模型类：

from django.contrib.auth import get_user_model
UserModel = get_user_model()

之后，您可以使用此 UserModel 按照 @Lakshman Prasad 的建议添加功能：UserModel.add_to_class('get_lated_foo_models', get_lated_foo_models)< /代码>。

---

为了让代码只执行一次，我更喜欢使用 Django 应用程序配置类 (https: //docs.djangoproject.com/es/1.9/ref/applications/），所以一个完整的工作示例将是：

# myapp/__init__.py
default_app_config = 'myapp.apps.MyAppConfig'

# myapp/apps.py
from django.apps import AppConfig
from django.contrib.auth import get_user_model


class MyAppConfig(AppConfig):
    name = 'myapp'
    verbose_name = 'MyApp'

    def ready(self):
        # Add some functions to user model:
        def custom_function(self):
            # Do whatsoever
            pass

        UserModel = get_user_model()
        UserModel.add_to_class('custom_function', custom_function)

Answer 2

您可以使用代理模型向用户模型添加方法，而无需更改它的类。

from django.contrib.auth import models

class UserProxy(models.User):
    class Meta:
        proxy = True
    
    def get_related_foo_models(self):
        pass

获取视图中代理模型的实例 - models.UserProxy.objects.get(pk=request.user.pk)

要立即从请求中获取代理模型的实例，您需要创建如本问题所述的自定义 ModelBackend - Django - 用户代理来自请求的模型。

来源 https://docs.djangoproject.com/en/ 4.1/topics/db/models/#proxy-models

Answer 3

您可以向 User 添加一个方法

from django.contrib import auth
auth.models.User.add_to_class('get_related_foo_models', get_related_foo_models)

，确保您在 models.py 或其他在 django 启动时导入的文件中拥有此代码。

Answer 4

这是@Lakshman Prasad 答案的更新。但完整的示例是：

在任何 apps:: 中创建一个文件 monkey_patching.py，

#app/monkey_patching.py
from django.contrib.auth.models import User 

def get_user_name(self):
    if self.first_name or self.last_name:
        return self.first_name + " " + self.last_name
    return self.username

User.add_to_class("get_user_name",get_user_name)

并将其导入到应用程序的 __init__.py 文件中。 IE：：

#app/__init__.py
import monkey_patching