diff --git a/config-template.py b/config-template.py index 4f68606d..af199168 100644 --- a/config-template.py +++ b/config-template.py @@ -32,7 +32,7 @@ openai_config = { }, } -# 管理员QQ号,用于接收报错等通知及执行管理员级别指令,为0时关闭此功能 +# [必需] 管理员QQ号,用于接收报错等通知及执行管理员级别指令 admin_qq = 0 # 情景预设(机器人人格) diff --git a/main.py b/main.py index 688adc7e..8633f574 100644 --- a/main.py +++ b/main.py @@ -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