如何为Django Segment模型QuerySet添加默认过滤排除归档数据

要实现每次查询Segment模型时自动排除status=ARCHIVE的记录，最稳妥的方式是自定义模型管理器（Manager），这也是Django官方推荐的做法，能避免全局修改带来的意外问题。

方法一：自定义Manager（推荐） #

1. 先定义一个自定义Manager类，重写get_queryset方法添加过滤逻辑：

from django.db import models
from .models import SegmentStatuses  # 确保导入你的状态枚举类

class SegmentManager(models.Manager):
    def get_queryset(self):
        # 默认查询自动排除ARCHIVE状态的记录
        return super().get_queryset().exclude(status=SegmentStatuses.ARCHIVE)
    
    def with_archived(self):
        # 额外提供一个方法，允许查询包含归档状态的全量数据（按需调用）
        return super().get_queryset()

2. 修改Segment模型，将默认的objects替换为这个自定义管理器：

class Segment(BaseSegmentModel):
    payload_json = JSONField(null=True, default=None, blank=True)
    owner = models.ForeignKey(CustomUser, null=True, blank=True, on_delete=models.SET_NULL, related_name='segments')
    name = models.CharField(max_length=100, null=True, blank=False)
    status = models.CharField(max_length=100, null=True, blank=True, choices=SegmentStatuses.choices, default=SegmentStatuses.STOPPED)
    created_date = models.DateTimeField(null=True, auto_now_add=True)
    
    # 替换默认管理器，启用自定义过滤逻辑
    objects = SegmentManager()

完成后，常规的Segment.objects.all()、Segment.objects.filter(...)等查询都会自动排除归档状态的记录。如果偶尔需要查询包含归档的全量数据，直接调用Segment.objects.with_archived()即可。

方法二：重写模型的get_queryset（不推荐，需注意继承影响） #

如果你的BaseSegmentModel没有自定义管理器，也可以直接在Segment模型中重写get_queryset，但这种方式灵活性较差，若父类模型已重写该方法可能会覆盖原有逻辑：

class Segment(BaseSegmentModel):
    # 原有字段定义...
    
    def get_queryset(self):
        return super().get_queryset().exclude(status=SegmentStatuses.ARCHIVE)

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