Django Queryset annotate如何通过关联字段动态给日期加指定月份
解决方案
你之前的方案失效本质是relativeDelta属于Python层面的日期计算工具,不支持解析Django的F()表达式(表达式是直接转成SQL在数据库层执行的),所以没法直接在annotate里混合使用。可以用以下两种方案实现数据库层面的动态计算:
方案1:用Django 3.2及以上版本内置的AddMonths函数(推荐)
Django官方从3.2版本开始内置了专门的月份加减数据库函数,天然支持跨表取字段作为参数,写法如下:
from django.db.models import F, ExpressionWrapper, DateTimeField from django.db.models.functions import AddMonths # 注解出到期日期 coupon_uses = CouponUses.objects.annotate( expireDate=ExpressionWrapper( AddMonths(F('redeemDate'), F('coupon__numMonths')), output_field=DateTimeField() ) )
后续你可以直接用expireDate字段做过滤、聚合:
# 示例:过滤已经过期的优惠券使用记录 from django.utils import timezone expired_records = coupon_uses.filter(expireDate__lt=timezone.now())
这个方案是数据库无关的,Django会自动适配PostgreSQL、MySQL、SQLite等主流数据库的语法。
方案2:低版本Django自定义数据库函数实现
如果你的Django版本低于3.2,可以针对你使用的数据库写自定义Func实现:
以MySQL为例:
from django.db.models import Func, F, ExpressionWrapper, DateTimeField class MySQLAddMonths(Func): function = 'DATE_ADD' template = "%(function)s(%(expressions)s, INTERVAL %(months)s MONTH)" # 注解逻辑 coupon_uses = CouponUses.objects.annotate( expireDate=ExpressionWrapper( MySQLAddMonths(F('redeemDate'), months=F('coupon__numMonths')), output_field=DateTimeField() ) )
以PostgreSQL为例:
class PGAddMonths(Func): function = 'MAKE_INTERVAL' template = "%(expressions)s + %(function)s(months => %(months)s)" # 注解逻辑同上,替换MySQLAddMonths为PGAddMonths即可
注意事项
- 如果你的
redeemDate是DateField类型,把output_field改成DateField()即可 - 如果计算时需要处理月末对齐的问题(比如1月31日加1个月要得到2月28/29日),以上两种方案都默认遵循对应数据库的日期计算规则,和你用
relativeDelta的计算结果完全一致。
内容的提问来源于stack exchange,提问作者Yoel Acevedo
相关产品推荐
相关产品推荐

