From 57e50ee326db4fddc3834c7e47a05c6da7dd35a0 Mon Sep 17 00:00:00 2001
From: ThomasV <thomasv@gitorious>
Date: Tue, 19 Aug 2014 11:03:24 +0200
Subject: [PATCH] handle exceptions raised if trezor is not found

---
 gui/qt/installwizard.py | 6 +++++-
 plugins/trezor.py       | 6 +++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/gui/qt/installwizard.py b/gui/qt/installwizard.py
index 2b074f81d..617686e30 100644
--- a/gui/qt/installwizard.py
+++ b/gui/qt/installwizard.py
@@ -421,7 +421,11 @@ class InstallWizard(QDialog):
                 wallet.add_master_public_key("remote/", xpub2)
 
             elif action == 'create_accounts':
-                wallet.create_accounts(password)
+                try:
+                    wallet.create_accounts(password)
+                except BaseException as e:
+                    QMessageBox.information(None, _('Error'), str(e), _('OK'))
+                    return
                 self.waiting_dialog(wallet.synchronize)
 
             elif action == 'create_cold_seed':
diff --git a/plugins/trezor.py b/plugins/trezor.py
index 921b69afe..487209f94 100644
--- a/plugins/trezor.py
+++ b/plugins/trezor.py
@@ -70,7 +70,11 @@ class Plugin(BasePlugin):
 
     def installwizard_restore(self, wizard, storage):
         wallet = TrezorWallet(storage)
-        wallet.create_accounts(None)
+        try:
+            wallet.create_accounts(None)
+        except BaseException as e:
+            QMessageBox.information(None, _('Error'), str(e), _('OK'))
+            return
         return wallet
 
     def send_tx(self, tx):