110 lines
3.9 KiB
Python
110 lines
3.9 KiB
Python
|
import os
|
|||
|
|
from utils.execute_sql import execute_sql
|
|||
|
|
from utils.date_time import get_parameter
|
|||
|
|
from utils.log import Log
|
|||
|
|
from utils.loadconfig import get_path
|
|||
|
|
|
|||
|
|
|
|||
|
|
def read_sql_files(directory):
|
|||
|
|
log = Log().getlog()
|
|||
|
|
|
|||
|
|
|
|||
|
|
directory = get_path() + '/sql/' + directory
|
|||
|
|
"""
|
|||
|
|
读取指定目录下所有.sql结尾的文件内容并打印
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
directory (str): 要读取文件的目录路径
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
None
|
|||
|
|
"""
|
|||
|
|
# 检查目录是否存在
|
|||
|
|
if not os.path.exists(directory):
|
|||
|
|
log.warning(f"目录 {directory} 不存在，请检查路径。")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 检查目录是否为空
|
|||
|
|
if not os.listdir(directory):
|
|||
|
|
log.warning(f"目录 {directory} 为空。")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 遍历目录及其子目录下的所有文件
|
|||
|
|
for root, dirs, files in os.walk(directory):
|
|||
|
|
for file in files:
|
|||
|
|
if file.endswith('.sql'):
|
|||
|
|
file_path = os.path.join(root, file)
|
|||
|
|
try:
|
|||
|
|
with open(file_path, 'r', encoding='utf-8') as f:
|
|||
|
|
content = f.read()
|
|||
|
|
# 遍历字典，将字符串中的占位符替换为对应的值
|
|||
|
|
for key, value in get_parameter().items():
|
|||
|
|
content = content.replace(key, str(value))
|
|||
|
|
log.info(f"文件路径: {file_path}")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
returnstr = execute_sql(content)
|
|||
|
|
log.info(f"执行结果: {returnstr}")
|
|||
|
|
log.info("-" * 50)
|
|||
|
|
except Exception as e:
|
|||
|
|
log.error(f"执行 SQL 文件 {file_path} 时出现错误: {e}")
|
|||
|
|
|
|||
|
|
except UnicodeDecodeError:
|
|||
|
|
log.error(f"文件 {file_path} 编码格式可能不是utf-8，无法正确读取，请检查。")
|
|||
|
|
except Exception as e:
|
|||
|
|
log.error(f"读取文件 {file_path} 时出现错误: {e}")
|
|||
|
|
|
|||
|
|
def read_one_game_sql_files(directory,game_information):
|
|||
|
|
log = Log().getlog()
|
|||
|
|
|
|||
|
|
directory = get_path() + '/one_game_sql/' + directory
|
|||
|
|
"""
|
|||
|
|
读取指定目录下所有.sql结尾的文件内容并打印
|
|||
|
|
|
|||
|
|
参数:
|
|||
|
|
directory (str): 要读取文件的目录路径
|
|||
|
|
|
|||
|
|
返回:
|
|||
|
|
None
|
|||
|
|
"""
|
|||
|
|
# 检查目录是否存在
|
|||
|
|
if not os.path.exists(directory):
|
|||
|
|
log.warning(f"目录 {directory} 不存在，请检查路径。")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 检查目录是否为空
|
|||
|
|
if not os.listdir(directory):
|
|||
|
|
log.warning(f"目录 {directory} 为空。")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 遍历目录及其子目录下的所有文件
|
|||
|
|
for root, dirs, files in os.walk(directory):
|
|||
|
|
for file in files:
|
|||
|
|
if file.endswith('.sql'):
|
|||
|
|
file_path = os.path.join(root, file)
|
|||
|
|
try:
|
|||
|
|
with open(file_path, 'r', encoding='utf-8') as f:
|
|||
|
|
content = f.read()
|
|||
|
|
parameter=get_parameter()
|
|||
|
|
parameter.update(game_information)
|
|||
|
|
# 遍历字典，将字符串中的占位符替换为对应的值
|
|||
|
|
for key, value in parameter.items():
|
|||
|
|
content = content.replace(key, str(value))
|
|||
|
|
log.info(f"文件路径: {file_path}")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
returnstr = execute_sql(content)
|
|||
|
|
log.info(f"执行结果: {returnstr}")
|
|||
|
|
log.info("-" * 50)
|
|||
|
|
except Exception as e:
|
|||
|
|
log.error(f"执行 SQL 文件 {file_path} 时出现错误: {e}")
|
|||
|
|
|
|||
|
|
except UnicodeDecodeError:
|
|||
|
|
log.error(f"文件 {file_path} 编码格式可能不是utf-8，无法正确读取，请检查。")
|
|||
|
|
except Exception as e:
|
|||
|
|
log.error(f"读取文件 {file_path} 时出现错误: {e}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
read_sql_files()
|