trezor-firmware/core/src/trezor/pin.py

from trezor import ui

if False:
    from typing import Any


def pin_to_int(pin: str) -> int:
    return int("1" + pin)


_previous_progress = None

keepalive_callback = None  # type: Any


def show_pin_timeout(seconds: int, progress: int, message: str) -> bool:
    global _previous_progress

    if callable(keepalive_callback):
        keepalive_callback()

    if progress == 0:
        if progress != _previous_progress:
            # avoid overdraw in case of repeated progress calls
            ui.display.clear()
        ui.display.text_center(
            ui.WIDTH // 2, 37, message, ui.BOLD, ui.FG, ui.BG, ui.WIDTH
        )
    ui.display.loader(progress, False, 0, ui.FG, ui.BG)
    if seconds == 0:
        ui.display.text_center(
            ui.WIDTH // 2, ui.HEIGHT - 22, "Done", ui.BOLD, ui.FG, ui.BG, ui.WIDTH
        )
    elif seconds == 1:
        ui.display.text_center(
            ui.WIDTH // 2,
            ui.HEIGHT - 22,
            "1 second left",
            ui.BOLD,
            ui.FG,
            ui.BG,
            ui.WIDTH,
        )
    else:
        ui.display.text_center(
            ui.WIDTH // 2,
            ui.HEIGHT - 22,
            "%d seconds left" % seconds,
            ui.BOLD,
            ui.FG,
            ui.BG,
            ui.WIDTH,
        )
    ui.display.refresh()

    _previous_progress = progress
    return False