Question

我在sqlalchemy中有一张下表：

class FieldType(enum.Enum):
   INT_FIELD = 0
   FLOAT_FIELD = 1
   STRING_FIELD = 2

class EAVTable(Base):
     __tablename__ = 'EAVTable'

     field_name = Column(Stirng, primary_key=True)
     field_type = Column(Enum(FieldType))
     int_field = Column(Integer)
     float_field = Column(Float)
     string_field = Column(String)

这是为了模拟符合我业务目的的EAV model。

现在要在代码中轻松使用它，我有以下hybrid_property。

@hybrid_propderty
def value(self):
    if self.field_type == FieldType.INT_FIELD:
         return self.int_field
    ...

@value.setter
def value(self, value):
    if type(value) == int:
        self.field_type = FieldType.INT_FIELD
        self.int_field = value
    ...

当我尝试获取并设置Python代码中的字段时，此方法工作正常。但是我仍然有一个问题：

session.query(EAVTable).filter(EAVTable.value == 123)

这不是开箱即用的，但是我有一个在案例中使用hybrid.expression的想法：

@value.expression
def value(cls):
    return case(
        [
            (cls.field_type == FieldType.INT_FIELD, cls.int_field),
            (cls.field_type == FieldType.FLOAT_FIELD, cls.float_field),
            ...
        ]
    )

从理论上讲，这是可行的，例如，为查询session.query(EAVTable.value = 123生成的SQL如下：

select * from where case 
    when field_type = INT_FIELD then int_field
    when field_type = FLOAT_FIELD then float_field
    when field_type = STRING_FIELD then string_field 
    end = 123;

在语义上看起来像我喜欢的样子，但是后来我发现case表达式要求所有情况都具有相同的类型，或者将它们转换为相同的类型。

我理解这是SQL语言的要求，与sqlachemy无关，但是对于经验丰富的sqlalchemy用户，有什么简单的方法可以实现我想要的目标？有没有办法解决这个限制？

Answer 1

您可以使用custom comparator在CASE表达式内移动比较：

from sqlalchemy.ext.hybrid import Comparator

class PolymorphicComparator(Comparator):
    def __init__(self, cls):
        self.cls = cls

    def __clause_element__(self):
        # Since SQL doesn't allow polymorphism here, don't bother trying.
        raise NotImplementedError(
            f"{type(self).__name__} cannot be used as a clause")

    def operate(self, op, other):
        cls = self.cls
        return case(
            [
                (cls.field_type == field_type, op(field, other))
                for field_type, field in [
                    (FieldType.INT_FIELD, cls.int_field),
                    (FieldType.FLOAT_FIELD, cls.float_field),
                    (FieldType.STRING_FIELD, cls.string_field),
                ]
            ],
            else_=False
        )

class EAVTable(Base):
     ...

     # This replaces @value.expression
     @value.comparator
     def value(cls):
         return PolymorphicComparator(cls)

这样，普通类型就是布尔值。

如何动态选择要在where子句中使用的不同类型的列？

1 个答案: