BTCPrivate
/
python-trezor
mirror of https://github.com/BTCPrivate/python-trezor.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
trezorctl
View File

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