corp20250923

add_sqlite3_import.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+
+"""
+为api.py添加sqlite3导入
+"""
+
+import logging
+
+# 配置日志
+handlers = [logging.FileHandler('add_import_log.log'), logging.StreamHandler()]
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s',
+    handlers=handlers
+)
+logger = logging.getLogger(__name__)
+
+def main():
+    """
+    主函数：为api.py添加sqlite3导入
+    """
+    logger.info("开始为api.py添加sqlite3导入")
+    
+    try:
+        # 读取当前api.py文件内容
+        with open('api.py', 'r', encoding='utf-8') as f:
+            lines = f.readlines()
+        
+        # 检查是否已经导入了sqlite3
+        has_sqlite3_import = False
+        for line in lines:
+            if line.strip() == 'import sqlite3':
+                has_sqlite3_import = True
+                break
+        
+        # 如果没有导入，添加导入语句
+        if not has_sqlite3_import:
+            # 添加到文件开头
+            lines.insert(0, 'import sqlite3\n')
+            
+            # 写入更新后的文件
+            with open('api.py', 'w', encoding='utf-8') as f:
+                f.writelines(lines)
+            
+            logger.info("成功为api.py添加sqlite3导入")
+        else:
+            logger.info("api.py已经导入了sqlite3")
+            
+    except Exception as e:
+        logger.error(f"添加导入时发生错误: {str(e)}")
+
+if __name__ == "__main__":
+    main()