mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 19:37:36 +08:00
47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
import os
|
|
import shutil
|
|
import json
|
|
|
|
from .. import model as file_model
|
|
|
|
|
|
class JSONConfigFile(file_model.ConfigFile):
|
|
"""JSON配置文件"""
|
|
|
|
config_file_name: str = None
|
|
"""配置文件名"""
|
|
|
|
template_file_name: str = None
|
|
"""模板文件名"""
|
|
|
|
def __init__(self, config_file_name: str, template_file_name: str) -> None:
|
|
self.config_file_name = config_file_name
|
|
self.template_file_name = template_file_name
|
|
|
|
def exists(self) -> bool:
|
|
return os.path.exists(self.config_file_name)
|
|
|
|
async def create(self):
|
|
shutil.copyfile(self.template_file_name, self.config_file_name)
|
|
|
|
async def load(self) -> dict:
|
|
|
|
if not self.exists():
|
|
await self.create()
|
|
|
|
with open(self.config_file_name, 'r', encoding='utf-8') as f:
|
|
cfg = json.load(f)
|
|
|
|
# 从模板文件中进行补全
|
|
with open(self.template_file_name, 'r', encoding='utf-8') as f:
|
|
template_cfg = json.load(f)
|
|
|
|
for key in template_cfg:
|
|
if key not in cfg:
|
|
cfg[key] = template_cfg[key]
|
|
|
|
return cfg
|
|
|
|
async def save(self, cfg: dict):
|
|
with open(self.config_file_name, 'w', encoding='utf-8') as f:
|
|
json.dump(cfg, f, indent=4, ensure_ascii=False) |