update:格式化代码

This commit is contained in:
hrz
2025-02-14 00:54:59 +08:00
parent eae590eb26
commit c5180df9ac
11 changed files with 34 additions and 34 deletions
+7 -2
View File
@@ -1,10 +1,15 @@
import os
import argparse
from core.utils.util import read_config
from core.utils.util import read_config, get_project_dir
def load_config():
"""加载配置文件"""
parser = argparse.ArgumentParser(description="Server configuration")
parser.add_argument("--config_path", type=str, default="config.yaml")
default_config_file = "config.yaml"
# 判断是否存在私有的配置文件
if os.path.exists(get_project_dir() + "." + default_config_file):
default_config_file = "." + default_config_file
parser.add_argument("--config_path", type=str, default=default_config_file)
args = parser.parse_args()
return read_config(args.config_path)