mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-29 17:48:10 +00:00
71 lines
1.9 KiB
Python
71 lines
1.9 KiB
Python
import sys
|
|
import utime
|
|
from micropython import const
|
|
from typing import Any
|
|
|
|
_DEBUG = const(10)
|
|
_INFO = const(20)
|
|
_WARNING = const(30)
|
|
_ERROR = const(40)
|
|
_CRITICAL = const(50)
|
|
|
|
_leveldict = {
|
|
_DEBUG: ("DEBUG", "32"),
|
|
_INFO: ("INFO", "36"),
|
|
_WARNING: ("WARNING", "33"),
|
|
_ERROR: ("ERROR", "31"),
|
|
_CRITICAL: ("CRITICAL", "1;31"),
|
|
}
|
|
|
|
level = _DEBUG
|
|
color = True
|
|
|
|
|
|
def _log(name: str, mlevel: int, msg: str, *args: Any) -> None:
|
|
if __debug__ and mlevel >= level:
|
|
if color:
|
|
fmt = (
|
|
"%d \x1b[35m%s\x1b[0m \x1b["
|
|
+ _leveldict[mlevel][1]
|
|
+ "m%s\x1b[0m "
|
|
+ msg
|
|
)
|
|
else:
|
|
fmt = "%d %s %s " + msg
|
|
print(fmt % ((utime.ticks_us(), name, _leveldict[mlevel][0]) + args))
|
|
|
|
|
|
def debug(name: str, msg: str, *args: Any) -> None:
|
|
_log(name, _DEBUG, msg, *args)
|
|
|
|
|
|
def info(name: str, msg: str, *args: Any) -> None:
|
|
_log(name, _INFO, msg, *args)
|
|
|
|
|
|
def warning(name: str, msg: str, *args: Any) -> None:
|
|
_log(name, _WARNING, msg, *args)
|
|
|
|
|
|
def error(name: str, msg: str, *args: Any) -> None:
|
|
_log(name, _ERROR, msg, *args)
|
|
|
|
|
|
def exception(name: str, exc: BaseException) -> None:
|
|
# we are using `__class__.__name__` to avoid importing ui module
|
|
# we also need to instruct typechecker to ignore the missing argument
|
|
# in ui.Result exception
|
|
if exc.__class__.__name__ == "Result":
|
|
_log(
|
|
name,
|
|
_DEBUG,
|
|
"ui.Result: %s",
|
|
exc.value, # type: ignore[Cannot access member "value" for type "BaseException"]
|
|
)
|
|
elif exc.__class__.__name__ == "Cancelled":
|
|
_log(name, _DEBUG, "ui.Cancelled")
|
|
else:
|
|
_log(name, _ERROR, "exception:")
|
|
# since mypy 0.770 we cannot override sys, so print_exception is unknown
|
|
sys.print_exception(exc) # type: ignore ["print_exception" is not a known member of module]
|