41 lines
1.3 KiB
41 lines
1.3 KiB
from nonebot import require, export
|
|
from nonebot.permission import Permission, SUPERUSER
|
|
from nonebot.adapters.cqhttp import Bot, Event, MessageEvent
|
|
|
|
MySQLdbMethods = require("botdb").MySQLdbMethods
|
|
mysql = MySQLdbMethods()
|
|
export = export()
|
|
|
|
|
|
async def _teacher(bot: Bot, event: Event):
|
|
"""分析是否为教师或超级用户，如果是则满足条件"""
|
|
if event.get_type() == "message":
|
|
event: MessageEvent
|
|
user_id = event.get_user_id()
|
|
# 是否为超级用户
|
|
if user_id in bot.config.superusers:
|
|
return True
|
|
else:
|
|
# 是否为教师
|
|
await mysql.execute("select qq from teacher where qq=%s" % event.user_id)
|
|
return event.user_id in set(mysql.form()["qq"])
|
|
return False
|
|
|
|
|
|
async def _class_group(bot: Bot, event: Event):
|
|
if event.get_type() == "message":
|
|
event: MessageEvent
|
|
user_id = event.get_user_id()
|
|
# 是否为超级用户
|
|
if user_id in bot.config.superusers:
|
|
return True
|
|
else:
|
|
await mysql.execute("select qq from user_info union select qq from teacher")
|
|
return event.user_id in set(mysql.form()["qq"])
|
|
return False
|
|
|
|
CLASS_GROUP = Permission(_class_group)
|
|
TEACHER = Permission(_teacher)
|
|
export.CLASS_GROUP = CLASS_GROUP
|
|
export.TEACHER = TEACHER
|