Django中如何让自定义QuerySet管理器调用模型类的方法？

我编写了一个重写update方法的QuerySet类QuerySetUpdateOverriden，代码如下：

class QuerySetUpdateOverriden(QuerySet, object):
    def update(self, *args, **kwargs):
        super().update(*args, **kwargs)
        if hasattr(self, 'method_from_object'):
            self.method_from_object()
            return

随后在MyObject类中通过as_manager()将其设为objects管理器：

class MyObject:
    objects = QuerySetUpdateOverriden.as_manager()
    def method_from_object(self):
        print("called")

调用update时，MyObject的method_from_object方法从未执行。问题出在QuerySet实例并非MyObject的实例，找不到该实例方法，且不存在让管理器继承模型的说法。需要实现调用该方法的需求。

场景1：方法无需操作具体实例（类级逻辑） #

如果method_from_object不需要依赖单个实例的属性，改成类方法后，直接在QuerySet中调用模型类方法即可：

修改MyObject的方法定义（注意：Django模型必须继承models.Model，原代码遗漏了这一点）：

from django.db import models

class MyObject(models.Model):
    objects = QuerySetUpdateOverriden.as_manager()
    
    @classmethod
    def method_from_object(cls):
        print("called")

修改QuerySet的update方法：

from django.db.models.query import QuerySet

class QuerySetUpdateOverriden(QuerySet):
    def update(self, *args, **kwargs):
        result = super().update(*args, **kwargs)
        # 通过self.model关联到模型类，调用类方法
        self.model.method_from_object()
        return result

场景2：需要对每个被更新的实例执行方法 #

Django的update()是批量SQL操作，不会加载实例到内存，也不会触发save()方法或信号。如果要对每个被更新的实例执行方法，需要先获取实例并逐个处理（数据量大时会影响性能，需谨慎使用）：

class QuerySetUpdateOverriden(QuerySet):
    def update(self, *args, **kwargs):
        # 先获取待更新的实例集合
        target_instances = self.all()
        # 执行批量更新
        result = super().update(*args, **kwargs)
        # 遍历实例，刷新数据后调用方法
        for instance in target_instances:
            instance.refresh_from_db()
            instance.method_from_object()
        return result

关键说明 #

• Django模型必须继承models.Model，否则无法正常使用ORM功能。
• 管理器与模型是关联关系，而非继承关系，管理器通过self.model属性绑定到对应模型类，无需让管理器继承模型。

内容的提问来源于stack exchange，提问作者Octavian Niculescu