arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-01-09 06:50:58 +00:00
Code Issues Releases Wiki Activity

tools: support Black-compatible repr format

Browse Source
This commit is contained in:
matejcik 2018-08-15 17:33:33 +02:00
parent 59063e0cc2
commit 7ab69b18cd
1 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tools/coin_gen.py
View File

@ -56,6 +56,18 @@ def c_str_filter(b):
return json.dumps(b) return json.dumps(b)
def black_repr_filter(val):
if isinstance(val, str):
if '"' in val:
return repr(val)
else:
return c_str_filter(val)
elif isinstance(val, bytes):
return "b" + c_str_filter(val)
else:
return repr(val)
def ascii_filter(s): def ascii_filter(s):
return re.sub("[^ -\x7e]", "_", s) return re.sub("[^ -\x7e]", "_", s)
@ -69,7 +81,11 @@ def make_support_filter(support_info):
return supported_on return supported_on
MAKO_FILTERS = {"c_str": c_str_filter, "ascii": ascii_filter} MAKO_FILTERS = {
"c_str": c_str_filter,
"ascii": ascii_filter,
"black_repr": black_repr_filter,
}
def render_file(src, dst, coins, support_info): def render_file(src, dst, coins, support_info):
@ -82,7 +98,7 @@ def render_file(src, dst, coins, support_info):
support_info=support_info, support_info=support_info,
supported_on=make_support_filter(support_info), supported_on=make_support_filter(support_info),
**coins, **coins,
**MAKO_FILTERS **MAKO_FILTERS,
) )
dst.write(result) dst.write(result)