Have close_wallet hook take a wallet.

2015-12-31 10:31:49 +09:00
parent b8f2d9dfd4
commit 93d05e8cfe
4 changed files with 5 additions and 10 deletions
--- a/plugins/trezor/plugin.py
+++ b/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