nonebot2/none/message.py

32 lines
901 B
Python
Raw Normal View History

2018-06-15 06:58:24 +08:00
from typing import Dict, Any
from aiocqhttp import CQHttp
from aiocqhttp.message import MessageSegment
2018-06-27 22:05:12 +08:00
from .command import handle_command
2018-06-27 22:50:01 +08:00
from .log import logger
2018-06-30 10:58:56 +08:00
from .helpers import send
2018-06-15 06:58:24 +08:00
2018-06-27 16:36:40 +08:00
async def handle_message(bot: CQHttp, ctx: Dict[str, Any]) -> None:
2018-06-15 06:58:24 +08:00
if ctx['message_type'] != 'private':
# group or discuss
2018-06-30 10:58:56 +08:00
ctx['to_me'] = False
first_message_seg = ctx['message'][0]
if first_message_seg == MessageSegment.at(ctx['self_id']):
ctx['to_me'] = True
del ctx['message'][0]
2018-06-15 06:58:24 +08:00
if not ctx['message']:
ctx['message'].append(MessageSegment.text(''))
2018-06-30 10:58:56 +08:00
else:
ctx['to_me'] = True
2018-06-15 06:58:24 +08:00
2018-06-27 16:36:40 +08:00
handled = await handle_command(bot, ctx)
2018-06-15 06:58:24 +08:00
if handled:
logger.debug('Message is handled as a command')
2018-06-27 16:36:40 +08:00
return
2018-06-30 10:58:56 +08:00
elif ctx['to_me']:
await send(bot, ctx, '你在说什么我看不懂诶')
# TODO: NLP