akari-bot/core/bots/graia/bot.py

import asyncio

from graia.application.event.mirai import NewFriendRequestEvent, BotInvitedJoinGroupRequestEvent
from graia.application.friend import Friend
from graia.application.group import Group, Member
from graia.application.message.chain import MessageChain
from graia.application import GraiaMiraiApplication

from config import Config
from core.bots.graia.template import Template as BotTemplate
from core.bots.graia.broadcast import bcc, app
from core.elements import MsgInfo, Session, Module
from core.loader import Modules
from core.parser.message import parser


@bcc.receiver('GroupMessage')
async def group_message_handler(message: MessageChain, group: Group, member: Member):
    msg = BotTemplate(target=MsgInfo(targetId=f"QQ|Group|{group.id}", senderId=f'QQ|{member.id}', senderName=member.name,
                         targetFrom='QQ|Group', senderFrom="QQ"), session=Session(message=message, target=group, sender=member))
    await parser(msg)


@bcc.receiver('FriendMessage')
async def group_message_handler(message: MessageChain, friend: Friend):
    msg = BotTemplate(target=MsgInfo(targetId=f"QQ|{friend.id}", senderId=f'QQ|{friend.id}', senderName=friend.nickname,
                         targetFrom='QQ', senderFrom='QQ'), session=Session(message=message, target=friend, sender=friend))
    await parser(msg)


@bcc.receiver("NewFriendRequestEvent")
async def NFriend(event: NewFriendRequestEvent):
    await event.accept()


@bcc.receiver("BotInvitedJoinGroupRequestEvent")
async def NGroup(event: BotInvitedJoinGroupRequestEvent):
    await event.accept()


@bcc.receiver('ApplicationLaunched')
async def autorun_handler(app: GraiaMiraiApplication):
    gather_list = []
    for x in Modules:
        if isinstance(x, Module) and Modules[x].autorun:
            gather_list.append(asyncio.ensure_future(Modules[x].function(app)))
    await asyncio.gather(*gather_list)


if Config('qq_host') and Config('qq_account'):
    app.launch_blocking()