1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-22 22:38:08 +00:00

trezorctl: cleanup set_homescreen call

This commit is contained in:
Pavol Rusnak 2017-12-16 21:29:52 +01:00
parent b42fc6fb1f
commit de95c44ad1
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D

View File

@ -209,11 +209,13 @@ def set_flags(connect, flags):
@click.option('-f', '--filename', default=None)
@click.pass_obj
def set_homescreen(connect, filename):
if filename and filename.endswith('.toif'):
if filename is None:
img = b'\x00'
elif filename.endswith('.toif'):
img = open(filename, 'rb').read()
if img[:8] != b'TOIf\x90\x00\x90\x00':
raise ValueError('File is not a TOIF file with size of 144x144')
elif filename is not None:
raise CallException(types.Failure_DataError, 'File is not a TOIF file with size of 144x144')
else:
from PIL import Image
im = Image.open(filename)
if im.size != (128, 64):
@ -227,8 +229,6 @@ def set_homescreen(connect, filename):
o = (i + j * 128)
img[o // 8] |= (1 << (7 - o % 8))
img = bytes(img)
else:
img = b'\x00'
return connect().apply_settings(homescreen=img)