SQLAlchemy:如何根据其后端有条件地选择列的类型

Question

我想使用HSTORE如果列使用 PostgreSQL 作为后端，则输入列的类型，或 PickleType否则。问题是我们无法确定在定义模式时(在 Python 中)将使用哪个后端。当实际在后端数据库上创建表时，如何确定这一点并有条件地选择数据类型？

Answer 1

您可以使用 TypeEngine.with_variant 完成类似的事情:

from sqlalchemy.types import PickleType
from sqlalchemy.dialects import postgresql

HybridType = PickleType()

HybridType = HybridType.with_variant(postgresql.HSTORE(), 'postgresql')

这会创建一个新类型 HybridType，您可以像任何其他类型一样使用它，但需要注意的是，它将在 Postgres 上生成一个 HSTORE 列和一个 PickleType 其他地方。