34 lines
926 B
Python
34 lines
926 B
Python
import os
|
|
|
|
import yaml
|
|
|
|
# 模块级单例
|
|
_config = None
|
|
_engine = None
|
|
|
|
|
|
def load_config():
|
|
global _config
|
|
if _config is not None:
|
|
return _config
|
|
|
|
config_path = 'config.yaml'
|
|
template_path = 'config_template.yaml'
|
|
|
|
if not os.path.exists(config_path):
|
|
if os.path.exists(template_path):
|
|
# 如果配置文件不存在但模板存在，则复制模板
|
|
import shutil
|
|
shutil.copy(template_path, config_path)
|
|
print(f"已从 {template_path} 创建新的配置文件 {config_path}，请完善其中的配置信息后再运行。")
|
|
exit(1)
|
|
else:
|
|
print(f"错误：配置文件 {config_path} 和模板 {template_path} 均不存在。")
|
|
print(f"请创建 {template_path} 文件后再运行。")
|
|
exit(1)
|
|
|
|
with open(config_path, 'r') as f:
|
|
_config = yaml.safe_load(f)
|
|
return _config
|
|
|