47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
import tomli
|
||
import tomli_w
|
||
import sys
|
||
from pathlib import Path
|
||
import os
|
||
|
||
def sync_configs():
|
||
# 读取两个配置文件
|
||
try:
|
||
with open('bot_config_dev.toml', 'rb') as f: # tomli需要使用二进制模式读取
|
||
dev_config = tomli.load(f)
|
||
|
||
with open('bot_config.toml', 'rb') as f:
|
||
prod_config = tomli.load(f)
|
||
except FileNotFoundError as e:
|
||
print(f"错误：找不到配置文件 - {e}")
|
||
sys.exit(1)
|
||
except tomli.TOMLDecodeError as e:
|
||
print(f"错误：TOML格式解析失败 - {e}")
|
||
sys.exit(1)
|
||
|
||
# 递归合并配置
|
||
def merge_configs(source, target):
|
||
for key, value in source.items():
|
||
if key not in target:
|
||
target[key] = value
|
||
elif isinstance(value, dict) and isinstance(target[key], dict):
|
||
merge_configs(value, target[key])
|
||
|
||
# 将dev配置的新属性合并到prod配置中
|
||
merge_configs(dev_config, prod_config)
|
||
|
||
# 保存更新后的配置
|
||
try:
|
||
with open('bot_config.toml', 'wb') as f: # tomli_w需要使用二进制模式写入
|
||
tomli_w.dump(prod_config, f)
|
||
print("配置文件同步完成！")
|
||
except Exception as e:
|
||
print(f"错误：保存配置文件失败 - {e}")
|
||
sys.exit(1)
|
||
|
||
if __name__ == '__main__':
|
||
# 确保在正确的目录下运行
|
||
script_dir = Path(__file__).parent
|
||
os.chdir(script_dir)
|
||
sync_configs()
|