chunzhi/chatgpt_academic
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

env read failure reminder

Browse Source
This commit is contained in:
binary-husky 2023-05-02 15:33:07 +08:00
parent 896077009a
commit a69ef7f8c5
1 changed files with 28 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
toolbox.py
View File

@ -536,10 +536,14 @@ def read_env_variable(arg, default_value):
""" """
from colorful import print亮红, print亮绿 from colorful import print亮红, print亮绿
arg_with_prefix = "GPT_ACADEMIC_" + arg arg_with_prefix = "GPT_ACADEMIC_" + arg
if arg_with_prefix in os.environ: env_arg = os.environ[arg_with_prefix] if arg_with_prefix in os.environ:
elif arg in os.environ: env_arg = os.environ[arg] env_arg = os.environ[arg_with_prefix]
elif arg in os.environ:
env_arg = os.environ[arg]
else: else:
raise KeyError raise KeyError
print(f"[ENV_VAR] 尝试加载{arg}，默认值：{default_value} --> 修正值：{env_arg}")
try:
if isinstance(default_value, bool): if isinstance(default_value, bool):
r = bool(env_arg) r = bool(env_arg)
elif isinstance(default_value, int): elif isinstance(default_value, int):
@ -556,8 +560,12 @@ def read_env_variable(arg, default_value):
assert arg == "proxies" assert arg == "proxies"
r = eval(env_arg) r = eval(env_arg)
else: else:
print亮红(f"[ENV_VAR] 环境变量{arg}不支持通过环境变量设置！{default_value}") print亮红(f"[ENV_VAR] 环境变量{arg}不支持通过环境变量设置! ")
raise KeyError raise KeyError
except:
print亮红(f"[ENV_VAR] 环境变量{arg}加载失败! ")
raise KeyError(f"[ENV_VAR] 环境变量{arg}加载失败! ")
print亮绿(f"[ENV_VAR] 成功读取环境变量{arg}") print亮绿(f"[ENV_VAR] 成功读取环境变量{arg}")
return r return r