feat: 现在管理员是必需配置项

This commit is contained in:
Rock Chin 2023-01-05 18:08:18 +08:00
parent 5dfd0a9b50
commit d2922afce2
2 changed files with 22 additions and 2 deletions

View File

@ -32,7 +32,7 @@ openai_config = {
},
}
# 管理员QQ号用于接收报错等通知及执行管理员级别指令为0时关闭此功能
# [必需] 管理员QQ号用于接收报错等通知及执行管理员级别指令
admin_qq = 0
# 情景预设(机器人人格)

22
main.py
View File

@ -73,7 +73,27 @@ def main(first_time_init=False):
# 检查是否设置了管理员
if not (hasattr(config, 'admin_qq') and config.admin_qq != 0):
logging.warning("未设置管理员QQ,管理员权限指令及运行告警将无法使用,如需设置请修改config.py中的admin_qq字段")
# logging.warning("未设置管理员QQ,管理员权限指令及运行告警将无法使用,如需设置请修改config.py中的admin_qq字段")
while True:
try:
config.admin_qq = int(input("未设置管理员QQ,管理员权限指令及运行告警将无法使用,请输入管理员QQ号: "))
# 写入到文件
# 读取文件
config_file_str = ""
with open("config.py", "r", encoding="utf-8") as f:
config_file_str = f.read()
# 替换
config_file_str = config_file_str.replace("admin_qq = 0", "admin_qq = " + str(config.admin_qq))
# 写入
with open("config.py", "w", encoding="utf-8") as f:
f.write(config_file_str)
print("管理员QQ已设置如需修改请修改config.py中的admin_qq字段")
time.sleep(4)
break
except ValueError:
print("请输入数字")
import pkg.openai.manager
import pkg.database.manager