mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 03:32:33 +08:00
feat: 支持json格式的配置文件 (#265)
This commit is contained in:
parent
d67dbe8062
commit
cfc97fb22d
3
.gitignore
vendored
3
.gitignore
vendored
|
@ -13,4 +13,5 @@ sensitive.json
|
||||||
temp/
|
temp/
|
||||||
current_tag
|
current_tag
|
||||||
scenario/
|
scenario/
|
||||||
!scenario/default-template.json
|
!scenario/default-template.json
|
||||||
|
override.json
|
14
main.py
14
main.py
|
@ -1,4 +1,5 @@
|
||||||
import importlib
|
import importlib
|
||||||
|
import json
|
||||||
import os
|
import os
|
||||||
import shutil
|
import shutil
|
||||||
import threading
|
import threading
|
||||||
|
@ -143,9 +144,22 @@ def main(first_time_init=False):
|
||||||
setattr(config, key, getattr(config_template, key))
|
setattr(config, key, getattr(config_template, key))
|
||||||
logging.warning("[{}]不存在".format(key))
|
logging.warning("[{}]不存在".format(key))
|
||||||
is_integrity = False
|
is_integrity = False
|
||||||
|
|
||||||
if not is_integrity:
|
if not is_integrity:
|
||||||
logging.warning("配置文件不完整,请依据config-template.py检查config.py")
|
logging.warning("配置文件不完整,请依据config-template.py检查config.py")
|
||||||
logging.warning("以上配置已被设为默认值,将在5秒后继续启动... ")
|
logging.warning("以上配置已被设为默认值,将在5秒后继续启动... ")
|
||||||
|
|
||||||
|
# 检查override.json覆盖
|
||||||
|
if os.path.exists("override.json"):
|
||||||
|
override_json = json.load(open("override.json", "r", encoding="utf-8"))
|
||||||
|
for key in override_json:
|
||||||
|
if hasattr(config, key):
|
||||||
|
setattr(config, key, override_json[key])
|
||||||
|
logging.info("覆写配置[{}]为[{}]".format(key, override_json[key]))
|
||||||
|
else:
|
||||||
|
logging.error("无法覆写配置[{}]为[{}],该配置不存在,请检查override.json是否正确".format(key, override_json[key]))
|
||||||
|
|
||||||
|
if not is_integrity:
|
||||||
time.sleep(5)
|
time.sleep(5)
|
||||||
|
|
||||||
import pkg.utils.context
|
import pkg.utils.context
|
||||||
|
|
Loading…
Reference in New Issue
Block a user