LittlePaimon/hoshino/modules/Genshin_Paimon/daily_note/__init__.py

import json,os,re
from hoshino import R,MessageSegment,logger, Service
from hoshino.typing import CQEvent, Message
from hoshino.util import filt_message
from ..util import get_uid_by_qq, get_cookie, check_uid_by_qq, update_last_query_to_qq
from ..get_data import get_daily_note_data
from .get_img import draw_daily_note_card

sv = Service('原神实时便签')

@sv.on_prefix(('ssbq','实时便笺','实时便签'))
async def main(bot,ev):
    uid = ev.message.extract_plain_text().strip()
    qq = str(ev.user_id)
    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 uid and not check_uid_by_qq(qq, uid):
        await bot.send(ev,'派蒙没有这个uid的绑定信息哦',at_sender=True)
        return
    if not uid:
        uid = get_uid_by_qq(qq)
        if not uid:
            await bot.send(ev,'你还没把信息绑定给派蒙哦',at_sender=True)
            return
    if len(uid) != 9 or not uid.isdigit():
        await bot.send(ev,f'uid {filt_message(uid)} 不合规,是不是打错了呀',at_sender=True)
        return
    cookie = await get_cookie(qq, uid, only_private = True, only_match_uid = True)
    update_last_query_to_qq(qq, uid)
    if not cookie:
        await bot.send(ev,'你没有绑定cookie或者cookie失效了噢！',at_sender=True)
    else:
        try:
            data = await get_daily_note_data(uid, cookie)
            daily_note_card = await draw_daily_note_card(data, uid)
            await bot.send(ev, daily_note_card, at_sender=True)
        except Exception as e:
            await bot.send(ev, f'派蒙出现了问题：{e}',at_sender=True)