Archived
1
0
Fork 0
This repository has been archived on 2024-04-26. You can view files and clone it, but cannot push or open issues or pull requests.
akari-bot/modules/minecraft_news/__init__.py

69 lines
3.3 KiB
Python
Raw Normal View History

2021-08-24 15:58:57 +00:00
import asyncio
import ujson as json
2021-08-24 15:58:57 +00:00
import traceback
import aiohttp
import random
from config import Config
from database import BotDBUtil
from core.utils import get_url, download_to_cache
from core.scheduler import Scheduler
from core.loader.decorator import command
from core.elements import FetchTarget, Plain, Image
from core.logger import Logger
@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)
2021-08-25 15:15:20 +00:00
Logger.info(str(title_list))
2021-08-24 15:58:57 +00:00
@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
2021-08-25 15:15:20 +00:00
Logger.info('Checking Minecraft news...' + str(title_list))
2021-08-24 15:58:57 +00:00
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)
2021-08-26 16:53:46 +00:00
articletext = f'Minecraft官网发布了新的文章\n{title}\n{link}\n{desc}'
2021-08-24 15:58:57 +00:00
image = await download_to_cache(webrender + 'source?url=' + baseurl + image)
for x in user_list:
2021-08-25 15:15:20 +00:00
try:
2021-08-25 17:09:17 +00:00
await x.sendMessage(articletext)
2021-08-25 15:15:20 +00:00
if image:
await x.sendMessage([Image(image)])
2021-08-26 16:53:46 +00:00
await asyncio.sleep(random.randint(1, 10))
2021-08-25 15:15:20 +00:00
except Exception:
traceback.print_exc()
2021-08-24 15:58:57 +00:00
Logger.info('Minecraft news checked.')
else:
Logger.info('Check minecraft news failed:' + str(status))