如何用sqlalchemy建模由整数支持的枚举? - python

我使用sqlalchemy作为对数据库进行建模的易读方法,我只想从模型中为多个引擎生成数据库定义。

我表中的某些列的类型为Enum,由于它具有本机枚举支持,因此在诸如MySQL之类的引擎中运行良好。但是,对于SQL Server,它将列生成为VARCHAR并设置约束以检查值是否在我指定的预期枚举值之内。

我想用基于数字的后备替换此替代方法,以使列类型实际上是数字,并且约束检查数字值是否在枚举大小的范围内(假设顺序值从0开始)。

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

有没有一种方法可以实现此目的而不必修补sqlalchemy本身?

请注意,我对使用python查询数据库不感兴趣,因为在生成之后,我完成了操作,因此也许可以简化。

参考方案

这是您需要的:

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的支持。

在返回'Response'(Python)中传递多个参数 - python

我在Angular工作,正在使用Http请求和响应。是否可以在“响应”中发送多个参数。角度文件:this.http.get("api/agent/applicationaware").subscribe((data:any)... python文件:def get(request): ... return Response(seriali…

Python exchangelib在子文件夹中读取邮件 - python

我想从Outlook邮箱的子文件夹中读取邮件。Inbox ├──myfolder 我可以使用account.inbox.all()阅读收件箱,但我想阅读myfolder中的邮件我尝试了此页面folder部分中的内容,但无法正确完成https://pypi.python.org/pypi/exchangelib/ 参考方案 您需要首先掌握Folder的myfo…

Python GPU资源利用 - python

我有一个Python脚本在某些深度学习模型上运行推理。有什么办法可以找出GPU资源的利用率水平?例如,使用着色器,float16乘法器等。我似乎在网上找不到太多有关这些GPU资源的文档。谢谢! 参考方案 您可以尝试在像Renderdoc这样的GPU分析器中运行pyxthon应用程序。它将分析您的跑步情况。您将能够获得有关已使用资源,已用缓冲区,不同渲染状态上…

R'relaimpo'软件包的Python端口 - python

我需要计算Lindeman-Merenda-Gold(LMG)分数,以进行回归分析。我发现R语言的relaimpo包下有该文件。不幸的是,我对R没有任何经验。我检查了互联网,但找不到。这个程序包有python端口吗?如果不存在,是否可以通过python使用该包? python参考方案 最近,我遇到了pingouin库。

Python uuid4,如何限制唯一字符的长度 - python

在Python中,我正在使用uuid4()方法创建唯一的字符集。但是我找不到将其限制为10或8个字符的方法。有什么办法吗?uuid4()ffc69c1b-9d87-4c19-8dac-c09ca857e3fc谢谢。 参考方案 尝试:x = uuid4() str(x)[:8] 输出:"ffc69c1b" Is there a way to…