如何使用sqlachemy对整数支持的枚举进行建模？

Question

我使用sqlalchemy作为一种可读的方式来建模我的数据库，我只对从我的模型生成几个引擎的数据库定义感兴趣。

我的表中的一些列具有Enum类型，它在MySQL等引擎中工作得很好，因为它具有本机枚举支持。但是对于SQL Server，它将列生成为VARCHAR，并设置一个约束来检查这些值是否在我指定的预期枚举值范围内。

我想用基于数字的后备替换这个替代方法，这样列类型实际上是数字的，并且约束检查数值是否在枚举大小的范围内(假设从0开始的连续值)。

我尝试用Enum作为impl创建一个TypeDecorator，但这还不够，或者我不知道如何使它工作。我还尝试只复制Boolean类型的代码，并将其与Enum类型混合，以创建我自己的类型，但似乎也需要数据库编译器支持。

有没有一种方法可以在不修补sqlalchemy本身的情况下实现这一点？

请注意，我对使用python查询数据库不感兴趣，在它生成之后，我就完成了，所以这可能会简化。

Answer 1

这是你需要的：

import sqlalchemy as sa

class IntEnum(sa.types.TypeDecorator):
    impl = sa.Integer
    def __init__(self, enumtype, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self._enumtype = enumtype

    def process_bind_param(self, value, dialect):
        return value.value

    def process_result_value(self, value, dialect):
        return self._enumtype(value)

然后像这样使用它：

from enum import Enum
from sqlalchemy.ext.declarative import declarative_base

class MyEnum(Enum):
    one = 1
    two = 2
    three = 3

engine = sa.create_engine('sqlite:///:memory:')
session = sa.orm.sessionmaker(bind=engine)()
Base = declarative_base()

class Stuff(Base):
    __tablename__ = 'stuff'

    id = sa.Column('id', sa.Integer, primary_key=True)
    thing = sa.Column('num', IntEnum(MyEnum))

Base.metadata.create_all(engine)

session.add(Stuff(thing=MyEnum.one))
session.add(Stuff(thing=MyEnum.two))
session.add(Stuff(thing=MyEnum.three))
session.commit()
engine.execute(sa.text('insert into stuff values(4, 2);'))

for thing in session.query(Stuff):
    print(thing.id, thing.thing)

实际上，您唯一的问题是impl需要是一个sa.Integer，因为这才是真正支持枚举的东西，而不是enum.Enum。