81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
|
import hashlib
|
||
|
|
import threading
|
||
|
|
from functools import partial
|
||
|
|
|
||
|
|
from PyQt4.QtGui import *
|
||
|
|
from PyQt4.QtCore import *
|
||
|
|
import PyQt4.QtCore as QtCore
|
||
|
|
import PyQt4.QtGui as QtGui
|
||
|
|
|
||
|
|
from electrum.plugins import hook
|
||
|
|
from electrum.i18n import _
|
||
|
|
from electrum_gui.qt import HelpButton, EnterButton
|
||
|
|
from electrum_gui.qt.util import ThreadedButton, Buttons, CancelButton, OkButton
|
||
|
|
|
||
|
|
from labels import LabelsPlugin
|
||
|
|
|
||
|
|
|
||
|
|
class Plugin(LabelsPlugin):
|
||
|
|
|
||
|
|
def __init__(self, *args):
|
||
|
|
LabelsPlugin.__init__(self, *args)
|
||
|
|
self.obj = QObject()
|
||
|
|
|
||
|
|
def requires_settings(self):
|
||
|
|
return True
|
||
|
|
|
||
|
|
def settings_widget(self, window):
|
||
|
|
return EnterButton(_('Settings'),
|
||
|
|
partial(self.settings_dialog, window))
|
||
|
|
|
||
|
|
def settings_dialog(self, window):
|
||
|
|
d = QDialog(window)
|
||
|
|
vbox = QVBoxLayout(d)
|
||
|
|
layout = QGridLayout()
|
||
|
|
vbox.addLayout(layout)
|
||
|
|
layout.addWidget(QLabel("Label sync options: "), 2, 0)
|
||
|
|
self.upload = ThreadedButton("Force upload",
|
||
|
|
partial(self.push_thread, window.wallet),
|
||
|
|
self.done_processing)
|
||
|
|
layout.addWidget(self.upload, 2, 1)
|
||
|
|
self.download = ThreadedButton("Force download",
|
||
|
|
partial(self.pull_thread, window.wallet, True),
|
||
|
|
self.done_processing)
|
||
|
|
layout.addWidget(self.download, 2, 2)
|
||
|
|
self.accept = OkButton(d, _("Done"))
|
||
|
|
vbox.addLayout(Buttons(CancelButton(d), self.accept))
|
||
|
|
if d.exec_():
|
||
|
|
return True
|
||
|
|
else:
|
||
|
|
return False
|
||
|
|
|
||
|
|
def on_pulled(self, wallet):
|
||
|
|
self.obj.emit(SIGNAL('labels_changed'), wallet)
|
||
|
|
|
||
|
|
def done_processing(self):
|
||
|
|
QMessageBox.information(None, _("Labels synchronised"),
|
||
|
|
_("Your labels have been synchronised."))
|
||
|
|
|
||
|
|
@hook
|
||
|
|
def on_new_window(self, window):
|
||
|
|
window.connect(window.app, SIGNAL('labels_changed'), window.update_tabs)
|
||
|
|
wallet = window.wallet
|
||
|
|
nonce = self.get_nonce(wallet)
|
||
|
|
self.print_error("wallet", wallet.basename(), "nonce is", nonce)
|
||
|
|
mpk = ''.join(sorted(wallet.get_master_public_keys().values()))
|
||
|
|
if not mpk:
|
||
|
|
return
|
||
|
|
password = hashlib.sha1(mpk).digest().encode('hex')[:32]
|
||
|
|
iv = hashlib.sha256(password).digest()[:16]
|
||
|
|
wallet_id = hashlib.sha256(mpk).digest().encode('hex')
|
||
|
|
self.wallets[wallet] = (password, iv, wallet_id)
|
||
|
|
# If there is an auth token we can try to actually start syncing
|
||
|
|
t = threading.Thread(target=self.pull_thread, args=(wallet, False))
|
||
|
|
t.setDaemon(True)
|
||
|
|
t.start()
|
||
|
|
|
||
|
|
@hook
|
||
|
|
def on_close_window(self, window):
|
||
|
|
self.wallets.pop(window.wallet)
|
||
|
|
|