重写Django RelatedManager与QuerySet，基于关联模型字段过滤

我明白你的痛点了——你想通过LinkMixin复用中间表的状态管理逻辑，不想在每个关联的模型上重复写管理器，但Django默认的多对多管理器只会从目标模型（这里是Gene）找指定的管理器，所以才会触发AttributeError: type object 'Gene' has no attribute 'statuses'这个错误。

下面是一套基于你的需求设计的解决方案，既能复用LinkMixin的逻辑，又能让你直接通过genes.active()这种简洁的方式获取过滤后的结果：

步骤1：创建自定义多对多关联管理器 #

这个管理器会代理中间模型（继承LinkMixin）的statuses QuerySet方法，让你无需在目标模型（Gene）中重复定义任何逻辑：

from django.db.models.fields.related import ManyToManyManager

class LinkMixinRelatedManager(ManyToManyManager):
    def __getattr__(self, name):
        # 检查中间模型的statuses管理器是否有对应的方法
        through_queryset = self.through.statuses
        if hasattr(through_queryset, name):
            method = getattr(through_queryset, name)
            if callable(method):
                # 包装方法，实现基于中间表过滤目标模型的逻辑
                def wrapper(*args, **kwargs):
                    # 获取中间模型中关联当前SampleBatch实例的过滤条件
                    source_field = self.rel.source_field_name  # 对应'samplebatch'字段
                    filtered_through = method(*args, **kwargs).filter(**{source_field: self.instance})
                    # 获取符合条件的Gene ID列表
                    target_field = self.rel.target_field_name  # 对应'gene'字段
                    related_gene_ids = filtered_through.values_list(f"{target_field}_id", flat=True)
                    # 返回过滤后的Gene QuerySet
                    return self.get_queryset().filter(pk__in=related_gene_ids)
                return wrapper
        # 如果找不到对应方法，调用父类的默认处理
        return super().__getattr__(name)

步骤2：在多对多字段中指定自定义管理器 #

修改SampleBatch模型的genes字段，添加manager参数指定我们刚创建的自定义管理器：

class SampleBatch(models.Model):
    sample = models.ForeignKey(Sample, on_delete=models.CASCADE)
    batch = models.ForeignKey(Batch, on_delete=models.CASCADE)
    genes = models.ManyToManyField(
        "Gene",
        through='SampleBatchGene',
        related_name='samplebatches',
        blank=True,
        through_fields=('samplebatch', 'gene'),
        manager=LinkMixinRelatedManager()  # 这里指定自定义管理器
    )
    # ... 其他原有代码保持不变

步骤3：测试效果 #

现在你就可以像预期的那样调用方法了：

SampleBatch.objects.get(sample__lab_number="18_11998").genes.active()
# 输出：<QuerySet [<Gene: BAP1>]>

SampleBatch.objects.get(sample__lab_number="18_11998").genes.inactive()
# 输出符合条件的非活跃Gene QuerySet

为什么这个方案可行？ #

• 我们的自定义管理器通过__getattr__动态代理了LinkMixin中statuses QuerySet的方法（active、inactive）
• 它会先过滤中间表SampleBatchGene中符合状态条件且关联当前SampleBatch的记录，再通过这些记录获取对应的Gene ID，最终返回过滤后的Gene QuerySet
• 完全复用了LinkMixin的逻辑，不需要在Gene或其他关联模型中重复写任何管理器代码

备选方案：手动使用Prefetch（如果不想自定义管理器） #

如果你暂时不想自定义管理器，也可以用Prefetch对象实现类似效果，但写法会繁琐一些：

from django.db.models import Prefetch

# 获取指定批次，并预取活跃的基因
batch = SampleBatch.objects.prefetch_related(
    Prefetch(
        'genes',
        queryset=Gene.objects.filter(samplebatchgene__active=1),
        to_attr='active_genes'
    )
).get(sample__lab_number="18_11998")

# 访问活跃基因
batch.active_genes

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