增加了删除图片的功能。
This commit is contained in:
@@ -306,6 +306,39 @@ class MySQLManager:
|
||||
finally:
|
||||
session.close()
|
||||
|
||||
def delete_records(self, datasource: str, record_ids: List[int]) -> int:
|
||||
"""
|
||||
删除指定记录
|
||||
|
||||
Args:
|
||||
datasource: 数据源
|
||||
record_ids: 要删除的记录ID列表
|
||||
|
||||
Returns:
|
||||
成功删除的记录数
|
||||
"""
|
||||
session: Session = self.Session()
|
||||
try:
|
||||
Model = self.get_model_by_datasource(datasource)
|
||||
if Model is None:
|
||||
return 0
|
||||
|
||||
# 批量删除
|
||||
deleted_count = session.query(Model)\
|
||||
.filter(Model.id.in_(record_ids))\
|
||||
.delete(synchronize_session=False)
|
||||
|
||||
session.commit()
|
||||
print(f"成功从MySQL删除 {deleted_count} 条记录 (数据源: {datasource})")
|
||||
return deleted_count
|
||||
|
||||
except Exception as e:
|
||||
session.rollback()
|
||||
print(f"删除MySQL记录失败: {e}")
|
||||
return 0
|
||||
finally:
|
||||
session.close()
|
||||
|
||||
def close(self):
|
||||
"""关闭数据库连接"""
|
||||
if self.engine:
|
||||
|
||||
Reference in New Issue
Block a user