From 3588481dbee4ec708d9ac4a25836860499d1e503 Mon Sep 17 00:00:00 2001 From: M1nd3r Date: Tue, 4 Feb 2025 15:18:39 +0100 Subject: [PATCH] feat(python): implement session based trezorctl [no changelog] --- python/src/trezorlib/cli/__init__.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/src/trezorlib/cli/__init__.py b/python/src/trezorlib/cli/__init__.py index 0f5697a4f2..b661ea354a 100644 --- a/python/src/trezorlib/cli/__init__.py +++ b/python/src/trezorlib/cli/__init__.py @@ -30,7 +30,7 @@ from .. import exceptions, messages, transport, ui from ..client import PASSPHRASE_ON_DEVICE, ProtocolVersion, TrezorClient from ..messages import Capability from ..transport import Transport -from ..transport.session import Session, SessionV1 +from ..transport.session import Session, SessionV1, SessionV2 LOG = logging.getLogger(__name__) @@ -139,6 +139,9 @@ class TrezorConnection: session = SessionV1.resume_from_id( client=client, session_id=self.session_id ) + elif client.protocol_version is ProtocolVersion.PROTOCOL_V2: + session = SessionV2(client, self.session_id) + # TODO fix resumption on THP else: raise Exception("Unsupported client protocol", client.protocol_version) if must_resume: