1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-14 10:28:10 +00:00
trezor-firmware/core/src/trezor/strings.py
grdddj b8ea21d24a feat(all): implement translations into Trezor
Co-authored-by matejcik <ja@matejcik.cz>
2024-02-12 14:49:32 +01:00

122 lines
3.8 KiB
Python

import utime
from micropython import const
_SECONDS_1970_TO_2000 = const(946684800)
def format_amount(amount: int, decimals: int) -> str:
if amount < 0:
amount = -amount
sign = "-"
else:
sign = ""
d = 10**decimals
integer = amount // d
decimal = amount % d
# TODO: bug in mpz: https://github.com/micropython/micropython/issues/8984
grouped_integer = f"{integer:,}".lstrip(",")
s = f"{sign}{grouped_integer}.{decimal:0{decimals}}".rstrip("0").rstrip(".")
return s
def format_ordinal(number: int) -> str:
return str(number) + {1: "st", 2: "nd", 3: "rd"}.get(
4 if 10 <= number % 100 < 20 else number % 10, "th"
)
def format_plural_english(string: str, count: int, plural: str) -> str:
"""
Adds plural form to a string based on `count`.
!! Does not work with irregular words !!
Example:
>>> format_plural_english("We need {count} more {plural}", 3, "share")
'We need 3 more shares'
>>> format_plural_english("We need {count} more {plural}", 1, "share")
'We need 1 more share'
>>> format_plural_english("{count} {plural}", 4, "candy")
'4 candies'
"""
if not all(s in string for s in ("{count}", "{plural}")):
# string needs to have {count} and {plural} inside
raise ValueError
if count == 0 or count > 1:
# candy -> candies, but key -> keys
if plural[-1] == "y" and plural[-2] not in "aeiouy":
plural = plural[:-1] + "ies"
elif plural[-1] in "hsxz":
plural = plural + "es"
else:
plural = plural + "s"
return string.format(count=count, plural=plural)
def format_plural(string: str, count: int, plurals: str) -> str:
"""
Adds plural form to a string based on `count`.
"""
if not all(s in string for s in ("{count}", "{plural}")):
# string needs to have {count} and {plural} inside
raise ValueError
plural_options = plurals.split("|")
if len(plural_options) not in (2, 3):
# plurals need to have 2 or 3 options
raise ValueError
# First one is for singular, last one is for MANY (or zero)
if count == 1:
plural = plural_options[0]
else:
plural = plural_options[-1]
# In case there are three options, the middle one is for FEW (2-4)
if len(plural_options) == 3:
# TODO: this is valid for czech - are there some other languages that have it differently?
# In that case we need to add language-specific rules
if 1 < count < 5:
plural = plural_options[1]
return string.format(count=count, plural=plural)
def format_duration_ms(milliseconds: int, unit_plurals: dict[str, str]) -> str:
"""
Returns human-friendly representation of a duration. Truncates all decimals.
"""
units: tuple[tuple[str, int], ...] = (
(unit_plurals["hour"], 60 * 60 * 1000),
(unit_plurals["minute"], 60 * 1000),
(unit_plurals["second"], 1000),
)
for unit, divisor in units:
if milliseconds >= divisor:
break
else:
unit = unit_plurals["millisecond"]
divisor = 1
return format_plural("{count} {plural}", milliseconds // divisor, unit)
def format_timestamp(timestamp: int) -> str:
"""
Returns human-friendly representation of a unix timestamp (in seconds format).
Minutes and seconds are always displayed as 2 digits.
Example:
>>> format_timestamp_to_human(0)
'1970-01-01 00:00:00'
>>> format_timestamp_to_human(1616051824)
'2021-03-18 07:17:04'
"""
# By doing the conversion to 2000-based epoch in Python, we take advantage of the
# bignum implementation, and get another 30 years out of the 32-bit mp_int_t
# that is used internally.
d = utime.gmtime2000(timestamp - _SECONDS_1970_TO_2000)
return f"{d[0]}-{d[1]:02d}-{d[2]:02d} {d[3]:02d}:{d[4]:02d}:{d[5]:02d}"