corp20250923

2025-09-23 17:28:37 +08:00
parent 89c79bdaa1
commit 5c8d07e5d7
13 changed files with 1366 additions and 86 deletions
@@ -0,0 +1,156 @@
+# -*- 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()