tmove freeze and prioritize functions into wallet class - electrum - Electrum Bitcoin wallet HTML git clone https://git.parazyd.org/electrum DIR Log DIR Files DIR Refs DIR Submodules --- DIR commit 34be8520619269fe01890c79c2116815065d924f DIR parent e463968da3373a5d8b2a68652c757da414a2debd HTML Author: thomasv <thomasv@gitorious> Date: Thu, 7 Jun 2012 11:18:11 +0200 move freeze and prioritize functions into wallet class Diffstat: M electrum | 30 +++++------------------------- M lib/gui_qt.py | 10 ++++------ M lib/wallet.py | 29 +++++++++++++++++++++++++++++ 3 files changed, 38 insertions(+), 31 deletions(-) --- DIR diff --git a/electrum b/electrum t@@ -505,37 +505,17 @@ if __name__ == '__main__': elif cmd == 'freeze': addr = args[1] - if addr in self.wallet.all_addresses() and addr not in self.wallet.frozen_addresses: - self.wallet.frozen_addresses.append(addr) - self.wallet.save() - print True - else: - print False - + print self.wallet.freeze(addr) + elif cmd == 'unfreeze': addr = args[1] - if addr in self.wallet.all_addresses() and addr in self.wallet.frozen_addresses: - self.wallet.frozen_addresses.remove(addr) - self.wallet.save() - print True - else: - print False + print self.wallet.unfreeze(addr) elif cmd == 'prioritize': addr = args[1] - if addr in self.wallet.all_addresses() and addr not in self.wallet.frozen_addresses: - self.wallet.prioritized_addresses.append(addr) - self.wallet.save() - print True - else: - print False + print self.wallet.prioritize(addr) elif cmd == 'unprioritize': addr = args[1] - if addr in self.wallet.all_addresses() and addr in self.wallet.frozen_addresses: - self.wallet.prioritized_addresses.remove(addr) - self.wallet.save() - print True - else: - print False + print self.wallet.unprioritize(addr) DIR diff --git a/lib/gui_qt.py b/lib/gui_qt.py t@@ -547,10 +547,9 @@ class ElectrumWindow(QMainWindow): addr = self.get_current_addr(True) if not addr: return if addr in self.wallet.frozen_addresses: - self.wallet.frozen_addresses.remove(addr) + self.wallet.unfreeze(addr) else: - self.wallet.frozen_addresses.append(addr) - self.wallet.save() + self.wallet.freeze(addr) self.update_receive_tab() self.freezeButton = b = EnterButton(_("Freeze"), toggle_freeze) t@@ -560,10 +559,9 @@ class ElectrumWindow(QMainWindow): addr = self.get_current_addr(True) if not addr: return if addr in self.wallet.prioritized_addresses: - self.wallet.prioritized_addresses.remove(addr) + self.wallet.unprioritize(addr) else: - self.wallet.prioritized_addresses.append(addr) - self.wallet.save() + self.wallet.prioritize(addr) self.update_receive_tab() self.prioritizeButton = b = EnterButton(_("Prioritize"), toggle_priority) DIR diff --git a/lib/wallet.py b/lib/wallet.py t@@ -995,5 +995,34 @@ class Wallet: self.interface.subscribe(self.all_addresses()) + def freeze(self,addr): + if addr in self.all_addresses() and addr not in self.frozen_addresses: + self.frozen_addresses.append(addr) + self.save() + return True + else: + return False + def unfreeze(self,addr): + if addr in self.all_addresses() and addr in self.frozen_addresses: + self.frozen_addresses.remove(addr) + self.save() + return True + else: + return False + def prioritize(self,addr): + if addr in self.all_addresses() and addr not in self.frozen_addresses and addr not in self.prioritized_addresses: + self.prioritized_addresses.append(addr) + self.save() + return True + else: + return False + + def unprioritize(self,addr): + if addr in self.all_addresses() and addr in self.prioritized_addresses: + self.prioritized_addresses.remove(addr) + self.save() + return True + else: + return False