Django中如何使用类的自定义方法作为过滤条件筛选查询集？

现有方案的核心问题 #

你当前的实现会先把所有Tutor表数据全部拉取到内存，再逐行调用自定义方法过滤，当家教数据量较大时，内存占用和查询耗时都会很高。

优化方案 #

1. 优先将过滤逻辑下沉到数据库层（最优解） #

你自定义方法里的三个判断规则，大部分都可以通过Django ORM直接转换成数据库查询，提前过滤掉绝大多数不符合要求的记录，仅把少量候选数据拉到内存做最终校验：

• 共同科目校验：如果科目是多对多关联字段，可直接用Tutor.objects.filter(subjects__in=student.get_subjects()).distinct()过滤
• 共同授课地点校验：逻辑和科目一致，用对应字段加in查询过滤
• 距离校验：如果使用PostGIS等地理字段扩展，可直接用filter(coordinate__distance_lte=(student.get_coordinates(), D(km=30)))完成数据库层面的距离过滤

经过上述三层过滤后，剩余的数据量会非常小，再调用自定义方法做二次校验即可，性能提升会非常明显。

2. 优化自定义判断方法的执行效率 #

你原方法中使用list.count()判断元素是否存在，时间复杂度为O(n)，可转成集合后用交集判断，代码更简洁，执行速度也更快：

def satisfies_min_requirements(self, student: Student) -> bool:
    # 校验是否有共同科目
    if not set(self.get_subjects()) & set(student.get_subjects()):
        return False
    # 校验是否有共同授课地点
    if not set(self.get_tutoring_places()) & set(student.get_tutoring_places()):
        return False
    # 校验距离是否符合要求
    return self.get_distance(student.get_coordinates()) <= 30

3. 轻量优化现有内存过滤写法 #

如果暂时无法将逻辑下沉到数据库层，可将原有的filter+lambda写法换成列表推导式，可读性更高，执行效率也有小幅提升：

# 数据量较大时可把Tutor.objects.all()换成Tutor.objects.all().iterator()，减少内存占用
tutors = [tutor for tutor in Tutor.objects.all() if tutor.satisfies_min_requirements(student)]

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