mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-22 23:48:12 +00:00
36 lines
898 B
Python
Executable File
36 lines
898 B
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
|
|
"""
|
|
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()
|