nonebot2/demo/plugins/tuling.py
2018-12-27 20:23:45 +08:00

38 lines
999 B
Python

from nonebot import (
on_command, CommandSession,
on_natural_language, NLPSession, NLPResult
)
@on_command('tuling', aliases=('聊天', '对话'))
async def tuling(session: CommandSession):
message = session.get('message', prompt='我已经准备好啦,来跟我聊天吧~')
finish = message in ('结束', '拜拜', '再见')
if finish:
session.finish('拜拜啦,你忙吧,下次想聊天随时找我哦~')
return
# call tuling api
reply = f'你说了:{message}'
one_time = session.get_optional('one_time', False)
if one_time:
session.finish(reply)
else:
session.pause(reply)
@tuling.args_parser
async def _(session: CommandSession):
if session.current_key == 'message':
session.args[session.current_key] = session.current_arg_text.strip()
@on_natural_language
async def _(session: NLPSession):
return NLPResult(60.0, 'tuling', {
'message': session.msg,
'one_time': True
})