ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2024-12-12 14:35:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Support multiple nicknames

Browse Source
This commit is contained in:
Richard Chien 2018-07-04 09:39:50 +08:00
parent 030a31e63f
commit b00e31ad84
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
none/natural_language.py
View File

@ -80,8 +80,12 @@ async def handle_natural_language(bot: NoneBot, ctx: Dict[str, Any]) -> bool:
""" """
msg = str(ctx['message']) msg = str(ctx['message'])
if bot.config.NICKNAME: if bot.config.NICKNAME:
# check if the user is calling to me with my nickname # check if the user is calling me with my nickname
m = re.search(rf'^{bot.config.NICKNAME}[\s,，]+', msg) if not isinstance(bot.config.NICKNAME, Iterable):
nicknames = (bot.config.NICKNAME,)
else:
nicknames = filter(lambda n: n, bot.config.NICKNAME)
m = re.search(rf'^({"|".join(nicknames)})[\s,，]+', msg)
if m: if m:
ctx['to_me'] = True ctx['to_me'] = True
msg = msg[m.end():] msg = msg[m.end():]

2
none_demo/config.py
View File

@ -6,6 +6,6 @@ HOST = '0.0.0.0'
SECRET = 'abc' SECRET = 'abc'
SUPERUSERS = {1002647525} SUPERUSERS = {1002647525}
NICKNAME = '奶茶' NICKNAME = {'奶茶', '小奶茶'}
COMMAND_START = {'', '/', '!', '／', '！', re.compile(r'^>+\s*')} COMMAND_START = {'', '/', '!', '／', '！', re.compile(r'^>+\s*')}
COMMAND_SEP = {'/', '.', re.compile(r'#|::?')} COMMAND_SEP = {'/', '.', re.compile(r'#|::?')}