wuzhiqiang/ai-admin-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: optimize code

Browse Source
This commit is contained in:
martsforever
2026-03-20 14:46:25 +08:00
parent 7b9d1253a5
commit ab073d09b3
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/utils/create_module_service.py
+4 -4
View File
@@ -402,7 +402,7 @@ def create_model_service(
if not row_dict.get('id'):
raise HTTPException(status_code=400, detail="ID不能为空")
# 根据id查询要更新的记录
update_cls = (await session.exec(self.select_cls().where(Cls.id == row_dict.get('id')))).first()
update_cls = (await session.execute(self.select_cls().where(Cls.id == row_dict.get('id')))).scalars().first()
if not update_cls:
# 若记录不存在，抛出异常
raise HTTPException(status_code=500, detail="Update row not found")
@@ -442,7 +442,7 @@ def create_model_service(
# 提取所有待更新记录的id
update_id_list = [row_dict['id'] for row_dict in row_dict_list]
# 根据id查询所有待更新的记录
update_cls_list = (await session.exec(self.select_cls().where(Cls.id.in_(update_id_list)))).all()
update_cls_list = (await session.execute(self.select_cls().where(Cls.id.in_(update_id_list)))).all()
# 若查询到的记录数量与待更新数量不一致，说明部分id不存在
if len(update_cls_list) != len(row_dict_list):
# 抛出异常并提示不存在的id
@@ -480,7 +480,7 @@ def create_model_service(
await before_delete(row_dict, session)
# 根据id查询要删除的记录
delete_cls = (await session.exec(self.select_cls().where(Cls.id == row_dict.get('id')))).first()
delete_cls = (await session.execute(self.select_cls().where(Cls.id == row_dict.get('id')))).scalars().first()
if not delete_cls:
# 若记录不存在，返回删除失败
return False
@@ -509,7 +509,7 @@ def create_model_service(
row_id_list = [row_dict.get("id") for row_dict in row_dict_list]
# 根据id查询所有待删除的记录
delete_cls_list = (await session.exec(self.select_cls().where(Cls.id.in_(row_id_list)))).all()
delete_cls_list = (await session.execute(self.select_cls().where(Cls.id.in_(row_id_list)))).all()
# 若查询到的记录数量与待删除数量不一致，说明部分id不存在
if len(delete_cls_list) != len(row_id_list):
# 抛出异常并提示不存在的id