From e4a6a2962ee03e42ea80498521714aa0a5f64a59 Mon Sep 17 00:00:00 2001
From: m0mchil <pishtov@gmail.com>
Date: Sat, 12 Jul 2014 00:28:26 +0300
Subject: [PATCH] enable trezor plugin if about to create/restore wallet

---
 plugins/trezor.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/plugins/trezor.py b/plugins/trezor.py
index b0504be4..0798cb8e 100644
--- a/plugins/trezor.py
+++ b/plugins/trezor.py
@@ -36,6 +36,7 @@ class Plugin(BasePlugin):
     def __init__(self, gui, name):
         BasePlugin.__init__(self, gui, name)
         self._is_available = self._init()
+        self.wallet = None
 
     def _init(self):
         return TREZOR
@@ -43,9 +44,24 @@ class Plugin(BasePlugin):
     def is_available(self):
         return self._is_available
 
+    def set_enabled(self, enabled):
+        self.wallet.storage.put('use_' + self.name, enabled)
+
+    def is_enabled(self):
+        if not self.is_available():
+            return False
+
+        if not self.wallet:
+            return True
+
+        return self.wallet.storage.get('use_' + self.name) is True
+
     def enable(self):
         return BasePlugin.enable(self)
 
+    def load_wallet(self, wallet):
+        self.wallet = wallet
+
     def add_wallet_types(self, wallet_types):
         wallet_types.append(('trezor', _("Trezor wallet"), TrezorWallet))