arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-18 20:38:10 +00:00
Code Issues Releases Wiki Activity

src/trezor/wire: allow passing MessageType classes to register()

Browse Source
This commit is contained in:
Jan Pochyla 2018-04-05 12:34:39 +02:00
parent 2731742d34
commit 60bec0b4d1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/trezor/wire/__init__.py
View File

@ -13,6 +13,8 @@ workflow_handlers = {}
def register(mtype, handler, *args): def register(mtype, handler, *args):
'''Register `handler` to get scheduled after `mtype` message is received.''' '''Register `handler` to get scheduled after `mtype` message is received.'''
if isinstance(mtype, type) and issubclass(mtype, protobuf.MessageType):
mtype = mtype.MESSAGE_WIRE_TYPE
if mtype in workflow_handlers: if mtype in workflow_handlers:
raise KeyError raise KeyError
workflow_handlers[mtype] = (handler, args) workflow_handlers[mtype] = (handler, args)