使用 dict 方法
这种方式的不需要模型配合，便可以直接把查询数据转为字典。不过对 sqlalchemy 来讲字典中会多一个 _sa_instance_state 需要移除。
def get(self, table, column, value) -> dict:
""" 获取单条信息"""
with self.session.begin() as session:
table_column = getattr(table, column)
stmt = select(table).where(table_column == value)
result = session.execute(stmt).scalar()
if not result:
return {}
result = result.__dict__ # 转为字典
if "_sa_instance_state" in result:
del result["_sa_instance_state"] # 删除指定键
return result使用 dict 方法
def get(self, table, column, value) -> dict:
""" 获取单条信息"""
with self.session.begin() as session:
table_column = getattr(table, column)
stmt = select(table).where(table_column == value)
result = session.execute(stmt).scalar()
if not result:
return {}
return dict(result)使用函数
可以自定义需要返回的数据，甚至是可以
class Infos(Base, GetDict):
name = Column(String(32))
age = Column(Integer)
def to_dict(self):
return {
"name": self.name,
"age": self.age,
}模版
class GetDict:
def keys(self):
k_list = []
for key in self.__mapper__.c.keys():
k_list.append(key)
return k_list
def __getitem__(self, item):
return getattr(self, item)对应模型
class Infos(Base, GetDict):
name = Column(String(32))
age = Column(Integer)调用
infos = Infos()
print(dict(infos))调用时需要使用 dict。
分表
class Test():
def model(suffix):
tablename = f"test{suffix}"
classname = f"test{suffix}"
# 这里也可以传递上面的 GetDict
Model = type(classname, (Base, ), {
'__module__': __name__,
'__name__': classname,
'__tablename__': tablename,
"id" : Column(String(36), primary_key=True,
default=str(uuid.uuid4())),
"name": Column(Integer),
"create_time" : Column(DateTime, server_default=func.now(),
comment="创建时间"),
"update_time" : Column(DateTime, server_default=func.now(),
onupdate=func.now(),
comment='修改时间'),
})
return Model调用
# model 返回的便是类（表模型）
model = Test.model("42")
db.get_all(model, {})