1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-17 21:22:10 +00:00
trezor-firmware/core/tools/toif_convert.py
2022-12-13 17:15:15 +01:00

41 lines
1.0 KiB
Python
Executable File

#!/usr/bin/env python3
import click
from PIL import Image
from trezorlib import toif
@click.command()
@click.argument("infile", type=click.File("rb"))
@click.argument("outfile", type=click.File("wb"))
def toif_convert(infile, outfile):
"""Convert any image format to/from TOIF or vice-versa.
\b
Examples:
toif_convert.py somefile.jpg outfile.toif
toif_convert.py infile.toif outfile.png
\b
# ensure gray-scale output TOIF
mogrify -colorspace gray icon.png
toif_convert.py icon.png icon.toif
"""
if infile.name.endswith(".toif") or infile.name == "-":
toi = toif.from_bytes(infile.read())
im = toi.to_image()
im.save(outfile)
elif outfile.name.endswith(".toif") or outfile.name == "-":
im = Image.open(infile)
toi = toif.from_image(im)
outfile.write(toi.to_bytes())
else:
raise click.ClickException("At least one of the arguments must end with .toif")
if __name__ == "__main__":
toif_convert()