trezor-firmware/src/apps/ethereum/__init__.py

from trezor.messages.MessageType import (
    EthereumGetAddress,
    EthereumSignMessage,
    EthereumSignTx,
    EthereumVerifyMessage,
)
from trezor.wire import protobuf_workflow, register


def dispatch_EthereumGetAddress(*args, **kwargs):
    from .get_address import ethereum_get_address

    return ethereum_get_address(*args, **kwargs)


def dispatch_EthereumSignTx(*args, **kwargs):
    from .sign_tx import ethereum_sign_tx

    return ethereum_sign_tx(*args, **kwargs)


def dispatch_EthereumSignMessage(*args, **kwargs):
    from .sign_message import ethereum_sign_message

    return ethereum_sign_message(*args, **kwargs)


def dispatch_EthereumVerifyMessage(*args, **kwargs):
    from .verify_message import ethereum_verify_message

    return ethereum_verify_message(*args, **kwargs)


def boot():
    register(EthereumGetAddress, protobuf_workflow, dispatch_EthereumGetAddress)
    register(EthereumSignTx, protobuf_workflow, dispatch_EthereumSignTx)
    register(EthereumSignMessage, protobuf_workflow, dispatch_EthereumSignMessage)
    register(EthereumVerifyMessage, protobuf_workflow, dispatch_EthereumVerifyMessage)