From f65d15c08e7cb5f3297f8fb1c1417c205732bd19 Mon Sep 17 00:00:00 2001 From: Yoav Caspi Date: Sat, 7 Sep 2019 22:40:22 +0300 Subject: [PATCH] add regex flags to handle_message function. --- bot.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/bot.py b/bot.py index e32b413..2477c91 100644 --- a/bot.py +++ b/bot.py @@ -101,22 +101,22 @@ HANDLERS: List[Tuple[Pattern[str], Callable[[Match[str]], Response]]] 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: 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 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( *( f'^:(?P[^!]+).* ' f'PRIVMSG #(?P[^ ]+) ' f':(?P{message_prefix}.*)' for message_prefix in message_prefixes - ), + ), flags=flags, )