mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-10-21 16:27:15 +08:00
83 lines
3.6 KiB
Python
83 lines
3.6 KiB
Python
import re
|
||
from aiohttp import ClientSession
|
||
from nonebot import on_command, require, get_bot, logger
|
||
from nonebot.params import CommandArg
|
||
from nonebot.adapters.onebot.v11 import MessageEvent, MessageSegment
|
||
from ..utils.util import load_data, save_data, get_id
|
||
|
||
news60s_pic = on_command('早报', aliases={'今日早报', '今日新闻', '60s读世界'}, priority=13, block=True)
|
||
|
||
scheduler = require('nonebot_plugin_apscheduler').scheduler
|
||
|
||
|
||
@news60s_pic.handle()
|
||
async def news60s_pic_handler(event: MessageEvent, msg=CommandArg()):
|
||
msg = str(msg).strip()
|
||
if not msg:
|
||
url = 'https://api.iyk0.com/60s/'
|
||
async with ClientSession() as session:
|
||
res = await session.get(url)
|
||
res = await res.json()
|
||
await news60s_pic.finish(MessageSegment.image(file=res['imageUrl']))
|
||
elif msg.startswith('开启推送'):
|
||
# 匹配msg中的xx:xx时间
|
||
time_str = re.search(r'(\d{2}):(\d{2})', msg)
|
||
if time_str:
|
||
push_data = load_data('news60s_push.json')
|
||
push_id = str(get_id(event))
|
||
push_data[push_id] = {
|
||
'type': event.message_type,
|
||
'hour': int(time_str.group(1)),
|
||
'minute': int(time_str.group(2))
|
||
}
|
||
if event.message_type == 'guild':
|
||
push_data[push_id]['guild_id'] = event.guild_id
|
||
scheduler.add_job(
|
||
func=news60s_push_task,
|
||
trigger='cron',
|
||
hour=int(time_str.group(1)),
|
||
minute=int(time_str.group(2)),
|
||
id=str(get_id(event)),
|
||
args=(str(get_id(event)),
|
||
push_data[str(get_id(event))])
|
||
)
|
||
save_data(push_data, 'news60s_push.json')
|
||
await news60s_pic.finish('开启60s读世界推送成功', at_sender=True)
|
||
else:
|
||
await news60s_pic.finish('请给出正确的时间，格式为12:00', at_sender=True)
|
||
elif msg.startswith('关闭推送'):
|
||
push_data = load_data('news60s_push.json')
|
||
del push_data[str(event.group_id or event.channel_id or event.user_id)]
|
||
scheduler.remove_job(str(get_id(event)))
|
||
save_data(push_data, 'news60s_push.json')
|
||
await news60s_pic.finish('关闭60s读世界推送成功', at_sender=True)
|
||
|
||
|
||
async def news60s_push_task(push_id, push_data: dict):
|
||
try:
|
||
url = 'https://api.iyk0.com/60s/'
|
||
async with ClientSession() as session:
|
||
res = await session.get(url)
|
||
res = await res.json()
|
||
if push_data['type'] == 'group':
|
||
await get_bot().send_group_msg(group_id=push_id, message=MessageSegment.image(file=res['imageUrl']))
|
||
elif push_data['type'] == 'private':
|
||
await get_bot().send_private_msg(user_id=push_id, message=MessageSegment.image(file=res['imageUrl']))
|
||
elif push_data['type'] == 'guild':
|
||
await get_bot().send_guild_channel_msg(guild_id=push_data['guild_id'], channel_id=push_id,
|
||
message=MessageSegment.image(file=res['imageUrl']))
|
||
logger.info(f'{push_data["type"]}的{push_id}的60秒读世界推送成功')
|
||
except Exception as e:
|
||
logger.exception(f'{push_data["type"]}的{push_id}的60秒读世界推送失败：{e}')
|
||
|
||
|
||
for push_id, push_data in load_data('news60s_push.json').items():
|
||
scheduler.add_job(
|
||
func=news60s_push_task,
|
||
trigger='cron',
|
||
hour=push_data['hour'],
|
||
minute=push_data['minute'],
|
||
id=push_id,
|
||
args=(push_id,
|
||
push_data)
|
||
) |