mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-12-16 13:40:53 +08:00
48 lines
2.0 KiB
Python
48 lines
2.0 KiB
Python
import re, datetime
|
||
from aiocqhttp.exceptions import ActionFailed
|
||
from json import JSONDecodeError
|
||
from hoshino import logger, Service
|
||
from hoshino.util import filt_message
|
||
from ..util import get_uid_in_msg
|
||
from ..get_data import get_monthinfo_data
|
||
from .get_img import draw_monthinfo_card
|
||
|
||
help_msg='''
|
||
[myzj/每月札记/zj (uid) (月份)]查看该月份获得的原石、摩拉数
|
||
*绑定私人cookie之后才能使用,只能查看最近3个月的记录,默认为本月
|
||
'''
|
||
sv = Service('派蒙每月札记', bundle='派蒙', help_=help_msg)
|
||
|
||
@sv.on_prefix(('myzj', '每月札记', 'zj'))
|
||
async def main(bot,ev):
|
||
uid, msg, user_id, use_cache = await get_uid_in_msg(ev)
|
||
# 札记只能查看最近3个月的，构造正则来获取月份
|
||
month_now = datetime.datetime.now().month
|
||
if month_now == 1:
|
||
month_list = ['11','12','1']
|
||
elif month_now == 2:
|
||
month_list = ['12', '1', '2']
|
||
else:
|
||
month_list = [str(month_now - 2), str(month_now - 1)]
|
||
find_month = '(?P<month>' + '|'.join(month_list) + ')'
|
||
match = re.search(find_month, msg)
|
||
month = match.group('month') if match else month_now
|
||
try:
|
||
data = await get_monthinfo_data(uid, month, use_cache=use_cache)
|
||
if isinstance(data, str):
|
||
await bot.send(ev, data, at_sender=True)
|
||
else:
|
||
monthinfo_card = await draw_monthinfo_card(data)
|
||
await bot.send(ev, monthinfo_card, at_sender=True)
|
||
except ActionFailed:
|
||
logger.exception('账号可能被风控')
|
||
await bot.send(ev, '派蒙可能被风控，发不出消息')
|
||
except TypeError or AttributeError:
|
||
await bot.send(ev, '派蒙好像没有该UID的绑定信息')
|
||
except IndexError or KeyError:
|
||
await bot.send(ev, '派蒙获取信息失败，可能是米游社API有变动，请联系开发者')
|
||
except JSONDecodeError:
|
||
await bot.send(ev, '派蒙获取信息失败，重试一下吧')
|
||
except Exception as e:
|
||
await bot.send(ev, f'派蒙出现了问题：{e}')
|