arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-04-20 17:19:01 +00:00
Code Issues Releases Wiki Activity

fixup! wip

Browse Source
This commit is contained in:
M1nd3r 2025-04-09 15:19:09 +02:00
parent 9c21c0b368
commit 5dbe45f3cf
2 changed files with 10 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -228,10 +228,10 @@ class TrezorClient:
def _read(self, session_id: int | None = None) -> t.Any:
if isinstance(self.protocol, ProtocolV1Channel):
self.protocol.read()
return self.protocol.read()
elif isinstance(self.protocol, ProtocolV2Channel):
assert session_id is not None
self.protocol.read(session_id=session_id)
return self.protocol.read(session_id=session_id)
else:
raise Exception("Unknown client protocol")

27
python/src/trezorlib/debuglink.py
View File

@ -827,7 +827,6 @@ class DebugUI:
def __init__(self, debuglink: DebugLink) -> None:
self.debuglink = debuglink
self.pins: t.Iterator[str] | None = None
self.clear()
def clear(self) -> None:
@ -1098,20 +1097,16 @@ class TrezorClientDebugLink(TrezorClient):
self.sync_responses()
def __getattr__(self, name: str) -> t.Any:
return getattr(self._session, name)
def __setattr__(self, name: str, value: t.Any) -> None:
if hasattr(self._session, name):
setattr(self._session, name, value)
else:
self.__dict__[name] = value
# So that we can choose right screenshotting logic (T1 vs TT)
# and know the supported debug capabilities
self.debug.model = self.model
self.debug.version = self.version
self.reset_debug_features()
@property
def protocol_version(self) -> int:
return self.client.protocol_version
def layout_type(self) -> LayoutType:
return self.debug.layout_type
def get_new_client(self) -> TrezorClientDebugLink:
new_client = TrezorClientDebugLink(
@ -1313,10 +1308,8 @@ class TrezorClientDebugLink(TrezorClient):
actual_responses = self.actual_responses
# grab a copy of the inputflow generator to raise an exception through it
if isinstance(self.client, TrezorClientDebugLink) and isinstance(
self.client.ui, DebugUI
):
input_flow = self.client.ui.input_flow
if isinstance(self.ui, DebugUI):
input_flow = self.ui.input_flow
else:
input_flow = None
@ -1433,10 +1426,6 @@ class TrezorClientDebugLink(TrezorClient):
self.actual_responses.append(resp)
return resp
def reset_protocol(self):
super().reset_protocol()
self._seedless_session = self.get_seedless_session(new_session=True)
def load_device(
session: "Session",