akari-bot/bot.py

import asyncio

from graia.application import GraiaMiraiApplication
from graia.application.friend import Friend
from graia.application.group import Group, Member
from graia.application.message.chain import MessageChain

from core.broadcast import bcc, app
from core.loader import rss_loader
from core.parser import parser
import os

cache_path = os.path.abspath('./cache/')
if os.path.exists(cache_path):
    for x in os.listdir(cache_path):
        os.remove(cache_path + x)
    os.removedirs(cache_path)
    os.mkdir(cache_path)
else:
    os.mkdir(cache_path)

@bcc.receiver('GroupMessage')
async def group_message_handler(message: MessageChain, group: Group, member: Member):
    kwargs = {MessageChain: message, Group: group, Member: member}
    await parser(kwargs)


@bcc.receiver('FriendMessage')
async def group_message_handler(message: MessageChain, friend: Friend):
    kwargs = {MessageChain: message, Friend: friend}
    await parser(kwargs)


@bcc.receiver('ApplicationLaunched')
async def message_handler(app: GraiaMiraiApplication):
    rss_list = rss_loader()
    gather_list = []
    for x in rss_list:
        gather_list.append(asyncio.ensure_future(rss_list[x](app)))
    await asyncio.gather(*gather_list)


app.launch_blocking()