mirror of https://github.com/Mai-with-u/MaiBot.git
90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
"""
|
||
TOML文件工具函数 - 保留格式和注释
|
||
"""
|
||
|
||
import os
|
||
import tomlkit
|
||
from typing import Any
|
||
|
||
|
||
def save_toml_with_format(data: dict[str, Any], file_path: str) -> None:
|
||
"""
|
||
保存TOML数据到文件，保留现有格式（如果文件存在）
|
||
|
||
Args:
|
||
data: 要保存的数据字典
|
||
file_path: 文件路径
|
||
"""
|
||
# 如果文件不存在，直接创建
|
||
if not os.path.exists(file_path):
|
||
with open(file_path, "w", encoding="utf-8") as f:
|
||
tomlkit.dump(data, f)
|
||
return
|
||
|
||
# 如果文件存在，尝试读取现有文件以保留格式
|
||
try:
|
||
with open(file_path, "r", encoding="utf-8") as f:
|
||
existing_doc = tomlkit.load(f)
|
||
except Exception:
|
||
# 如果读取失败，直接覆盖
|
||
with open(file_path, "w", encoding="utf-8") as f:
|
||
tomlkit.dump(data, f)
|
||
return
|
||
|
||
# 递归更新，保留现有格式
|
||
_merge_toml_preserving_format(existing_doc, data)
|
||
|
||
# 保存
|
||
with open(file_path, "w", encoding="utf-8") as f:
|
||
tomlkit.dump(existing_doc, f)
|
||
|
||
|
||
def _merge_toml_preserving_format(target: dict[str, Any], source: dict[str, Any]) -> None:
|
||
"""
|
||
递归合并source到target，保留target中的格式和注释
|
||
|
||
Args:
|
||
target: 目标文档（保留格式）
|
||
source: 源数据（新数据）
|
||
"""
|
||
for key, value in source.items():
|
||
if key in target:
|
||
# 如果两个都是字典且都是表格，递归合并
|
||
if isinstance(value, dict) and isinstance(target[key], dict):
|
||
if hasattr(target[key], "items"): # 确实是字典/表格
|
||
_merge_toml_preserving_format(target[key], value)
|
||
else:
|
||
target[key] = value
|
||
else:
|
||
# 其他情况直接替换
|
||
target[key] = value
|
||
else:
|
||
# 新键直接添加
|
||
target[key] = value
|
||
|
||
|
||
def _update_toml_doc(target: dict[str, Any], source: dict[str, Any]) -> None:
|
||
"""
|
||
更新TOML文档中的字段，保留现有的格式和注释
|
||
|
||
这是一个递归函数，用于在部分更新配置时保留现有的格式和注释。
|
||
|
||
Args:
|
||
target: 目标表格（会被修改）
|
||
source: 源数据（新数据）
|
||
"""
|
||
for key, value in source.items():
|
||
if key in target:
|
||
# 如果两个都是字典，递归更新
|
||
if isinstance(value, dict) and isinstance(target[key], dict):
|
||
if hasattr(target[key], "items"): # 确实是表格
|
||
_update_toml_doc(target[key], value)
|
||
else:
|
||
target[key] = value
|
||
else:
|
||
# 直接更新值，保留注释
|
||
target[key] = value
|
||
else:
|
||
# 新键直接添加
|
||
target[key] = value
|