62 lines
3 KiB
Python
62 lines
3 KiB
Python
import ujson as json
|
||
import aiohttp
|
||
import ujson as json
|
||
|
||
from config import Config
|
||
from core.elements import FetchTarget, Image
|
||
from core.loader.decorator import command
|
||
from core.logger import Logger
|
||
from core.scheduler import Scheduler
|
||
from core.utils import get_url, download_to_cache
|
||
from database import BotDBUtil
|
||
|
||
|
||
@command('minecraft_news', autorun=True)
|
||
async def start_check_news(bot: FetchTarget):
|
||
baseurl = 'https://www.minecraft.net'
|
||
url = 'https://www.minecraft.net/content/minecraft-net/_jcr_content.articles.grid?tileselection=auto&tagsPath=minecraft:article/news,minecraft:article/insider,minecraft:article/culture,minecraft:article/merch,minecraft:stockholm/news,minecraft:stockholm/guides,minecraft:stockholm/deep-dives,minecraft:stockholm/merch,minecraft:stockholm/events,minecraft:stockholm/minecraft-builds,minecraft:stockholm/marketplace&offset=0&pageSize=10'
|
||
webrender = Config('infobox_render')
|
||
get = webrender + 'source?url=' + url
|
||
if not webrender:
|
||
return
|
||
getpage = await get_url(get)
|
||
title_list = []
|
||
if getpage:
|
||
o_nws = json.loads(getpage)['article_grid']
|
||
for o_article in o_nws:
|
||
o_default_tile = o_article['default_tile']['title']
|
||
title_list.append(o_default_tile)
|
||
Logger.info(str(title_list))
|
||
|
||
@Scheduler.scheduled_job('interval', seconds=600)
|
||
async def check_news():
|
||
user_list = []
|
||
get_all_enabled_user = BotDBUtil.Module.get_enabled_this('minecraft_news')
|
||
for x in get_all_enabled_user:
|
||
fetch = await bot.fetch_target(x)
|
||
if fetch:
|
||
user_list.append(fetch)
|
||
if not user_list:
|
||
return
|
||
Logger.info('Checking Minecraft news...' + str(title_list))
|
||
async with aiohttp.ClientSession() as session:
|
||
async with session.get(get) as resp:
|
||
status = resp.status
|
||
if status == 200:
|
||
nws = json.loads(await resp.read())['article_grid']
|
||
for article in nws:
|
||
default_tile = article['default_tile']
|
||
title = default_tile['title']
|
||
image = baseurl + default_tile['image']['imageURL']
|
||
desc = default_tile['sub_header']
|
||
link = baseurl + article['article_url']
|
||
if title not in title_list:
|
||
title_list.append(title)
|
||
articletext = f'Minecraft官网发布了新的文章：\n{title}\n{link}\n{desc}'
|
||
image = await download_to_cache(webrender + 'source?url=' + baseurl + image)
|
||
await bot.post_message('minecraft_news', articletext, user_list=user_list)
|
||
if image:
|
||
await bot.post_message('minecraft_news', [Image(image)], user_list=user_list)
|
||
Logger.info('Minecraft news checked.')
|
||
else:
|
||
Logger.info('Check minecraft news failed:' + str(status))
|