xenofem/python-zulip-api
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

bot lib: Add optional StateHandler contextmanager.

Browse source 
State can still be handled with get_state/set_state, and flushed
before the end of a 'with' block using set_state.
This commit is contained in:
neiljp (Neil Pilgrim) 2017-07-23 11:26:40 -07:00 committed by Tim Abbott
parent d9ac5ad88c
commit 101dd60829
1 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
zulip_bots/zulip_bots/lib.py
View file

@ -9,6 +9,8 @@ import re
from six.moves import configparser from six.moves import configparser
from contextlib import contextmanager
if False: if False:
from mypy_extensions import NoReturn from mypy_extensions import NoReturn
from typing import Any, Optional, List, Dict from typing import Any, Optional, List, Dict
@ -107,15 +109,22 @@ class ExternalBotHandler(object):
class StateHandler(object): class StateHandler(object):
def __init__(self): def __init__(self):
# type: () -> None # type: () -> None
self.state = None # type: Any self.state_ = None # type: Any
def set_state(self, state): def set_state(self, state):
# type: (Any) -> None # type: (Any) -> None
self.state = state self.state_ = state
def get_state(self): def get_state(self):
# type: () -> Any # type: () -> Any
return self.state return self.state_
@contextmanager
def state(self, default):
#type: (Any) -> Any
new_state = self.get_state() or default
yield new_state
self.set_state(new_state)
def run_message_handler_for_bot(lib_module, quiet, config_file): def run_message_handler_for_bot(lib_module, quiet, config_file):
# type: (Any, bool, str) -> Any # type: (Any, bool, str) -> Any