100 lines
3.8 KiB
Python
100 lines
3.8 KiB
Python
import eventlet
|
|
from graia.application import MessageChain, GroupMessage, FriendMessage
|
|
from graia.application.friend import Friend
|
|
from graia.application.group import Group, Member
|
|
from graia.application.message.elements.internal import Plain, Image, Source
|
|
from graia.broadcast.interrupt import InterruptControl
|
|
from graia.broadcast.interrupt.waiter import Waiter
|
|
|
|
from core.broadcast import app, bcc
|
|
from database import check_superuser
|
|
|
|
|
|
async def sendMessage(kwargs: dict, msgchain):
|
|
if isinstance(msgchain, str):
|
|
msgchain = MessageChain.create([Plain(msgchain)])
|
|
if Group in kwargs:
|
|
try:
|
|
eventlet.monkey_patch()
|
|
with eventlet.Timeout(15):
|
|
send = await app.sendGroupMessage(kwargs[Group], msgchain, quote=kwargs[MessageChain][Source][0].id)
|
|
return send
|
|
except eventlet.timeout.Timeout:
|
|
split_msg = msgchain.get(Plain)
|
|
sent_msgs = []
|
|
for msgs in split_msg:
|
|
send = await app.sendGroupMessage(kwargs[Group], MessageChain.create([msgs]),
|
|
quote=kwargs[MessageChain][Source][0].id)
|
|
sent_msgs.append(send)
|
|
split_img = msgchain.get(Image)
|
|
for imgs in split_img:
|
|
send = await app.sendGroupMessage(kwargs[Group], MessageChain.create([imgs]),
|
|
quote=kwargs[MessageChain][Source][0].id)
|
|
sent_msgs.append(send)
|
|
return sent_msgs
|
|
if Friend in kwargs:
|
|
try:
|
|
eventlet.monkey_patch()
|
|
with eventlet.Timeout(15):
|
|
send = await app.sendFriendMessage(kwargs[Friend], msgchain)
|
|
return send
|
|
except eventlet.timeout.Timeout:
|
|
split_msg = msgchain.get(Plain)
|
|
sent_msgs = []
|
|
for msgs in split_msg:
|
|
send = await app.sendFriendMessage(kwargs[Friend], MessageChain.create([msgs]))
|
|
sent_msgs.append(send)
|
|
split_img = msgchain.get(Image)
|
|
for imgs in split_img:
|
|
send = await app.sendFriendMessage(kwargs[Friend], MessageChain.create([imgs]))
|
|
sent_msgs.append(send)
|
|
return sent_msgs
|
|
|
|
|
|
async def wait_confirm(kwargs: dict):
|
|
inc = InterruptControl(bcc)
|
|
confirm_command = ["是", "对", 'yes', 'y']
|
|
if Group in kwargs:
|
|
@Waiter.create_using_function([GroupMessage])
|
|
def waiter(waiter_group: Group,
|
|
waiter_member: Member, waiter_message: MessageChain):
|
|
if all([
|
|
waiter_group.id == kwargs[Group].id,
|
|
waiter_member.id == kwargs[Member].id,
|
|
]):
|
|
print(111)
|
|
if waiter_message.asDisplay() in confirm_command:
|
|
return True
|
|
else:
|
|
return False
|
|
if Friend in kwargs:
|
|
@Waiter.create_using_function([FriendMessage])
|
|
def waiter(waiter_friend: Friend, waiter_message: MessageChain):
|
|
if all([
|
|
waiter_friend.id == kwargs[Friend].id,
|
|
]):
|
|
if waiter_message.asDisplay() in confirm_command:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
return await inc.wait(waiter)
|
|
|
|
|
|
async def revokeMessage(msgchain):
|
|
if isinstance(msgchain, list):
|
|
for msg in msgchain:
|
|
await app.revokeMessage(msg)
|
|
else:
|
|
await app.revokeMessage(msgchain)
|
|
|
|
|
|
def check_permission(kwargs):
|
|
if Group in kwargs:
|
|
if str(kwargs[Member].permission) in ['MemberPerm.Administrator', 'MemberPerm.Owner'] or check_superuser(
|
|
kwargs):
|
|
return True
|
|
if Friend in kwargs:
|
|
if check_superuser(kwargs[Friend].id):
|
|
return True
|
|
return False
|