# -*- 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()