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

feat(python): support derive_cardano setting

Browse Source
This commit is contained in:
matejcik 2021-10-20 13:15:33 +02:00 committed by matejcik
parent 846eca36ca
commit 9922e7ba48
1 changed files with 24 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
python/src/trezorlib/client.py
View File

@ -17,7 +17,7 @@
import logging import logging
import os import os
import warnings import warnings
from typing import Optional from typing import TYPE_CHECKING, Optional
from mnemonic import Mnemonic from mnemonic import Mnemonic
@ -25,6 +25,10 @@ from . import MINIMUM_FIRMWARE_VERSION, exceptions, mapping, messages, tools
from .log import DUMP_BYTES from .log import DUMP_BYTES
from .messages import Capability from .messages import Capability
if TYPE_CHECKING:
from .ui import TrezorClientUI
from .transport import Transport
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
VENDORS = ("bitcointrezor.com", "trezor.io") VENDORS = ("bitcointrezor.com", "trezor.io")
@ -85,16 +89,17 @@ class TrezorClient:
def __init__( def __init__(
self, self,
transport, transport: "Transport",
ui, ui: "TrezorClientUI",
session_id=None, session_id: Optional[bytes] = None,
derive_cardano: Optional[bool] = None,
): ):
LOG.info(f"creating client instance for device: {transport.get_path()}") LOG.info(f"creating client instance for device: {transport.get_path()}")
self.transport = transport self.transport = transport
self.ui = ui self.ui = ui
self.session_counter = 0 self.session_counter = 0
self.session_id = session_id self.session_id = session_id
self.init_device(session_id=session_id) self.init_device(session_id=session_id, derive_cardano=derive_cardano)
def open(self): def open(self):
if self.session_counter == 0: if self.session_counter == 0:
@ -257,7 +262,11 @@ class TrezorClient:
@tools.session @tools.session
def init_device( def init_device(
self, *, session_id: bytes = None, new_session: bool = False self,
*,
session_id: bytes = None,
new_session: bool = False,
derive_cardano: Optional[bool] = None,
) -> Optional[bytes]: ) -> Optional[bytes]:
"""Initialize the device and return a session ID. """Initialize the device and return a session ID.
@ -292,7 +301,15 @@ class TrezorClient:
elif session_id is not None: elif session_id is not None:
self.session_id = session_id self.session_id = session_id
resp = self.call_raw(messages.Initialize(session_id=self.session_id)) resp = self.call_raw(
messages.Initialize(
session_id=self.session_id,
derive_cardano=derive_cardano,
)
)
if isinstance(resp, messages.Failure):
# can happen if `derive_cardano` does not match the current session
raise exceptions.TrezorFailure(resp)
if not isinstance(resp, messages.Features): if not isinstance(resp, messages.Features):
raise exceptions.TrezorException("Unexpected response to Initialize") raise exceptions.TrezorException("Unexpected response to Initialize")