electrum

commit 5baa5feb114b21147fc36d3df2916455a251da90
parent 5e8d28a1872e07496114994d866251c250a1bce4
Author: Neil Booth <kyuupichan@gmail.com>
Date:   Sun, 10 Jan 2016 18:41:20 +0900

Put restore_wallet_class back in the derived classes

Diffstat:
M
lib/wallet.py
 | 
1
-
M
plugins/ledger/ledger.py
 | 
1
+
M
plugins/trezor/plugin.py
 | 
2
++

3 files changed, 3 insertions(+), 1 deletion(-)
diff --git a/lib/wallet.py b/lib/wallet.py
@@ -1746,7 +1746,6 @@ class BIP32_HD_Wallet(BIP32_Wallet):
 class BIP44_Wallet(BIP32_HD_Wallet):
     root_derivation = "m/44'/0'"
     wallet_type = 'bip44'
-    restore_wallet_class = BIP44_Wallet
 
     def can_sign_xpubkey(self, x_pubkey):
         xpub, sequence = BIP32_Account.parse_xpubkey(x_pubkey)
diff --git a/plugins/ledger/ledger.py b/plugins/ledger/ledger.py
@@ -32,6 +32,7 @@ except ImportError:
 class BTChipWallet(BIP44_Wallet):
     wallet_type = 'btchip'
     device = 'Ledger'
+    restore_wallet_class = BIP44_Wallet
 
     def __init__(self, storage):
         BIP44_Wallet.__init__(self, storage)
diff --git a/plugins/trezor/plugin.py b/plugins/trezor/plugin.py
@@ -32,6 +32,8 @@ class TrezorCompatibleWallet(BIP44_Wallet):
     #   - DEVICE_IDS
     #   - wallet_type
 
+    restore_wallet_class = BIP44_Wallet
+
     def __init__(self, storage):
         BIP44_Wallet.__init__(self, storage)
         # After timeout seconds we clear the device session