154 lines
6.4 KiB
Python
154 lines
6.4 KiB
Python
import asyncio
|
|
import os
|
|
import random
|
|
import re
|
|
import traceback
|
|
from os.path import abspath
|
|
|
|
import graia.application.interrupt as inter
|
|
from graia.application.message.chain import MessageChain
|
|
from graia.application.message.elements.internal import Plain, Image, Source
|
|
|
|
from modules.findimage import findimage
|
|
|
|
try:
|
|
cachepath = abspath('./assets/cache/')
|
|
cachefile = os.listdir(cachepath)
|
|
for file in cachefile:
|
|
os.remove(f'{cachepath}/{file}')
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
async def gen(bcc, app, message, target1, target2='0', msgtype='None', runfun='command'):
|
|
im = inter.InterruptControl(bcc)
|
|
command = __import__('CommandGen', fromlist=[runfun])
|
|
command = getattr(command, runfun)
|
|
if msgtype == 'Group':
|
|
run = await command(message.asDisplay(), target1.id)
|
|
else:
|
|
run = await command(message.asDisplay())
|
|
# print(run)
|
|
if run != None:
|
|
await msgproc(run, app, im, command, message, target1, target2, msgtype)
|
|
|
|
|
|
async def msgproc(resultmsgraw, app, im, command, message, target1, target2='0', msgtype='None'):
|
|
print(resultmsgraw)
|
|
msgchain = await makemsgchain(resultmsgraw)
|
|
send = await sendmessage(app, msgchain, target1, target2, msgtype,
|
|
message[Source][0] if msgtype == 'Group' else 0)
|
|
uimgcs = re.findall(r'\[\[uimgc:.*\]\]', resultmsgraw, re.I | re.M)
|
|
for uimgc in uimgcs:
|
|
uimgc = re.match(r'\[\[uimgc:(.*)\]\]', uimgc)
|
|
if uimgc:
|
|
await uimgsend(app, message, target1, target2, msgtype, uimgc.group(1))
|
|
r = re.findall(r'(https?://.*?/File:.*?\.(?:png|gif|jpg|jpeg|webp|bmp|ico))', resultmsgraw, re.I)
|
|
for d in r:
|
|
d1 = await findimage(d)
|
|
if d1 is not None:
|
|
await linkimgsend(app, d1, target1, target2, msgtype)
|
|
await afterproc(resultmsgraw, app, im, command, send, message, target1, target2, msgtype)
|
|
|
|
|
|
async def makemsgchain(msg):
|
|
msg = re.sub('\[wait\]', '', msg)
|
|
msgbase = re.sub(r'\[\[uimgc:.*\]\]', '', msg)
|
|
msgchain = MessageChain.create([Plain(msgbase)])
|
|
return msgchain
|
|
|
|
|
|
async def afterproc(resultmsgraw, app, im, command, send, message, target1, target2='0', msgtype='None'):
|
|
if resultmsgraw.find('[一分钟后撤回本消息]') != -1:
|
|
await asyncio.sleep(60)
|
|
await app.revokeMessage(send)
|
|
if resultmsgraw.find('[30秒后撤回本消息]') != -1:
|
|
await asyncio.sleep(30)
|
|
await app.revokeMessage(send)
|
|
if resultmsgraw.find('[wait]') != -1:
|
|
ranint = random.randint(1, 3)
|
|
if ranint == 2:
|
|
waitmsg = await makemsgchain('提示:你可以发送“是”字来将所有无效结果再次查询。(考虑到实现复杂性,恕不提供选择性查询)')
|
|
await sendmessage(app, waitmsg, target1, target2, msgtype)
|
|
MessageEventImport = __import__('graia.application', fromlist=[f'{msgtype}Message'])
|
|
MessageEvent = getattr(MessageEventImport, f'{msgtype}Message')
|
|
InterruptImport = __import__('graia.application.interrupt.interrupts',
|
|
fromlist=[f'{msgtype}MessageInterrupt'])
|
|
Interrupt = getattr(InterruptImport, f'{msgtype}MessageInterrupt')
|
|
if msgtype == 'Friend':
|
|
event: MessageEvent = await im.wait(Interrupt(target1.id))
|
|
else:
|
|
event: MessageEvent = await im.wait(Interrupt(target1, target2))
|
|
print(event)
|
|
if event.messageChain.asDisplay() == '是':
|
|
msg2 = await command(resultmsgraw)
|
|
await msgproc(msg2, app, im, command, message, target1, target2, msgtype)
|
|
else:
|
|
pass
|
|
|
|
|
|
async def uimgsend(app, message, target1, target2, msgtype, link):
|
|
exec('from graia.application.message.elements.internal import UploadMethods')
|
|
mth = eval(f'UploadMethods.{msgtype}')
|
|
try:
|
|
msgchain = MessageChain.create(
|
|
[Image.fromLocalFile(filepath=abspath(link), method=mth)])
|
|
print('Sending Image...')
|
|
await sendmessage(app, msgchain, target1, target2, msgtype,
|
|
message[Source][0] if msgtype == 'Group' else 0)
|
|
except Exception:
|
|
traceback.print_exc()
|
|
msgchain = MessageChain.create(
|
|
[Plain('上传过程中遇到了问题,图片发送失败。')])
|
|
await sendmessage(app, msgchain, target1, target2, msgtype,
|
|
message[Source][0] if msgtype == 'Group' else 0)
|
|
|
|
|
|
async def linkimgsend(app, sendlink, target1, target2, msgtype):
|
|
exec('from graia.application.message.elements.internal import UploadMethods')
|
|
mth = eval(f'UploadMethods.{msgtype}')
|
|
try:
|
|
msgchain = MessageChain.create([Image.fromNetworkAddress(url=sendlink, method=mth)])
|
|
print('Sending Image...')
|
|
await sendmessage(app, msgchain, target1, target2, msgtype)
|
|
except Exception:
|
|
traceback.print_exc()
|
|
msgchain = MessageChain.create(
|
|
[Plain('上传过程中遇到了问题,图片发送失败。')])
|
|
await sendmessage(app, msgchain, target1, target2, msgtype)
|
|
|
|
|
|
async def sendmessage(app, msgchain, target1, target2, msgtype, quoteid=0):
|
|
if msgtype == 'Friend':
|
|
friend = target1
|
|
send = await app.sendFriendMessage(friend, msgchain.asSendable())
|
|
if msgtype == 'Group':
|
|
group = target1
|
|
send = await app.sendGroupMessage(group, msgchain.asSendable(), quote=quoteid if quoteid != 0 else None)
|
|
if msgtype == 'Temp':
|
|
group = target1
|
|
member = target2
|
|
send = await app.sendTempMessage(group=group, target=member, message=msgchain.asSendable())
|
|
return send
|
|
|
|
|
|
"""
|
|
if msgtype == 'Group':
|
|
voice = re.findall(r'https?://.*?/File:.*?\.(?:ogg|m4a|mp3|flac|wav)', run, re.I)
|
|
for voicelink in voice:
|
|
try:
|
|
findvoicename = re.match(r'(https?://.*?/)File:(.*?\.(?:ogg|m4a|mp3|flac|wav))', voicelink, re.I)
|
|
downloadfile = await dfile(findvoicename.group(1), findvoicename.group(2))
|
|
print(downloadfile)
|
|
conventamr = await camr(downloadfile)
|
|
print(conventamr)
|
|
readfile = open(conventamr, 'rb+')
|
|
uploadvoice = await app.uploadVoice(readfile.read())
|
|
voicemsgchain = MessageChain.create([uploadvoice])
|
|
await app.sendGroupMessage(target1, voicemsgchain)
|
|
readfile.close()
|
|
os.remove(downloadfile)
|
|
os.remove(conventamr)
|
|
except Exception:
|
|
traceback.print_exc()
|
|
"""
|