akari-bot/modules/mcv_rss/__init__.py

import asyncio
import re
import traceback
from os.path import abspath

import aiohttp
from graia.application import MessageChain
from graia.application.message.elements.internal import Plain

from core.loader import logger_info
from database import BotDB
from modules.mcv.mcv import get_data


check_enable_modules_all = BotDB().check_enable_modules_all


def mcversion():
    w = open(abspath('./assets/mcversion.txt'), 'r+')
    s = w.read().split('\n')
    w.close()
    return s


def mcversion_jira():
    w = open(abspath('./assets/mcversion_jira.txt'), 'r+')
    s = w.read().split('\n')
    w.close()
    return s


async def mcv_rss(app):
    url = 'http://launchermeta.mojang.com/mc/game/version_manifest.json'
    logger_info('Subbot ver launched')
    while True:
        try:
            logger_info('Checking mcv...')
            verlist = mcversion()
            file = await get_data(url, 'json')
            release = file['latest']['release']
            snapshot = file['latest']['snapshot']
            if release not in verlist:
                logger_info(f'huh, we find {release}.')
                for qqgroup in check_enable_modules_all('group_permission', 'mcv_rss'):
                    try:
                        await app.sendGroupMessage(int(qqgroup), MessageChain.create(
                            [Plain('启动器已更新' + file['latest']['release'] + '正式版。')]))
                        await asyncio.sleep(0.5)
                    except Exception:
                        traceback.print_exc()
                for qqfriend in check_enable_modules_all('friend_permission', 'mcv_rss_self'):
                    try:
                        await app.sendFriendMessage(int(qqfriend), MessageChain.create(
                            [Plain('启动器已更新' + file['latest']['release'] + '正式版。')]))
                        await asyncio.sleep(0.5)
                    except Exception:
                        traceback.print_exc()
                addversion = open('./assets/mcversion.txt', 'a')
                addversion.write('\n' + release)
                addversion.close()
                verlist = mcversion()
                release_ = re.sub('\.', '-', release)
                article_link = 'https://www.minecraft.net/en-us/article/minecraft-java-edition-' + release_
                async with aiohttp.ClientSession() as getart:
                    async with getart.get(article_link) as getstatus:
                        if getstatus.status == 200:
                            for qqgroup in check_enable_modules_all('group_permission', 'mcv_rss'):
                                try:
                                    await app.sendGroupMessage(int(qqgroup), MessageChain.create(
                                        [Plain(f'Mojang已发布{release}的更新日志：{article_link}')]))
                                    await asyncio.sleep(0.5)
                                except Exception:
                                    traceback.print_exc()
                            for qqfriend in check_enable_modules_all('friend_permission', 'mcv_rss_self'):
                                try:
                                    await app.sendFriendMessage(int(qqfriend), MessageChain.create(
                                        [Plain(f'Mojang已发布{release}的更新日志：{article_link}')]))
                                    await asyncio.sleep(0.5)
                                except Exception:
                                    traceback.print_exc()
            if snapshot not in verlist:
                logger_info(f'huh, we find {snapshot}.')
                for qqgroup in check_enable_modules_all('group_permission', 'mcv_rss'):
                    try:
                        await app.sendGroupMessage(int(qqgroup), MessageChain.create(
                            [Plain('启动器已更新' + file['latest']['snapshot'] + '快照。')]))
                        await asyncio.sleep(0.5)
                    except Exception:
                        traceback.print_exc()
                for qqfriend in check_enable_modules_all('friend_permission', 'mcv_rss_self'):
                    try:
                        await app.sendFriendMessage(int(qqfriend), MessageChain.create(
                            [Plain('启动器已更新' + file['latest']['snapshot'] + '快照。')]))
                        await asyncio.sleep(0.5)
                    except Exception:
                        traceback.print_exc()
                addversion = open('./assets/mcversion.txt', 'a')
                addversion.write('\n' + snapshot)
                addversion.close()
                article_link = 'https://www.minecraft.net/en-us/article/minecraft-snapshot-' + snapshot
                async with aiohttp.ClientSession() as getart:
                    async with getart.get(article_link) as getstatus:
                        if getstatus.status == 200:
                            for qqgroup in check_enable_modules_all('group_permission', 'mcv_rss'):
                                try:
                                    await app.sendGroupMessage(int(qqgroup), MessageChain.create(
                                        [Plain(f'Mojang已发布{snapshot}的更新日志：{article_link}')]))
                                    await asyncio.sleep(0.5)
                                except Exception:
                                    traceback.print_exc()
                            for qqfriend in check_enable_modules_all('friend_permission', 'mcv_rss_self'):
                                try:
                                    await app.sendFriendMessage(int(qqfriend), MessageChain.create(
                                        [Plain(f'Mojang已发布{snapshot}的更新日志：{article_link}')]))
                                    await asyncio.sleep(0.5)
                                except Exception:
                                    traceback.print_exc()
            logger_info('mcv checked.')
            await asyncio.sleep(40)
        except Exception:
            traceback.print_exc()
            await asyncio.sleep(20)


async def mcv_jira_rss(app):
    url = 'https://bugs.mojang.com/rest/api/2/project/10400/versions'
    logger_info('Subbot jira launched')
    while True:
        try:
            logger_info('Checking Jira mcv...')
            verlist = mcversion_jira()
            file = await get_data(url, 'json')
            release = []
            for v in file:
                if not v['archived']:
                    release.append(v['name'])
            for x in release:
                if x not in verlist:
                    logger_info(f'huh, we find {x}.')
                    for qqgroup in check_enable_modules_all('group_permission', 'mcv_jira_rss'):
                        try:
                            await app.sendGroupMessage(int(qqgroup), MessageChain.create(
                                [Plain(f'Jira已更新{x}。\n（Jira上的信息仅作版本号预览用，不代表启动器已更新此版本）')]))
                            await asyncio.sleep(0.5)
                        except Exception:
                            traceback.print_exc()
                    for qqfriend in check_enable_modules_all('friend_permission', 'mcv_jira_rss_self'):
                        try:
                            await app.sendFriendMessage(int(qqfriend), MessageChain.create(
                                [Plain(f'Jira已更新{x}。\n（Jira上的信息仅作版本号预览用，不代表启动器已更新此版本）')]))
                            await asyncio.sleep(0.5)
                        except Exception:
                            traceback.print_exc()
                    addversion = open('./assets/mcversion_jira.txt', 'a')
                    addversion.write('\n' + x)
                    addversion.close()
            logger_info('jira mcv checked.')
            await asyncio.sleep(40)
        except Exception:
            traceback.print_exc()
            await asyncio.sleep(20)


rss = {'mcv_rss': mcv_rss, 'mcv_jira_rss': mcv_jira_rss}
options = ['mcv_rss', 'mcv_jira_rss']
friend_options = ['mcv_rss_self', 'mcv_jira_rss_self']
help = {'mcv_rss': {'help': '订阅Minecraft Java版游戏版本检测。（仅群聊）'},
        'mcv_jira_rss': {'help': '订阅Minecraft Java版游戏版本检测（Jira记录，仅作预览用）。（仅群聊）'}}