PostgreSQL 9+版本INET字段>=与<=操作符替代方案咨询

问题背景 #

在DRF应用中为PostgreSQL的INET字段自定义了查询逻辑，因Django原生不支持这类PostgreSQL专属字段，使用自定义IP字段支撑查询。目前多数查询正常，但以下两个自定义Lookup的查询失效：

class HostGreaterOrEqual(Lookup):
    """ Lookup to check if an IP address is greater than or equal. Used for net ranges"""
    lookup_name = "host_greater_or_equal"

    def as_sql(self, qn, connection):
        lhs, lhs_params = self.process_lhs(qn, connection)
        rhs, rhs_params = self.process_rhs(qn, connection)
        params = lhs_params + rhs_params
        return "%s >= %s" % (lhs, rhs), params
    
class HostLessOrEqual(Lookup):
    """ Lookup to check if an IP address is less than or equal. Used for net ranges"""
    lookup_name = "host_less_or_equal"

    def as_sql(self, qn, connection):
        lhs, lhs_params = self.process_lhs(qn, connection)
        rhs, rhs_params = self.process_rhs(qn, connection)
        params = lhs_params + rhs_params
        return "%s <= %s" % (lhs, rhs), params

模型中用address和address_end字段存储IP范围，希望通过以下过滤实现查询：

items = queryset.filter(
                Q(address__host_greater_or_equal=value)
                & Q(address_end__host_less_or_equal=value)
            )

查阅PostgreSQL 9版本文档可见INET字段的>=、<=操作符，但15版本文档中未找到相关条目，疑问：该功能是被移除还是有替代方案？

解答 #

PostgreSQL 15并没有移除INET/CIDR类型的大小比较功能，>=、<=操作符依然有效，只是15版文档调整了内容呈现方式，没有单独列出这些操作符（这类比较属于INET类型的默认行为，归并到类型的通用比较逻辑中）。

你的查询失效大概率是因为参数类型不匹配，数据库无法将传入的字符串自动转换为INET类型。可以通过以下两种方式修复：

1. 强制转换参数为INET类型 #

修改自定义Lookup的as_sql方法，将右侧参数显式转换为inet类型：

class HostGreaterOrEqual(Lookup):
    lookup_name = "host_greater_or_equal"

    def as_sql(self, qn, connection):
        lhs, lhs_params = self.process_lhs(qn, connection)
        rhs, rhs_params = self.process_rhs(qn, connection)
        params = lhs_params + rhs_params
        # 显式转换rhs为inet类型
        return "%s >= %s::inet" % (lhs, rhs), params
    
class HostLessOrEqual(Lookup):
    lookup_name = "host_less_or_equal"

    def as_sql(self, qn, connection):
        lhs, lhs_params = self.process_lhs(qn, connection)
        rhs, rhs_params = self.process_rhs(qn, connection)
        params = lhs_params + rhs_params
        return "%s <= %s::inet" % (lhs, rhs), params

2. 在自定义IP字段中处理类型转换 #

确保自定义IP字段的get_prep_value方法将输入的IP字符串转换为PostgreSQL可识别的INET格式，避免参数绑定后类型不兼容：

from django.db import models

class InetField(models.Field):
    def db_type(self, connection):
        return 'inet'

    def get_prep_value(self, value):
        # 确保传入的值是合法的IP字符串，直接返回即可（PostgreSQL会解析为inet）
        if value is None:
            return None
        return str(value)

额外验证 #

可以直接在PostgreSQL客户端执行测试语句，确认操作符有效性：

-- 测试INET比较
SELECT '192.168.1.10'::inet >= '192.168.1.5'::inet; -- 返回true
SELECT '192.168.1.10'::inet <= '192.168.1.15'::inet; -- 返回true

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