mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-29 09:38:08 +00:00
74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
import datetime
|
|
import sys
|
|
import time
|
|
from subprocess import run
|
|
|
|
|
|
class Device:
|
|
def __init__(self, uhub_location, device_port):
|
|
self.uhub_location = uhub_location
|
|
self.device_port = device_port
|
|
|
|
@staticmethod
|
|
def log(msg):
|
|
print(msg, flush=True, file=sys.stderr)
|
|
|
|
def run_trezorctl(self, cmd: str, **kwargs):
|
|
full_cmd = "trezorctl "
|
|
full_cmd += cmd
|
|
self.log("[software/trezorctl] Running '{}'".format(full_cmd))
|
|
return run(full_cmd, shell=True, check=True, **kwargs)
|
|
|
|
def check_model(self, model=None):
|
|
res = self.run_trezorctl("list", capture_output=True, text=True)
|
|
self.log(res.stdout)
|
|
self.log(res.stderr)
|
|
self.run_trezorctl("get-features | grep version")
|
|
lines = res.stdout.splitlines()
|
|
if len(lines) != 1:
|
|
raise RuntimeError("{} trezors connected".format(len(lines)))
|
|
if model and model not in lines[0]:
|
|
raise RuntimeError(
|
|
"invalid trezor model connected (expected {})".format(model)
|
|
)
|
|
return lines[0].split()[0]
|
|
|
|
def reboot(self):
|
|
self.power_off()
|
|
self.power_on()
|
|
|
|
def power_on(self):
|
|
self.now()
|
|
self.log("[hardware/usb] Turning power on...")
|
|
run(
|
|
"uhubctl -l {} -p {} -a on".format(self.uhub_location, self.device_port),
|
|
shell=True,
|
|
check=True,
|
|
)
|
|
self.wait(3)
|
|
|
|
def power_off(self):
|
|
self.now()
|
|
self.log("[hardware/usb] Turning power off...")
|
|
run(
|
|
"uhubctl -l {} -p {} -r 100 -a off".format(
|
|
self.uhub_location, self.device_port
|
|
),
|
|
shell=True,
|
|
check=True,
|
|
)
|
|
self.wait(3)
|
|
|
|
def touch(self, location, action):
|
|
raise NotImplementedError
|
|
|
|
@staticmethod
|
|
def wait(seconds):
|
|
Device.now()
|
|
Device.log("[software] Waiting for {} seconds...".format(seconds))
|
|
time.sleep(seconds)
|
|
|
|
@staticmethod
|
|
def now():
|
|
Device.log("\n[timestamp] {}".format(datetime.datetime.now()))
|