akari-bot/modules/cytoid/__init__.py

import re
from .profile import cytoid_profile
from .rating import get_rating
from core.template import sendMessage
from graia.application.message.elements.internal import Image
from graia.application import MessageChain
from database import write_time, check_time


async def cytoid(kwargs: dict):
    command = kwargs['trigger_msg']
    command = re.sub('cytoid ', '', command)
    command_split = command.split(' ')
    if command_split[0] == 'profile':
        kwargs['trigger_msg'] = re.sub(r'^profile ', '', command)
        await cytoid_profile(kwargs)
    if command_split[0] in ['b30', 'r30']:
        c = check_time(kwargs, 'cytoidrank', 300)
        if not c:
            write_time(kwargs, 'cytoidrank')
            uid = re.sub(r'^.*?30 ', '', command)
            img = await get_rating(uid, command_split[0])
            if 'path' in img:
                await sendMessage(kwargs, MessageChain.create([Image.fromLocalFile(img['path'])]))
            if 'text' in img:
                await sendMessage(kwargs, img['text'])
        else:
            await sendMessage(kwargs, f'距离上次执行已过去{-c}秒，本命令的冷却时间为300秒。')
    if command_split[0] == 'test':
        c = check_time(kwargs, 'test')
        if c:
            write_time(kwargs, 'test')
            await sendMessage(kwargs, '111')
        else:
            await sendMessage(kwargs, 'delay!')



command = {'cytoid': cytoid}
help = {'cytoid': {'help': {'~cytoid profile <uid> - 获取一个用户的Cytoid账号信息。\n' + \
                            '~cytoid b30 <uid> - 获取一个用户的Best30信息。\n' + \
                            '~cytoid r30 <uid> - 获取一个用户的Recent30信息。'}}}