nonebot/nonebot2
1
1
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2024-09-20 21:02:36 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Make superuser qq as a environment var

Browse Source
This commit is contained in:
Richard Chien 2016-12-07 21:43:00 +08:00
parent addcf40c13
commit 7c5d7630ea
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
command.py
View File

@ -1,5 +1,6 @@
import functools
import re
import os
from apiclient import client as api
from little_shit import SkipException, get_command_name_separators
@ -155,7 +156,7 @@ class CommandRegistry:
try:
if func.superuser_only:
check(str(ctx_msg.get('sender_qq')) == '1002647525')
check(str(ctx_msg.get('sender_qq')) == os.environ.get('SUPER_USER_QQ'))
if ctx_msg.get('type') == 'group_message':
allowed_roles = {'owner', 'admin', 'member'}
if func.group_admin_only:
@ -178,7 +179,7 @@ class CommandRegistry:
# This is strange, not likely happens
raise SkipException
except SkipException:
if not str(ctx_msg.get('sender_qq')) == '1002647525':
if not str(ctx_msg.get('sender_qq')) == os.environ.get('SUPER_USER_QQ'):
# Not allowed
return False