53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
# -*- 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() |