SQLAlchemy 2是否支持实现示例中的泛型Mixin?
在SQLAlchemy 2中实现泛型Mixin
完全可以实现,你给出的示例代码本身在SQLAlchemy 2.x版本中就是有效的写法,直接运行就能创建对应的数据库表。
SQLAlchemy 2.0对Python的泛型机制支持非常友好,结合typing.Generic和ORM的Mapped类型注解,就能轻松构建带泛型字段的Mixin类。你的代码逻辑清晰:
- 定义泛型Mixin
MyGenericMixin,继承自Generic[T],包含通用的主键字段id和泛型数据字段my_data - 实体类
MyIntImpl、MyStrImpl通过继承Base和指定泛型参数的MyGenericMixin,分别生成对应int、str类型的my_data列 - 最后通过
Base.metadata.create_all(engine)创建两张表my_int_impl和my_str_impl,其中my_data列的类型会自动匹配泛型参数对应的数据库类型(比如int对应INTEGER,str对应VARCHAR)
如果需要验证功能,可以补充如下测试代码:
from sqlalchemy.orm import Session with Session(engine) as session: # 插入测试数据 session.add(MyIntImpl(my_data=456)) session.add(MyStrImpl(my_data="hello sqlalchemy")) session.commit() # 查询验证 int_item = session.get(MyIntImpl, 1) print(int_item.my_data, type(int_item.my_data)) # 输出456 <class 'int'> str_item = session.get(MyStrImpl, 1) print(str_item.my_data, type(str_item.my_data)) # 输出hello sqlalchemy <class 'str'>
另外,如果需要对泛型字段做更精细的控制(比如指定长度、自定义类型),可以在mapped_column中添加参数,例如针对字符串类型指定长度:
class MyGenericMixin(Generic[T]): id: Mapped[int] = mapped_column(primary_key=True, unique=True, autoincrement=True, nullable=False) my_data: Mapped[T] = mapped_column(nullable=False, type_=String(255)) # 自定义字符串长度
内容的提问来源于stack exchange,提问作者Aart Stuurman
相关产品推荐
相关产品推荐

