如何用Django ORM实现周与年份的组合范围过滤

你之前用i_week__range和i_year__range组合过滤得到错误结果，核心原因是这两个条件是逻辑与关系，会筛选出i_year在指定年份区间且i_week在指定周数区间的记录，这和PostgreSQL中(i_week, i_year) BETWEEN (...) AND (...)的行比较逻辑完全不同。行比较是按元组整体排序判断：先比较年份，年份相同再比较周数，能精准匹配跨年度的周范围。

下面提供三种可行的实现方式：

方法1：使用RawSQL直接复用原生行比较逻辑 #

这种方式最直接，完全对应你给出的PostgreSQL语句：

from django.db.models import RawSQL
from myapp.models import Temp

# 定义筛选的起始和结束元组
start = (1, 2021)
end = (52, 2022)

# 执行查询并返回指定字段
results = Temp.objects.filter(
    RawSQL("(i_week, i_year) BETWEEN (%s, %s) AND (%s, %s)",
           [start[0], start[1], end[0], end[1]])
).values("i_week", "i_year")

方法2：用Django Func构造行表达式（ORM风格） #

在Django 3.2及以上版本，可以通过自定义Func类模拟PostgreSQL的ROW函数，实现数据库原生行比较：

from django.db.models import Func, F
from myapp.models import Temp

class Row(Func):
    function = 'ROW'
    arity = 2  # 指定参数个数为2

start = (1, 2021)
end = (52, 2022)

results = Temp.objects.filter(
    Row(F('i_week'), F('i_year')).__range__((start, end))
).values("i_week", "i_year")

Django会自动生成ROW(i_week, i_year) BETWEEN ROW(1, 2021) AND ROW(52, 2022)的SQL语句，和原生查询逻辑完全一致。

方法3：跨数据库兼容的Q对象组合逻辑 #

如果需要兼容非PostgreSQL数据库，可以手动拆分行比较的逻辑，用Q对象组合等价条件：

from django.db.models import Q
from myapp.models import Temp

start_year, start_week = 2021, 1
end_year, end_week = 2022, 52

results = Temp.objects.filter(
    # 中间年份的所有周
    Q(i_year__gt=start_year, i_year__lt=end_year) |
    # 起始年份的指定周及以后
    Q(i_year=start_year, i_week__gte=start_week) |
    # 结束年份的指定周及以前
    Q(i_year=end_year, i_week__lte=end_week)
).values("i_week", "i_year")

这种写法不依赖数据库特定语法，在MySQL、SQLite等数据库也能正常运行，但需要确保逻辑拆分准确。

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