-
Notifications
You must be signed in to change notification settings - Fork 2
/
fgo_news.py
120 lines (105 loc) · 5.43 KB
/
fgo_news.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
import re
import shutil
from aiocqhttp import ActionFailed
from hoshino import HoshinoBot
from .get.get_news import get_news
from .path_and_json import *
@sv_news.on_fullmatch(("帮助fgo新闻获取", "帮助FGO新闻获取", "帮助bgo新闻获取", "帮助BGO新闻获取"))
@sv_news.on_rex(re.compile(r"^[fb]go[新x][闻w][获h][取q][帮b][助z]$", re.IGNORECASE))
async def bangzhu(bot: HoshinoBot, ev: CQEvent):
helps = gen_node(sv_news_help.strip())
await bot.send_group_forward_msg(group_id=ev['group_id'], messages=helps)
@sv_news.on_rex(re.compile(r"^[获h更g][取q新x][fb]go[新x][闻w](\s\d+)?$", re.IGNORECASE))
async def get_official_news(bot: HoshinoBot, ev: CQEvent):
async with ClientSession(headers=headers) as session:
num = ev.message.extract_plain_text().split()
if len(num) > 1:
num = int(num[1])
else:
num = 6
news, status = await get_news(num, session)
if isinstance(status, Exception):
await bot.send(ev, f"获取新闻出错，原因：\n{status}")
return
if status:
await bot.send(ev, f"没有新的新闻~本地共有{news}条新闻~")
else:
await bot.send(ev, f"下载完成，本次共获取了{news}条新闻~")
@sv_news.on_rex(re.compile(r"^[查c][询x][fb]go[新x][闻w](\s.+)?$", re.IGNORECASE))
async def get_local_news(bot: HoshinoBot, ev: CQEvent):
if not os.path.exists(news_detail_path):
await bot.send(ev, "没有本地新闻~请先获取官网新闻~")
return
args = ev.message.extract_plain_text()
pic = True if "pic" in args else False
get_all = True if "all" in args or "全部" in args else False
match = re.search(r"\d+", args)
index = int(match.group(0)) if match else 0
if not index:
get_all = True
try:
news = json.load(open(news_detail_path, encoding="utf-8"))
except json.decoder.JSONDecodeError:
await bot.send(ev, "没有本地新闻~请先获取官网新闻~")
return
if not os.path.exists(news_img_path):
sv_news.logger.info("初始化新闻图片目录...")
os.mkdir(news_img_path)
news_num = len(news)
if not get_all:
index = int(index) - 1
link = f"标题：{news[index]['title']}\n电脑版网页：{news[index]['page']}\n手机版网页：{news[index]['mobile_page']}\n\n"
if pic:
img_path = os.path.join(news_img_path, f"{news[index]['id']}.png")
resp = await getpic(news[index]['page'], img_path)
if resp:
msg = f"{gen_ms_img(Image.open(img_path))}\n"
else:
sv_news.logger.error("获取新闻截图出错")
msg = "截图出错，请自行查看~"
_news = gen_node((link + msg).strip())
try:
await bot.send_group_forward_msg(group_id=ev['group_id'], messages=_news)
except ActionFailed:
await bot.send(ev, "转发消息失败……尝试直接发送~")
try:
await bot.send(ev, msg)
except ActionFailed:
await bot.send(ev, f"转发消息失败……可能是新闻太长了，试试直接去官网看看吧~\n"
f"标题：{news[index]['title']}\n"
f"电脑版网页：{news[index]['page']}\n"
f"手机版网页：{news[index]['mobile_page']}")
else:
msg = create_img(news[index]['content'].strip())
_news = [gen_node(link.strip()), gen_node(msg)]
await bot.send_group_forward_msg(group_id=ev['group_id'], messages=_news)
if get_all:
if not index:
await bot.send(ev, f"本地共有{news_num}条新闻，请用编号查对应新闻~")
news_all = []
for i in range(news_num):
link = f"标题：{news[i]['title']}\n电脑版网页：{news[i]['page']}\n手机版网页：{news[i]['mobile_page']}\n\n"
_news = gen_node(link.strip())
news_all.append(_news)
try:
await bot.send_group_forward_msg(group_id=ev['group_id'], messages=news_all)
except ActionFailed:
if news_num < 10:
await bot.send(ev, f"发送合集失败，尝试拆分发送！\n共有{news_num}条新闻~")
for i in range(news_num):
link = f"标题：{news[i]['title']}\n电脑版网页：{news[i]['page']}\n手机版网页：{news[i]['mobile_page']}\n\n"
_news = gen_node(link.strip())
try:
await bot.send_group_forward_msg(group_id=ev['group_id'], messages=_news)
except ActionFailed:
await bot.send(ev, f"转发消息失败……可能是新闻太长了，试试直接去官网看看吧~\n"
f"标题：{news[i]['title']}\n"
f"电脑版网页：{news[i]['page']}\n"
f"手机版网页：{news[i]['mobile_page']}")
else:
await bot.send(ev, f"新闻太多啦！\n共有{news_num}条新闻，请尝试用编号查对应新闻~")
@sv_news.on_fullmatch("清除新闻缓存")
async def delete_news_cache(bot: HoshinoBot, ev: CQEvent):
shutil.rmtree(news_img_path)
await bot.send(ev, "已清理新闻缓存")
return