You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
2.3 KiB
71 lines
2.3 KiB
from PyQt4.Qt import QDialog, QInputDialog, QLineEdit, QVBoxLayout, QLabel, SIGNAL |
|
import PyQt4.QtCore as QtCore |
|
import threading |
|
|
|
from electrum.plugins import BasePlugin, hook |
|
|
|
from ledger import LedgerPlugin, BTChipWallet |
|
|
|
class Plugin(LedgerPlugin): |
|
|
|
@hook |
|
def load_wallet(self, wallet, window): |
|
if type(wallet) != BTChipWallet: |
|
return |
|
wallet.plugin = self |
|
if self.handler is None: |
|
self.handler = BTChipQTHandler(window) |
|
if self.btchip_is_connected(wallet): |
|
if not wallet.check_proper_device(): |
|
window.show_error(_("This wallet does not match your Ledger device")) |
|
wallet.force_watching_only = True |
|
else: |
|
window.show_error(_("Ledger device not detected.\nContinuing in watching-only mode.")) |
|
wallet.force_watching_only = True |
|
|
|
|
|
class BTChipQTHandler: |
|
|
|
def __init__(self, win): |
|
self.win = win |
|
self.win.connect(win, SIGNAL('btchip_done'), self.dialog_stop) |
|
self.win.connect(win, SIGNAL('btchip_message_dialog'), self.message_dialog) |
|
self.win.connect(win, SIGNAL('btchip_auth_dialog'), self.auth_dialog) |
|
self.done = threading.Event() |
|
|
|
def stop(self): |
|
self.win.emit(SIGNAL('btchip_done')) |
|
|
|
def show_message(self, msg): |
|
self.message = msg |
|
self.win.emit(SIGNAL('btchip_message_dialog')) |
|
|
|
def prompt_auth(self, msg): |
|
self.done.clear() |
|
self.message = msg |
|
self.win.emit(SIGNAL('btchip_auth_dialog')) |
|
self.done.wait() |
|
return self.response |
|
|
|
def auth_dialog(self): |
|
response = QInputDialog.getText(None, "Ledger Wallet Authentication", self.message, QLineEdit.Password) |
|
if not response[1]: |
|
self.response = None |
|
else: |
|
self.response = str(response[0]) |
|
self.done.set() |
|
|
|
def message_dialog(self): |
|
self.d = QDialog() |
|
self.d.setModal(1) |
|
self.d.setWindowTitle('Ledger') |
|
self.d.setWindowFlags(self.d.windowFlags() | QtCore.Qt.WindowStaysOnTopHint) |
|
l = QLabel(self.message) |
|
vbox = QVBoxLayout(self.d) |
|
vbox.addWidget(l) |
|
self.d.show() |
|
|
|
def dialog_stop(self): |
|
if self.d is not None: |
|
self.d.hide() |
|
self.d = None
|
|
|