diff options
author | Kyomotoi <[email protected]> | 2022-11-04 10:12:58 +0800 |
---|---|---|
committer | Kyomotoi <[email protected]> | 2022-11-04 10:12:58 +0800 |
commit | 8f9611a989c54b6c24ba4f503f15ed6f71101552 (patch) | |
tree | bf8e73810229d91d5b16d88839196e2e9615833e | |
parent | 33bdf84881275bb4d0e2fa26b70a7343c36251a4 (diff) | |
download | ATRI-8f9611a989c54b6c24ba4f503f15ed6f71101552.tar.gz ATRI-8f9611a989c54b6c24ba4f503f15ed6f71101552.tar.bz2 ATRI-8f9611a989c54b6c24ba4f503f15ed6f71101552.zip |
🎨 完善权限更新
-rw-r--r-- | ATRI/permission.py | 11 |
1 files 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 |