34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
import re
|
||
|
||
from database import BotDBUtil
|
||
from core.loader.decorator import command
|
||
from core.elements import MessageSession, Image
|
||
|
||
from .profile import cytoid_profile
|
||
from .rating import get_rating
|
||
|
||
|
||
@command('cytoid', help_doc=('~cytoid (b30|r30) <UserID> {查询一个用户的b30/r30记录}',
|
||
'~cytoid profile <UserID> {查询一个用户的基本信息}'))
|
||
async def cytoid(msg: MessageSession):
|
||
if msg.parsed_msg['profile']:
|
||
await cytoid_profile(msg)
|
||
if msg.parsed_msg['b30']:
|
||
query = 'b30'
|
||
elif msg.parsed_msg['r30']:
|
||
query = 'r30'
|
||
else:
|
||
query = False
|
||
if query:
|
||
qc = BotDBUtil.CoolDown(msg, 'cytoid_rank')
|
||
c = qc.check(300)
|
||
if c == 0:
|
||
img = await get_rating(msg.parsed_msg['<UserID>'], query)
|
||
if 'path' in img:
|
||
await msg.sendMessage([Image(path=img['path'])])
|
||
if 'text' in img:
|
||
await msg.sendMessage(img['text'])
|
||
if img['status']:
|
||
qc.reset()
|
||
else:
|
||
await msg.sendMessage(f'距离上次执行已过去{int(c)}秒，本命令的冷却时间为300秒。')
|