156 lines
5.1 KiB
Python
156 lines
5.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
修复update_query_result函数中的问题
|
|
主要解决以下问题：
|
|
1. 受影响行数的获取逻辑不准确
|
|
2. 添加更详细的错误处理
|
|
3. 确保数据库事务的正确性
|
|
"""
|
|
|
|
import logging
|
|
|
|
# 配置日志
|
|
handlers = [logging.FileHandler('fix_log.log'), logging.StreamHandler()]
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s - %(levelname)s - %(message)s',
|
|
handlers=handlers
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def main():
|
|
"""
|
|
主函数：执行修复操作
|
|
"""
|
|
logger.info("开始修复update_query_result函数")
|
|
|
|
try:
|
|
# 读取当前api.py文件内容
|
|
with open('api.py', 'r', encoding='utf-8') as f:
|
|
content = f.read()
|
|
|
|
# 定义需要替换的代码块和新代码块
|
|
old_code = """ # 获取受影响的行数
|
|
affected_rows = cursor.rowcount
|
|
|
|
# 关闭数据库连接
|
|
close_db_connection(conn)
|
|
|
|
# 检查是否有记录被更新
|
|
if affected_rows == 0:
|
|
return jsonify({
|
|
'success': False,
|
|
'error': '未找到id为 {} 的记录'.format(query_id),
|
|
'data': None
|
|
}), 404"""
|
|
|
|
new_code = """ # 检查query_history表是否有记录被更新
|
|
# 重新查询来确认更新是否成功
|
|
cursor.execute("SELECT 1 FROM query_history WHERE id = ?", (query_id,))
|
|
query_exists = cursor.fetchone() is not None
|
|
|
|
# 关闭数据库连接
|
|
close_db_connection(conn)
|
|
|
|
# 检查是否存在该记录
|
|
if not query_exists:
|
|
return jsonify({
|
|
'success': False,
|
|
'error': '未找到id为 {} 的记录'.format(query_id),
|
|
'data': None
|
|
}), 404"""
|
|
|
|
# 执行替换
|
|
if old_code in content:
|
|
new_content = content.replace(old_code, new_code)
|
|
|
|
# 写入修复后的代码
|
|
with open('api.py', 'w', encoding='utf-8') as f:
|
|
f.write(new_content)
|
|
|
|
logger.info("成功修复update_query_result函数，解决了受影响行数的获取问题")
|
|
|
|
# 添加更详细的异常处理
|
|
add_detailed_exception_handling()
|
|
|
|
else:
|
|
logger.warning("未找到需要修复的代码块，可能已经被修改过")
|
|
|
|
except Exception as e:
|
|
logger.error(f"修复过程中发生错误: {str(e)}")
|
|
|
|
def add_detailed_exception_handling():
|
|
"""
|
|
为update_query_result函数添加更详细的异常处理
|
|
"""
|
|
try:
|
|
# 读取当前api.py文件内容
|
|
with open('api.py', 'r', encoding='utf-8') as f:
|
|
content = f.read()
|
|
|
|
# 定义需要替换的代码块和新代码块
|
|
old_except_block = """ except Exception as e:
|
|
# 确保关闭数据库连接
|
|
if 'conn' in locals():
|
|
close_db_connection(conn)
|
|
return jsonify({
|
|
'success': False,
|
|
'error': str(e),
|
|
'data': None
|
|
}), 500"""
|
|
|
|
new_except_block = """ except sqlite3.IntegrityError as e:
|
|
# 确保关闭数据库连接
|
|
if 'conn' in locals():
|
|
close_db_connection(conn)
|
|
logger.error(f"数据库完整性错误: {str(e)}")
|
|
return jsonify({
|
|
'success': False,
|
|
'error': f'数据库完整性错误: {str(e)}',
|
|
'data': None
|
|
}), 500
|
|
except sqlite3.Error as e:
|
|
# 确保关闭数据库连接
|
|
if 'conn' in locals():
|
|
close_db_connection(conn)
|
|
logger.error(f"数据库错误: {str(e)}")
|
|
return jsonify({
|
|
'success': False,
|
|
'error': f'数据库错误: {str(e)}',
|
|
'data': None
|
|
}), 500
|
|
except Exception as e:
|
|
# 确保关闭数据库连接
|
|
if 'conn' in locals():
|
|
close_db_connection(conn)
|
|
logger.error(f"服务器错误: {str(e)}")
|
|
return jsonify({
|
|
'success': False,
|
|
'error': f'服务器错误: {str(e)}',
|
|
'data': None
|
|
}), 500"""
|
|
|
|
# 执行替换
|
|
if old_except_block in content:
|
|
new_content = content.replace(old_except_block, new_except_block)
|
|
|
|
# 同时添加sqlite3导入
|
|
if 'import sqlite3' not in content:
|
|
import_line = "import sqlite3\n"
|
|
new_content = import_line + new_content
|
|
logger.info("成功添加sqlite3导入")
|
|
|
|
# 写入修复后的代码
|
|
with open('api.py', 'w', encoding='utf-8') as f:
|
|
f.write(new_content)
|
|
|
|
logger.info("成功为update_query_result函数添加了详细的异常处理")
|
|
else:
|
|
logger.warning("未找到需要修复的异常处理代码块")
|
|
|
|
except Exception as e:
|
|
logger.error(f"添加详细异常处理时发生错误: {str(e)}")
|
|
|
|
if __name__ == "__main__":
|
|
main() |