mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-12-16 13:40:53 +08:00
57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
import json,os,re
|
||
from hoshino import R,MessageSegment,logger, Service
|
||
from hoshino.typing import CQEvent, Message
|
||
from ..util import get_uid_by_qq, get_cookie, update_last_query_to_qq
|
||
from ..get_data import get_abyss_data
|
||
from .get_img import draw_abyss_card
|
||
|
||
sv = Service('原神深渊查询')
|
||
|
||
@sv.on_prefix(('sy','深渊查询','深境螺旋查询'))
|
||
async def main(bot,ev):
|
||
msg = ev.message.extract_plain_text().strip().split(' ')
|
||
uid = ''
|
||
if len(msg[0]) == 9 and msg[0].isdigit():
|
||
uid = msg[0]
|
||
del msg[0]
|
||
if not msg:
|
||
abyss_floor = []
|
||
else:
|
||
abyss_floor = msg
|
||
abyss_floor_true = []
|
||
for floor in abyss_floor:
|
||
if floor.isdigit() and (int(floor) <= 12 and int(floor) >= 9):
|
||
abyss_floor_true.append(int(floor))
|
||
abyss_floor_true.sort()
|
||
if len(abyss_floor_true)>2:
|
||
abyss_floor_true = [abyss_floor_true[0],abyss_floor_true[1]]
|
||
qq = str(ev.user_id)
|
||
# nickname = ev['sender']['nickname']
|
||
if ev.message_type == 'guild':
|
||
rm = str(ev.message)
|
||
else:
|
||
rm = str(ev.raw_message)
|
||
match = re.search(r"\[CQ:at,qq=(.*)\]", rm)
|
||
if match:
|
||
uid = ''
|
||
qq = str(match.group(1))
|
||
if not uid:
|
||
uid = get_uid_by_qq(qq)
|
||
if not uid:
|
||
await bot.send(ev,'请把uid给派蒙哦，比如sy100000001',at_sender=True)
|
||
return
|
||
cookie = await get_cookie(qq, uid)
|
||
update_last_query_to_qq(qq, uid)
|
||
if not cookie:
|
||
await bot.send(ev,'当前cookie池中没有可用的cookie，请联系开发者',at_sender=True)
|
||
else:
|
||
try:
|
||
data = await get_abyss_data(uid, cookie)
|
||
abyss_card = await draw_abyss_card(data, uid, abyss_floor_true)
|
||
await bot.send(ev,abyss_card,at_sender=True)
|
||
except Exception as e:
|
||
await bot.send(ev, f'派蒙出现了问题：{e}',at_sender=True)
|
||
|
||
|
||
|