askill
/
twitch-chat-bot
mirror of https://github.com/Askill/twitch-chat-bot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add regex flags to handle_message function.

This commit is contained in:
Yoav Caspi 2019-09-07 22:40:22 +03:00
parent f1558c034c
commit f65d15c08e
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
bot.py
View File

@ -101,22 +101,22 @@ HANDLERS: List[Tuple[Pattern[str], Callable[[Match[str]], Response]]]
HANDLERS = [] HANDLERS = []
def handler(*prefixes: str) -> Callable[[Callback], Callback]: def handler(*prefixes: str, flags: re.RegexFlag = re.U) -> Callable[[Callback], Callback]:
def handler_decorator(func: Callback) -> Callback: def handler_decorator(func: Callback) -> Callback:
for prefix in prefixes: for prefix in prefixes:
HANDLERS.append((re.compile(prefix + '\r\n$'), func)) HANDLERS.append((re.compile(prefix + '\r\n$', flags=flags), func))
return func return func
return handler_decorator return handler_decorator
def handle_message(*message_prefixes: str) -> Callable[[Callback], Callback]: def handle_message(*message_prefixes: str, flags: re.RegexFlag = re.U) -> Callable[[Callback], Callback]:
return handler( return handler(
*( *(
f'^:(?P<user>[^!]+).* ' f'^:(?P<user>[^!]+).* '
f'PRIVMSG #(?P<channel>[^ ]+) ' f'PRIVMSG #(?P<channel>[^ ]+) '
f':(?P<msg>{message_prefix}.*)' f':(?P<msg>{message_prefix}.*)'
for message_prefix in message_prefixes for message_prefix in message_prefixes
), ), flags=flags,
) )