From 41820eec298cbd13654eb03cc3ad7ae26816b569 Mon Sep 17 00:00:00 2001
From: Neil Booth <kyuupichan@gmail.com>
Date: Wed, 28 Oct 2015 10:55:04 +0900
Subject: [PATCH] Get correct domain in address_dialog

Fixes #1509
Also clean up imports.
---
 gui/qt/address_dialog.py | 27 ++++++---------------------
 gui/qt/history_widget.py |  7 +++++--
 2 files changed, 11 insertions(+), 23 deletions(-)

diff --git a/gui/qt/address_dialog.py b/gui/qt/address_dialog.py
index edf90f3f..d4812a1a 100644
--- a/gui/qt/address_dialog.py
+++ b/gui/qt/address_dialog.py
@@ -16,23 +16,11 @@
 # You should have received a copy of the GNU General Public License
 # along with this program. If not, see <http://www.gnu.org/licenses/>.
 
-import sys, time, datetime, re, threading
-from electrum.i18n import _, set_language
-from electrum.util import print_error, print_msg
-import os.path, json, ast, traceback
-import shutil
-import StringIO
-
-
-try:
-    import PyQt4
-except Exception:
-    sys.exit("Error: Could not import PyQt4 on Linux systems, you may try 'sudo apt-get install python-qt4'")
+from electrum.i18n import _
 
+import PyQt4
 from PyQt4.QtGui import *
 from PyQt4.QtCore import *
-import PyQt4.QtCore as QtCore
-
 
 from util import *
 from history_widget import HistoryWidget
@@ -63,16 +51,16 @@ class AddressDialog(QDialog):
 
         vbox.addWidget(QLabel(_("History")))
         self.hw = HistoryWidget(self.parent)
+        self.hw.get_domain = self.get_domain
         vbox.addWidget(self.hw)
 
         vbox.addStretch(1)
         vbox.addLayout(Buttons(CloseButton(self)))
         self.format_amount = self.parent.format_amount
+        self.hw.update()
 
-        h = self.wallet.get_history([self.address])
-        self.hw.update(h)
-
-
+    def get_domain(self):
+        return [self.address]
 
     def show_qr(self):
         text = self.address
@@ -80,6 +68,3 @@ class AddressDialog(QDialog):
             self.parent.show_qrcode(text, 'Address')
         except Exception as e:
             self.show_message(str(e))
-
-
-
diff --git a/gui/qt/history_widget.py b/gui/qt/history_widget.py
index 0266abb0..dc34ff2b 100644
--- a/gui/qt/history_widget.py
+++ b/gui/qt/history_widget.py
@@ -55,10 +55,13 @@ class HistoryWidget(MyTreeWidget):
             icon = QIcon(":icons/confirmed.png")
         return icon, time_str
 
+    def get_domain(self):
+        '''Replaced in address_dialog.py'''
+        return self.wallet.get_account_addresses(self.parent.current_account)
+
     def on_update(self):
         self.wallet = self.parent.wallet
-        domain = self.wallet.get_account_addresses(self.parent.current_account)
-        h = self.wallet.get_history(domain)
+        h = self.wallet.get_history(self.get_domain())
 
         item = self.currentItem()
         current_tx = item.data(0, Qt.UserRole).toString() if item else None