import PyQt4.QtCore as QtCore
from electrum.i18n import _
-from electrum import Wallet, mnemonic
+from electrum import Wallet, Wallet_2of3
-from seed_dialog import SeedDialog
+import seed_dialog
from network_dialog import NetworkDialog
from util import *
from amountedit import AmountEdit
import sys
import threading
+from electrum.plugins import run_hook
class InstallWizard(QDialog):
self.setWindowTitle('Electrum')
self.connect(self, QtCore.SIGNAL('accept'), self.accept)
+ self.stack = QStackedLayout()
+ self.setLayout(self.stack)
+
+
+ def set_layout(self, layout):
+ w = QWidget()
+ w.setLayout(layout)
+ self.stack.setCurrentIndex(self.stack.addWidget(w))
+
def restore_or_create(self):
grid = QGridLayout()
grid.setSpacing(5)
- msg = _("Electrum could not find an existing wallet.")+"\n\n"+_("Did you use Electrum before and want to restore a previous wallet or is this your first time and do you want to create a new wallet?"+"\n")
+ msg = _("Electrum could not find an existing wallet.") + "\n\n" \
+ + _("What do you want to do?") + "\n"
label = QLabel(msg)
label.setWordWrap(True)
grid.addWidget(label, 0, 0)
b1.setChecked(True)
b2 = QRadioButton(gb)
- b2.setText(_("Restore wallet from seed"))
-
- b3 = QRadioButton(gb)
- b3.setText(_("Restore wallet from master public key"))
+ b2.setText(_("Restore an existing wallet"))
grid.addWidget(b1,1,0)
grid.addWidget(b2,2,0)
- grid.addWidget(b3,3,0)
- vbox = QVBoxLayout(self)
- vbox.addLayout(grid)
+ vbox = QVBoxLayout()
+ self.set_layout(vbox)
+ vbox.addLayout(grid)
vbox.addStretch(1)
vbox.addLayout(ok_cancel_buttons(self, _('Next')))
if not self.exec_():
return
- if b1.isChecked():
- answer = 'create'
- elif b2.isChecked():
- answer = 'restore'
- else:
- answer = 'watching'
+ return 'create' if b1.isChecked() else 'restore'
- return answer
- def verify_seed(self, wallet):
- r = self.seed_dialog(False)
+ def verify_seed(self, seed, sid):
+ r = self.enter_seed_dialog(False, sid)
if not r:
return
- if r != wallet.seed:
+ if r != seed:
QMessageBox.warning(None, _('Error'), _('Incorrect seed'), _('OK'))
return False
else:
return True
- def seed_dialog(self, is_restore=True):
-
- if self.layout(): QWidget().setLayout(self.layout())
-
- vbox = QVBoxLayout(self)
- if is_restore:
- msg = _("Please enter your wallet seed.") + "\n"
- msg += _("Your seed can be entered as a sequence of words, or as a hexadecimal string."+ ' \n')
- else:
- msg = _("Your seed is important!") \
- + "\n" + _("To make sure that you have properly saved your seed, please retype it here." + ' ')
-
- logo = QLabel()
- logo.setPixmap(QPixmap(":icons/seed.png").scaledToWidth(56))
- logo.setMaximumWidth(60)
-
- label = QLabel(msg)
- label.setWordWrap(True)
-
- seed_e = QTextEdit()
- seed_e.setMaximumHeight(100)
+ def get_seed_text(self, seed_e):
+ return unicode(seed_e.toPlainText())
- vbox.addWidget(label)
- grid = QGridLayout()
- grid.addWidget(logo, 0, 0)
- grid.addWidget(seed_e, 0, 1)
-
- vbox.addLayout(grid)
+ def is_seed(self, seed_e):
+ text = self.get_seed_text(seed_e)
+ return Wallet.is_seed(text) or Wallet.is_mpk(text)
+ def enter_seed_dialog(self, is_restore, sid):
+ vbox, seed_e = seed_dialog.enter_seed_box(is_restore, sid)
vbox.addStretch(1)
- vbox.addLayout(ok_cancel_buttons(self, _('Next')))
-
+ hbox, button = ok_cancel_buttons2(self, _('Next'))
+ vbox.addLayout(hbox)
+ button.setEnabled(False)
+ seed_e.textChanged.connect(lambda: button.setEnabled(self.is_seed(seed_e)))
+ self.set_layout(vbox)
if not self.exec_():
return
+ return self.get_seed_text(seed_e)
- try:
- seed = str(seed_e.toPlainText())
- seed.decode('hex')
- except:
- try:
- seed = mnemonic.mn_decode( seed.split() )
- except:
- QMessageBox.warning(None, _('Error'), _('I cannot decode this'), _('OK'))
- return
- if not seed:
- QMessageBox.warning(None, _('Error'), _('No seed'), _('OK'))
- return
+ def double_seed_dialog(self):
+ vbox = QVBoxLayout()
+ vbox1, seed_e1 = seed_dialog.enter_seed_box(True, 'hot')
+ vbox2, seed_e2 = seed_dialog.enter_seed_box(True, 'cold')
+ vbox.addLayout(vbox1)
+ vbox.addLayout(vbox2)
+ vbox.addStretch(1)
+ hbox, button = ok_cancel_buttons2(self, _('Next'))
+ vbox.addLayout(hbox)
+ button.setEnabled(False)
+ f = lambda: button.setEnabled(self.is_seed(seed_e1) and self.is_seed(seed_e2))
+ seed_e1.textChanged.connect(f)
+ seed_e2.textChanged.connect(f)
+ self.set_layout(vbox)
+ if not self.exec_():
+ return
+ return self.get_seed_text(seed_e1), self.get_seed_text(seed_e2)
- return seed
task()
self.emit(QtCore.SIGNAL('accept'))
- if self.layout(): QWidget().setLayout(self.layout())
- vbox = QVBoxLayout(self)
+ vbox = QVBoxLayout()
self.waiting_label = QLabel(msg)
vbox.addWidget(self.waiting_label)
+ self.set_layout(vbox)
t = threading.Thread(target = target)
t.start()
self.exec_()
- def mpk_dialog(self):
-
- if self.layout(): QWidget().setLayout(self.layout())
-
- vbox = QVBoxLayout(self)
-
- vbox.addWidget(QLabel(_("Please enter your master public key.")))
-
- grid = QGridLayout()
- grid.setSpacing(8)
-
- label = QLabel(_("Key"))
- grid.addWidget(label, 0, 0)
- mpk_e = QTextEdit()
- mpk_e.setMaximumHeight(100)
- grid.addWidget(mpk_e, 0, 1)
-
- label = QLabel(_("Chain"))
- grid.addWidget(label, 1, 0)
- chain_e = QTextEdit()
- chain_e.setMaximumHeight(100)
- grid.addWidget(chain_e, 1, 1)
-
- vbox.addLayout(grid)
-
- vbox.addStretch(1)
- vbox.addLayout(ok_cancel_buttons(self, _('Next')))
-
- if not self.exec_(): return None, None
-
- mpk = str(mpk_e.toPlainText()).strip()
- chain = str(chain_e.toPlainText()).strip()
- return mpk, chain
-
def network_dialog(self):
- if self.layout(): QWidget().setLayout(self.layout())
-
grid = QGridLayout()
grid.setSpacing(5)
label = QLabel(_("Electrum communicates with remote servers to get information about your transactions and addresses. The servers all fulfil the same purpose only differing in hardware. In most cases you simply want to let Electrum pick one at random if you have a preference though feel free to select a server manually.") + "\n\n" \
- + _("How do you want to connect to a server: "))
+ + _("How do you want to connect to a server:")+" ")
label.setWordWrap(True)
grid.addWidget(label, 0, 0)
grid.addWidget(b2,2,0)
#grid.addWidget(b3,3,0)
- vbox = QVBoxLayout(self)
+ vbox = QVBoxLayout()
vbox.addLayout(grid)
vbox.addStretch(1)
vbox.addLayout(ok_cancel_buttons(self, _('Next')))
+ self.set_layout(vbox)
if not self.exec_():
return
self.config.set_key('auto_cycle', False, True)
return
-
-
- def show_seed(self, wallet):
- from seed_dialog import make_seed_dialog
- vbox = make_seed_dialog(wallet.seed, wallet.imported_keys)
- vbox.addLayout(ok_cancel_buttons(self, _("Next")))
+ def show_message(self, msg):
+ vbox = QVBoxLayout()
+ vbox.addWidget(QLabel(msg))
+ vbox.addStretch(1)
+ vbox.addLayout(close_button(self, _('Next')))
+ self.set_layout(vbox)
+ if not self.exec_():
+ return None
+
+ def question(self, msg):
+ vbox = QVBoxLayout()
+ vbox.addWidget(QLabel(msg))
+ vbox.addStretch(1)
+ vbox.addLayout(ok_cancel_buttons(self, _('OK')))
+ self.set_layout(vbox)
+ if not self.exec_():
+ return None
+ return True
- if self.layout(): QWidget().setLayout(self.layout())
- self.setLayout(vbox)
- if not self.exec_():
- exit()
+ def show_seed(self, seed, sid):
+ vbox = seed_dialog.show_seed_box(seed, sid)
+ vbox.addLayout(ok_cancel_buttons(self, _("Next")))
+ self.set_layout(vbox)
+ return self.exec_()
def password_dialog(self, wallet):
msg = _("Please choose a password to encrypt your wallet keys.")+'\n'\
+_("Leave these fields empty if you want to disable encryption.")
from password_dialog import make_password_dialog, run_password_dialog
- if self.layout(): QWidget().setLayout(self.layout())
- make_password_dialog(self, wallet, msg)
- run_password_dialog(self, wallet, self)
+ self.set_layout( make_password_dialog(self, wallet, msg) )
+ return run_password_dialog(self, wallet, self)
+
+ def choose_wallet_type(self):
+ grid = QGridLayout()
+ grid.setSpacing(5)
+
+ msg = _("Choose your wallet.")
+ label = QLabel(msg)
+ label.setWordWrap(True)
+ grid.addWidget(label, 0, 0)
+
+ gb = QGroupBox()
+
+ b1 = QRadioButton(gb)
+ b1.setText(_("Standard wallet (protected by password)"))
+ b1.setChecked(True)
+
+ b2 = QRadioButton(gb)
+ b2.setText(_("Multi-signature wallet (two-factor authentication)"))
- def run(self):
+ grid.addWidget(b1,1,0)
+ grid.addWidget(b2,2,0)
+
+ vbox = QVBoxLayout()
+
+ vbox.addLayout(grid)
+ vbox.addStretch(1)
+ vbox.addLayout(ok_cancel_buttons(self, _('Next')))
+
+ self.set_layout(vbox)
+ if not self.exec_():
+ return
+
+ if b1.isChecked():
+ return 'standard'
+ elif b2.isChecked():
+ return '2of3'
- action = self.restore_or_create()
- if not action: exit()
- wallet = Wallet(self.storage)
- gap = self.config.get('gap_limit', 5)
- if gap != 5:
- wallet.gap_limit = gap
- wallet.storage.put('gap_limit', gap, True)
+ def run(self, action = None):
+
+ if action is None:
+ action = self.restore_or_create()
+
+ if action is None:
+ return
+
+ if action == 'create':
+ t = self.choose_wallet_type()
+ if not t:
+ return
+
+ if t == '2of3':
+ run_hook('create_cold_seed', self.storage, self)
+ return
+
+
+ if action in ['create', 'create2of3']:
+
+ wallet = Wallet(self.storage)
- if action == 'create':
wallet.init_seed(None)
- self.show_seed(wallet)
- if self.verify_seed(wallet):
- def create():
- wallet.save_seed()
- wallet.create_accounts()
- wallet.synchronize() # generate first addresses offline
- self.waiting_dialog(create)
- else:
+ seed = wallet.get_mnemonic(None)
+ sid = 'hot' if action == 'create2of3' else None
+ if not self.show_seed(seed, sid):
return
-
- elif action == 'restore':
- # ask for seed and gap.
- seed = self.seed_dialog()
- if not seed:
+ if not self.verify_seed(seed, sid):
return
- wallet.init_seed(str(seed))
- wallet.save_seed()
+ ok, old_password, password = self.password_dialog(wallet)
+ wallet.save_seed(password)
+
+ if action == 'create2of3':
+ run_hook('create_third_key', wallet, self)
+ if not wallet.master_public_keys.get("remote/"):
+ return
- elif action == 'watching':
- # ask for seed and gap.
- K, chain = self.mpk_dialog()
- if not K:
+ wallet.create_accounts(password)
+ # generate first addresses offline
+ self.waiting_dialog(wallet.synchronize)
+
+ elif action == 'restore':
+ # dialog box will accept either seed or xpub.
+ # use two boxes for 2of3
+ t = self.choose_wallet_type()
+ if not t:
return
- wallet.seed = ''
- wallet.create_watching_only_wallet(chain,K)
+
+ if t == 'standard':
+ text = self.enter_seed_dialog(True, None)
+ if Wallet.is_seed(text):
+ wallet = Wallet.from_seed(text, self.storage)
+ ok, old_password, password = self.password_dialog(wallet)
+ wallet.save_seed(password)
+ wallet.create_accounts(password)
+ elif Wallet.is_mpk(text):
+ wallet = Wallet.from_mpk(text, self.storage)
+ else:
+ return
+
+ elif t in ['2of2', '2of3']:
+ r = self.double_seed_dialog()
+ if not r:
+ return
+ text1, text2 = r
+ wallet = Wallet_2of3(self.storage)
+
+ if Wallet.is_seed(text1):
+ xpriv, xpub = bip32_root(text1)
+ elif Wallet.is_mpk(text1):
+ xpub = text1
+ wallet.add_master_public_key("m/", xpub)
+
+ if Wallet.is_seed(text2):
+ xpriv2, xpub2 = bip32_root(text2)
+ elif Wallet.is_mpk(text2):
+ xpub2 = text2
+ wallet.add_master_public_key("cold/", xpub2)
+
+ run_hook('restore_third_key', wallet, self)
+
+ wallet.create_accounts(None)
+
+
else: raise
#if not self.config.get('server'):
- self.network_dialog()
+ if self.network:
+ if self.network.interfaces:
+ self.network_dialog()
+ else:
+ QMessageBox.information(None, _('Warning'), _('You are offline'), _('OK'))
+ self.network.stop()
+ self.network = None
# start wallet threads
wallet.start_threads(self.network)
self.waiting_dialog(lambda: wallet.restore(self.waiting_label.setText))
- if wallet.is_found():
- QMessageBox.information(None, _('Information'), _("Recovery successful"), _('OK'))
+ if self.network:
+ if wallet.is_found():
+ QMessageBox.information(None, _('Information'), _("Recovery successful"), _('OK'))
+ else:
+ QMessageBox.information(None, _('Information'), _("No transactions found for this seed"), _('OK'))
else:
- QMessageBox.information(None, _('Information'), _("No transactions found for this seed"), _('OK'))
-
- wallet.fill_addressbook()
-
- self.password_dialog(wallet)
+ QMessageBox.information(None, _('Information'), _("This wallet was restored offline. It may contain more addresses than displayed."), _('OK'))
return wallet
git://git.novaco.in / electrum-nvc.git / blobdiff