arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-07-25 16:08:32 +00:00
Code Issues Releases Wiki Activity

fixup! feat(python): implement session based trezorlib

Browse Source
This commit is contained in:
M1nd3r 2025-03-24 14:08:03 +01:00
parent 39fbaca9f8
commit 3e4522ffd4
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -22,7 +22,6 @@ import warnings
from enum import IntEnum from enum import IntEnum
from . import exceptions, mapping, messages, models from . import exceptions, mapping, messages, models
from .mapping import ProtobufMapping
from .tools import parse_path from .tools import parse_path
from .transport import Transport, get_transport from .transport import Transport, get_transport
from .transport.thp.protocol_and_channel import Channel from .transport.thp.protocol_and_channel import Channel
@ -58,6 +57,7 @@ class TrezorClient:
) = None ) = None
pin_callback: t.Callable[[Session, messages.PinMatrixRequest], t.Any] | None = None pin_callback: t.Callable[[Session, messages.PinMatrixRequest], t.Any] | None = None
_model: models.TrezorModel
_seedless_session: Session | None = None _seedless_session: Session | None = None
_features: messages.Features | None = None _features: messages.Features | None = None
_protocol_version: int _protocol_version: int
@ -66,21 +66,27 @@ class TrezorClient:
def __init__( def __init__(
self, self,
transport: Transport, transport: Transport,
protobuf_mapping: ProtobufMapping | None = None,
protocol: Channel | None = None, protocol: Channel | None = None,
model: models.TrezorModel | None = None,
) -> None: ) -> None:
""" """
Transport needs to be opened before calling a method (or accessing Transport needs to be opened before calling a method (or accessing
an attribute) for the first time. It should be closed after you're an attribute) for the first time. It should be closed after you're
done using the client. done using the client.
""" """
LOG.info(f"creating client instance for device: {transport.get_path()}")
# Here, self.model could be set to None. Unless _init_device is False, it will
# get correctly reconfigured as part of the init_device flow.
self._model = model # type: ignore ["None" is incompatible with "TrezorModel"]
if self._model:
self.mapping = self.model.default_mapping
else:
self.mapping = mapping.DEFAULT_MAPPING
self._is_invalidated: bool = False self._is_invalidated: bool = False
self.transport = transport self.transport = transport
if protobuf_mapping is None:
self.mapping = mapping.DEFAULT_MAPPING
else:
self.mapping = protobuf_mapping
if protocol is None: if protocol is None:
self.protocol = self._get_protocol() self.protocol = self._get_protocol()
else: else: