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

enable trezor plugin if about to create/restore wallet

This commit is contained in:
m0mchil 2014-07-12 00:28:26 +03:00
parent 8a7f81a9de
commit e4a6a2962e
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
plugins/trezor.py
View File

@ -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))