Austin/QChatGPT
1
0
Fork 0
mirror of https://github.com/RockChinQ/QChatGPT.git synced 2024-11-16 03:32:33 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

暂时解决reload后的config无法加载问题

Browse Source
This commit is contained in:
LINSTCL 2023-03-23 16:42:15 +08:00
parent 196e424c88
commit f56c6876d1
1 changed files with 19 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
main.py
View File

@ -102,6 +102,24 @@ def reset_logging():
pkg.utils.context.context['logger_handler'] = sh
return sh
# 临时函数，用于加载config和上下文，未来统一放在config类
def load_config():
# 完整性校验
is_integrity = True
config_template = importlib.import_module('config-template')
config = importlib.import_module('config')
for key in dir(config_template):
if not key.startswith("__") and not hasattr(config, key):
setattr(config, key, getattr(config_template, key))
logging.warning("[{}]不存在".format(key))
is_integrity = False
if not is_integrity:
logging.warning("配置文件不完整，请依据config-template.py检查config.py")
logging.warning("以上配置已被设为默认值，将在5秒后继续启动... ")
time.sleep(5)
# 存进上下文
pkg.utils.context.set_config(config)
def start(first_time_init=False):
"""启动流程，reload之后会被执行"""
@ -109,6 +127,7 @@ def start(first_time_init=False):
global known_exception_caught
import pkg.utils.context
load_config() #暂时在start里加载config应对重载，未来在config抽象类中统一处理reload(未来删除该行)
config = pkg.utils.context.get_config()
# 更新openai库到最新版本
if not hasattr(config, 'upgrade_dependencies') or config.upgrade_dependencies:
@ -294,26 +313,6 @@ def stop():
raise e
# 临时函数，用于加载config和上下文，未来统一放在config类
def load_config():
# 完整性校验
is_integrity = True
config_template = importlib.import_module('config-template')
config = importlib.import_module('config')
for key in dir(config_template):
if not key.startswith("__") and not hasattr(config, key):
setattr(config, key, getattr(config_template, key))
logging.warning("[{}]不存在".format(key))
is_integrity = False
if not is_integrity:
logging.warning("配置文件不完整，请依据config-template.py检查config.py")
logging.warning("以上配置已被设为默认值，将在5秒后继续启动... ")
time.sleep(5)
# 存进上下文
pkg.utils.context.set_config(config)
def check_file():
# 配置文件存在性校验
if not os.path.exists('config.py'):