Fix labels plugin wallet popping bug.

2015-12-05 21:47:17 +09:00
parent 50dcee69d0
commit a0b031c7df
2 changed files with 4 additions and 2 deletions
--- a/plugins/labels/labels.py
+++ b/plugins/labels/labels.py
@@ -147,3 +147,6 @@ class LabelsPlugin(BasePlugin):
        t = threading.Thread(target=self.pull_thread, args=(wallet, False))
        t.setDaemon(True)
        t.start()
+
+    def stop_wallet(self, wallet):
+        self.wallets.pop(wallet, None)
--- a/plugins/labels/qt.py
+++ b/plugins/labels/qt.py
@@ -62,5 +62,4 @@ class Plugin(LabelsPlugin):

    @hook
    def on_close_window(self, window):
-        self.wallets.pop(window.wallet)
-
+        self.stop_wallet(window.wallet)