ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2024-11-28 05:59:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update natural_language.py

Browse Source
This commit is contained in:
Richard Chien 2019-07-01 09:59:49 +08:00 committed by GitHub
parent 60101e7634
commit c2de849ea7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nonebot/natural_language.py
View File

@ -27,7 +27,7 @@ class NLProcessor:
self.allow_empty_message = allow_empty_message self.allow_empty_message = allow_empty_message
def on_natural_language(keywords: Union[Optional[Iterable], Callable] = None, def on_natural_language(keywords: Union[Optional[Iterable], str, Callable] = None,
*, permission: int = perm.EVERYBODY, *, permission: int = perm.EVERYBODY,
only_to_me: bool = True, only_to_me: bool = True,
only_short_message: bool = True, only_short_message: bool = True,
@ -42,9 +42,6 @@ def on_natural_language(keywords: Union[Optional[Iterable], Callable] = None,
:param allow_empty_message: handle empty messages :param allow_empty_message: handle empty messages
""" """
if isinstance(keywords, str):
keywords = (keywords,)
def deco(func: Callable) -> Callable: def deco(func: Callable) -> Callable:
nl_processor = NLProcessor(func=func, keywords=keywords, nl_processor = NLProcessor(func=func, keywords=keywords,
permission=permission, permission=permission,
@ -58,6 +55,8 @@ def on_natural_language(keywords: Union[Optional[Iterable], Callable] = None,
# here "keywords" is the function to be decorated # here "keywords" is the function to be decorated
return on_natural_language()(keywords) return on_natural_language()(keywords)
else: else:
if isinstance(keywords, str):
keywords = (keywords,)
return deco return deco