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

zulip-bots: Use context manager for incrementor.

Browse source
This commit is contained in:
PIG208 2021-05-15 21:03:39 +08:00 committed by Tim Abbott
parent 86fa9f5e35
commit b8389b78c1
1 changed files with 19 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1,7 +1,7 @@
# See readme.md for instructions on running this code. # See readme.md for instructions on running this code.
from typing import Dict from typing import Dict
from zulip_bots.lib import BotHandler from zulip_bots.lib import BotHandler, use_storage
class IncrementorHandler: class IncrementorHandler:
META = { META = {
@ -24,8 +24,8 @@ class IncrementorHandler:
storage.put('message_id', None) storage.put('message_id', None)
def handle_message(self, message: Dict[str, str], bot_handler: BotHandler) -> None: def handle_message(self, message: Dict[str, str], bot_handler: BotHandler) -> None:
storage = bot_handler.storage with use_storage(bot_handler.storage, ['number']) as storage:
num = storage.get('number') num = storage.get("number")
# num should already be an int, but we do `int()` to force an # num should already be an int, but we do `int()` to force an
# explicit type check # explicit type check