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

bots: Switch Incrementor to use state_handler.state() contextmanager.

Browse source
This commit is contained in:
neiljp (Neil Pilgrim) 2017-07-23 11:47:54 -07:00 committed by Tim Abbott
parent 101dd60829
commit 7a8b41b63e
1 changed files with 10 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
zulip_bots/zulip_bots/bots/incrementor/incrementor.py
View file

@ -12,18 +12,16 @@ class IncrementorHandler(object):
''' '''
def handle_message(self, message, bot_handler, state_handler): def handle_message(self, message, bot_handler, state_handler):
state = state_handler.get_state() or {'number': 0, 'message_id': None} with state_handler.state({'number': 0, 'message_id': None}) as state:
state['number'] += 1 state['number'] += 1
state_handler.set_state(state) if state['message_id'] is None:
if state['message_id'] is None: result = bot_handler.send_reply(message, str(state['number']))
result = bot_handler.send_reply(message, str(state['number'])) state['message_id'] = result['id']
state['message_id'] = result['id'] else:
state_handler.set_state(state) bot_handler.update_message(dict(
else: message_id = state['message_id'],
bot_handler.update_message(dict( content = str(state['number'])
message_id = state['message_id'], ))
content = str(state['number'])
))
handler_class = IncrementorHandler handler_class = IncrementorHandler