check wallet type in installwizard_load_wallet hook

9 years ago · 87168282b9
3 changed files with 6 additions and 0 deletions
--- a/plugins/btchipwallet.py
+++ b/plugins/btchipwallet.py
@ -95,6 +95,8 @@ class Plugin(BasePlugin):

    @hook
    def installwizard_load_wallet(self, wallet, window):
+        if type(wallet) != BTChipWallet:
+            return
        self.load_wallet(wallet, window)

    @hook
--- a/plugins/keepkey.py
+++ b/plugins/keepkey.py
@ -147,6 +147,8 @@ class Plugin(BasePlugin):

    @hook
    def installwizard_load_wallet(self, wallet, window):
+        if type(wallet) != KeepKeyWallet:
+            return
        self.load_wallet(wallet, window)

    @hook
--- a/plugins/trezor.py
+++ b/plugins/trezor.py
@ -147,6 +147,8 @@ class Plugin(BasePlugin):

    @hook
    def installwizard_load_wallet(self, wallet, window):
+        if type(wallet) != TrezorWallet:
+            return
        self.load_wallet(wallet, window)

    @hook