trezor-firmware/core/src/trezor/messages/PublicKey.py

# Automatically generated by pb2py
# fmt: off
import protobuf as p

from .HDNodeType import HDNodeType

if __debug__:
    try:
        from typing import Dict, List  # noqa: F401
        from typing_extensions import Literal  # noqa: F401
    except ImportError:
        pass


class PublicKey(p.MessageType):
    MESSAGE_WIRE_TYPE = 12

    def __init__(
        self,
        *,
        node: HDNodeType,
        xpub: str,
        root_fingerprint: int = None,
    ) -> None:
        self.node = node
        self.xpub = xpub
        self.root_fingerprint = root_fingerprint

    @classmethod
    def get_fields(cls) -> Dict:
        return {
            1: ('node', HDNodeType, p.FLAG_REQUIRED),
            2: ('xpub', p.UnicodeType, p.FLAG_REQUIRED),
            3: ('root_fingerprint', p.UVarintType, None),
        }