From 8f9611a989c54b6c24ba4f503f15ed6f71101552 Mon Sep 17 00:00:00 2001 From: Kyomotoi <0w0@imki.moe> Date: Fri, 4 Nov 2022 10:12:58 +0800 Subject: =?UTF-8?q?=F0=9F=8E=A8=20=E5=AE=8C=E5=96=84=E6=9D=83=E9=99=90?= =?UTF-8?q?=E6=9B=B4=E6=96=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/permission.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/ATRI/permission.py b/ATRI/permission.py index 7121322..97eb6b6 100644 --- a/ATRI/permission.py +++ b/ATRI/permission.py @@ -17,15 +17,19 @@ if not MASTER_FILE_PATH.is_file(): MASTER_LIST = set() -def is_master(bot: Bot, event: Event) -> bool: +def init_permission(): global MASTER_LIST + data = json.loads(MASTER_FILE_PATH.read_bytes()) + MASTER_LIST = set.union(set(data), conf.BotConfig.superusers) + + +def is_master(bot: Bot, event: Event) -> bool: + init_permission() try: user_id = event.get_user_id() except Exception: return False - data = json.loads(MASTER_FILE_PATH.read_bytes()) - MASTER_LIST = set.union(set(data), conf.BotConfig.superusers) return True if user_id in MASTER_LIST else False @@ -65,5 +69,6 @@ class Admin: return is_master(bot, event) +init_permission() MASTER = Permission(MasterList()).set_name("Master") ADMIN = Permission(Admin()).set_name("Admin") \ No newline at end of file -- cgit v1.2.3