Have close_wallet hook take a wallet.

gui/qt/main_window.py

@@ -244,7 +244,7 @@ class ElectrumWindow(QMainWindow, MessageBoxMixin, PrintError):
             self.print_error('close_wallet', self.wallet.storage.path)
             self.wallet.storage.put('accounts_expanded', self.accounts_expanded)
             self.wallet.stop_threads()
-        run_hook('close_wallet')
+        run_hook('close_wallet', self.wallet)
 
     def load_wallet(self, wallet):
         self.wallet = wallet

plugins/ledger/ledger.py

@@ -434,13 +434,11 @@ class LedgerPlugin(BasePlugin):
         return True
 
     @hook
-    def close_wallet(self):
-        pass
+    def close_wallet(self, wallet):
+        self.client = None
 
     @hook
     def installwizard_load_wallet(self, wallet, window):
-        if type(wallet) != BTChipWallet:
-            return
         self.load_wallet(wallet, window)
 
     @hook

plugins/trezor/plugin.py

@@ -193,9 +193,9 @@ class TrezorCompatiblePlugin(BasePlugin):
         return self.get_client().atleast_version(major, minor, patch)
 
     @hook
-    def close_wallet(self):
-        self.print_error("clear session")
+    def close_wallet(self, wallet):
         if self.client:
+            self.print_error("clear session")
             self.client.clear_session()
             self.client.transport.close()
             self.client = None

plugins/trezor/qt_generic.py

@@ -124,9 +124,6 @@ class QtPlugin(TrezorPlugin):
 
     @hook
     def installwizard_load_wallet(self, wallet, window):
-        if type(wallet) != self.wallet_class:
-            return
-        wallet.plugin = self
         self.load_wallet(wallet, window)
 
     @hook