AaronFiore/electrum
1
0
Fork 0
Code Activity

trezor: check if device is connected before asking passphrase

Browse Source
This commit is contained in:
ThomasV
2014-09-14 11:48:30 +02:00
parent ad4449dc38
commit ca6194e563
1 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
plugins/trezor.py
View File

@@ -79,16 +79,31 @@ class Plugin(BasePlugin):
def enable(self):
return BasePlugin.enable(self)
def trezor_is_connected(self):
try:
self.wallet.get_client().ping('t')
except:
return False
return True
@hook
def init_qt(self, gui):
self.window = gui.main_window
@hook
def close_wallet(self):
print_error("trezor: clear session")
self.wallet.client.clear_session()
if self.wallet.client:
self.wallet.client.clear_session()
@hook
def load_wallet(self, wallet):
self.wallet = wallet
if not self.wallet.check_proper_device():
QMessageBox.information(None, _('Error'), _("This wallet does not match your Trezor device"), _('OK'))
if self.trezor_is_connected():
if not self.wallet.check_proper_device():
QMessageBox.information(self.window, _('Error'), _("This wallet does not match your Trezor device"), _('OK'))
else:
QMessageBox.information(self.window, _('Error'), _("Trezor device not detected.\nContinuing in watching-only mode."), _('OK'))
@hook
def installwizard_restore(self, wizard, storage):