arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-14 03:30:02 +00:00
Code Issues Releases Wiki Activity

wip trezorlib wait-for-emulator to new UdpTransport

Browse Source
This commit is contained in:
M1nd3r 2024-09-13 12:04:35 +02:00
parent 9942f06b11
commit 28f25b6130
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
python/src/trezorlib/cli/trezorctl.py
View File

@ -29,7 +29,7 @@ from ..client import TrezorClient
from ..transport import DeviceIsBusy, new_enumerate_devices
from ..transport.new import channel_database
from ..transport.new.client import NewTrezorClient
from ..transport.udp import UdpTransport
from ..transport.new.udp import UdpTransport
from . import (
AliasedGroup,
NewTrezorConnection,
@ -404,7 +404,7 @@ def usb_reset() -> None:
@cli.command()
@click.option("-t", "--timeout", type=float, default=10, help="Timeout in seconds")
@click.pass_obj
def wait_for_emulator(obj: TrezorConnection, timeout: float) -> None:
def wait_for_emulator(obj: NewTrezorConnection, timeout: float) -> None:
"""Wait until Trezor Emulator comes up.
Tries to connect to emulator and returns when it succeeds.
@ -416,7 +416,7 @@ def wait_for_emulator(obj: TrezorConnection, timeout: float) -> None:
path = path.replace("udp:", "")
start = time.monotonic()
UdpTransport(path, skip_protocol_detection=True).wait_until_ready(timeout)
UdpTransport(path).wait_until_ready(timeout)
end = time.monotonic()
LOG.info(f"Waited for {end - start:.3f} seconds")