Django结合PostgREST使用Postgres DateRangeField存同日期范围异常求助

这个坑我之前踩过！核心原因是PostgreSQL的daterange类型默认是左闭右开区间（[start, end)），当你的起止日期是同一天时，这个区间其实没有包含任何有效日期，自然会被识别为empty=True的空范围。而你在shell里测试时应该是显式指定了闭合区间的边界，所以能正常存储。下面给你几个具体的解决办法：

1. 给模型字段指定闭合区间边界 #

直接在定义DateRangeField时设置bounds='[]'，强制使用闭合区间（[start, end]），这样同一天的起止日期会被正确存储为包含当天的范围：

from django.db import models
from django.contrib.postgres.fields import DateRangeField

class YourModel(models.Model):
    # 设置bounds为闭合区间，确保同一天的起止被正确存储
    date_range = DateRangeField(bounds='[]', null=True, blank=True)
    # 其他字段...

2. 构造DateRange对象时显式指定边界 #

如果不想全局修改字段的默认边界，你可以在创建/更新模型实例时，手动构造带闭合边界的DateRange对象：

from psycopg2.extras import DateRange
from datetime import date

# 创建同一天的闭合区间范围
same_day_range = DateRange(date(2024, 5, 20), date(2024, 5, 20), bounds='[]')

# 保存到模型
obj = YourModel.objects.create(date_range=same_day_range)
obj.save()

3. 在模型save方法中自动处理边界 #

如果希望模型自动处理这种情况，无需每次手动指定边界，可以重写模型的save方法：

from psycopg2.extras import DateRange

class YourModel(models.Model):
    date_range = DateRangeField(null=True, blank=True)

    def save(self, *args, **kwargs):
        # 检查起止日期是否相同，且当前不是闭合区间
        if (self.date_range is not None 
            and self.date_range.start == self.date_range.end 
            and self.date_range.bounds != '[]'):
            # 重新构造闭合区间的DateRange
            self.date_range = DateRange(
                self.date_range.start,
                self.date_range.end,
                bounds='[]'
            )
        super().save(*args, **kwargs)

验证效果 #

修改后你可以在shell里测试：

obj = YourModel.objects.get(pk=1)
print(obj.date_range)  # 应该输出 [2024-05-20,2024-05-20]
print(obj.date_range.empty)  # 输出 False，说明不是空范围

这样就能解决保存同一天起止日期变成空范围的问题啦！

内容的提问来源于stack exchange，提问作者César Montilla