Change the environment variable name ADMIN to ADMIN_LIST

yym68686 · Dec 7, 2023 · 6a385fd · 6a385fd
1 parent a01263e
commit 6a385fd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/config.py b/config.py
@@ -36,7 +36,7 @@
     claudeBot = claudebot(api_key=f"{ClaudeAPI}")
 
 whitelist = os.environ.get('whitelist', None)
-ADMIN = os.environ.get('ADMIN', None)
+ADMIN_LIST = os.environ.get('ADMIN_LIST', None)
 if whitelist:
     whitelist = [int(id) for id in whitelist.split(",")]
 

diff --git a/utils/decorators.py b/utils/decorators.py
@@ -19,9 +19,9 @@ async def wrapper(*args, **kwargs):
 # 判断是否是管理员
 def AdminAuthorization(func):
     async def wrapper(*args, **kwargs):
-        if config.ADMIN == None:
+        if config.ADMIN_LIST == None:
             return await func(*args, **kwargs)
-        if (args[0].effective_chat.id not in config.ADMIN):
+        if (args[0].effective_chat.id not in config.ADMIN_LIST):
             message = (
                 f"`Hi, {args[0].effective_user.username}!`\n\n"
                 f"id: `{args[0].effective_user.id}`\n\n"