Update modules/config.py

modules/config.py

@@ -73,15 +73,21 @@ if os.environ.get("XMCHAT_API_KEY", None) == None:
     os.environ["XMCHAT_API_KEY"] = xmchat_api_key
 
 ## 多账户机制
-multi_api_key = config.get("multi_api_key", False) # 是否开启多账户机制
-if multi_api_key:
-    api_key_list = config.get("api_key_list", [])
+keylist_var = os.environ.get('key_list')
+if bool(keylist_var):
+    multi_api_key = bool(keylist_var)
+    api_key_list = keylist_var.split(',')
     if len(api_key_list) == 0:
         logging.error("多账号模式已开启，但api_key_list为空，请检查config.json")
         sys.exit(1)
     shared.state.set_api_key_queue(api_key_list)
 
 auth_list = config.get("users", []) # 实际上是使用者的列表
+users_var = os.environ.get('users')
+if users_var:
+    users_pairs = users_var.split(',')
+    users = [pair.split(':') for pair in users_pairs]
+    auth_list = users
 authflag = len(auth_list) > 0  # 是否开启认证的状态值，改为判断auth_list长度
 
 # 处理自定义的api_host，优先读环境变量的配置，如果存在则自动装配