import PyQt4.QtCore as QtCore
from electrum.i18n import _
-from electrum import Wallet, mnemonic
+from electrum import Wallet, Wallet_2of2, Wallet_2of3
+from electrum import bitcoin
+from electrum import util
-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
+
+
+MSG_ENTER_ANYTHING = _("Please enter a wallet seed, a master public key, a list of Bitcoin addresses, or a list of private keys")
+MSG_SHOW_MPK = _("This is your master public key")
+MSG_ENTER_MPK = _("Please enter your master public key")
+MSG_ENTER_COLD_MPK = _("Please enter the master public key of your cosigner wallet")
+MSG_ENTER_SEED_OR_MPK = _("Please enter a wallet seed, or master public key")
+MSG_VERIFY_SEED = _("Your seed is important!") + "\n" + _("To make sure that you have properly saved your seed, please retype it here.")
+
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):
+ vbox = QVBoxLayout()
+
+ main_label = QLabel(_("Electrum could not find an existing wallet."))
+ vbox.addWidget(main_label)
+
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")
- label = QLabel(msg)
+ label = QLabel(_("What do you want to do?"))
label.setWordWrap(True)
grid.addWidget(label, 0, 0)
- gb = QGroupBox()
+ gb1 = QGroupBox()
+ grid.addWidget(gb1, 0, 0)
- b1 = QRadioButton(gb)
+ group1 = QButtonGroup()
+
+ b1 = QRadioButton(gb1)
b1.setText(_("Create new wallet"))
b1.setChecked(True)
- b2 = QRadioButton(gb)
- b2.setText(_("Restore wallet from seed"))
+ b2 = QRadioButton(gb1)
+ b2.setText(_("Restore an existing wallet"))
- b3 = QRadioButton(gb)
- b3.setText(_("Restore wallet from master public key"))
+ group1.addButton(b1)
+ group1.addButton(b2)
- grid.addWidget(b1,1,0)
- grid.addWidget(b2,2,0)
- grid.addWidget(b3,3,0)
-
- vbox = QVBoxLayout(self)
+ grid.addWidget(b1, 1, 0)
+ grid.addWidget(b2, 2, 0)
vbox.addLayout(grid)
+ grid2 = QGridLayout()
+ grid2.setSpacing(5)
+
+ class ClickableLabel(QLabel):
+ def mouseReleaseEvent(self, ev):
+ self.emit(SIGNAL('clicked()'))
+
+ label2 = ClickableLabel(_("Wallet type:") + " [+]")
+ hbox = QHBoxLayout()
+ hbox.addWidget(label2)
+ grid2.addLayout(hbox, 0, 0)
+
+ gb2 = QGroupBox()
+ grid.addWidget(gb2, 3, 0)
+ group2 = QButtonGroup()
+
+ self.wallet_types = [
+ ('standard', _("Standard wallet"), Wallet),
+ ('2of2', _("Multisig wallet (2 of 2)"), Wallet_2of2),
+ ('2of3', _("Multisig wallet (2 of 3)"), Wallet_2of3)
+ ]
+ run_hook('add_wallet_types', self.wallet_types)
+
+ for i, (t,l,c) in enumerate(self.wallet_types):
+ button = QRadioButton(gb2)
+ button.setText(l)
+ grid2.addWidget(button, i+1, 0)
+ group2.addButton(button)
+ group2.setId(button, i)
+ if i==0:
+ button.setChecked(True)
+ #else:
+ # button.setHidden(True)
+
+
+ def toggle():
+ buttons = group2.buttons()
+ x = buttons[1].isHidden()
+ label2.setText(_("Wallet type:") + (' [+]' if x else ' [-]'))
+ for b in buttons[1:]:
+ b.setHidden(not x)
+
+ self.connect(label2, SIGNAL('clicked()'), toggle)
+ grid2.addWidget(label2)
+
+ vbox.addLayout(grid2)
vbox.addStretch(1)
- vbox.addLayout(ok_cancel_buttons(self, _('Next')))
+ hbox, button = ok_cancel_buttons2(self, _('Next'))
+ vbox.addLayout(hbox)
+ self.set_layout(vbox)
+ self.show()
+ self.raise_()
+ button.setDefault(True)
if not self.exec_():
- return
+ return None, None
- if b1.isChecked():
- answer = 'create'
- elif b2.isChecked():
- answer = 'restore'
- else:
- answer = 'watching'
+ action = 'create' if b1.isChecked() else 'restore'
+ wallet_type = self.wallet_types[group2.checkedId()][0]
+ return action, wallet_type
- return answer
-
- def verify_seed(self, wallet):
- r = self.seed_dialog(False)
+ def verify_seed(self, seed, sid):
+ r = self.enter_seed_dialog(MSG_VERIFY_SEED, 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)
+ def get_seed_text(self, seed_e):
+ text = unicode(seed_e.toPlainText()).strip()
+ text = ' '.join(text.split())
+ return text
- seed_e = QTextEdit()
- seed_e.setMaximumHeight(100)
+ def is_any(self, seed_e):
+ text = self.get_seed_text(seed_e)
+ return Wallet.is_seed(text) or Wallet.is_old_mpk(text) or Wallet.is_xpub(text) or Wallet.is_xprv(text) or Wallet.is_address(text) or Wallet.is_private_key(text)
- vbox.addWidget(label)
-
- grid = QGridLayout()
- grid.addWidget(logo, 0, 0)
- grid.addWidget(seed_e, 0, 1)
-
- vbox.addLayout(grid)
+ def is_mpk(self, seed_e):
+ text = self.get_seed_text(seed_e)
+ return Wallet.is_xpub(text) or Wallet.is_old_mpk(text)
+ def is_xpub(self, seed_e):
+ text = self.get_seed_text(seed_e)
+ return Wallet.is_xpub(text)
+ def enter_seed_dialog(self, msg, sid):
+ vbox, seed_e = seed_dialog.enter_seed_box(msg, 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_any(seed_e)))
+ self.set_layout(vbox)
+ if not self.exec_():
+ return
+ return self.get_seed_text(seed_e)
+
+
+ def multi_mpk_dialog(self, xpub_hot, n):
+ vbox = QVBoxLayout()
+ vbox0, seed_e0 = seed_dialog.enter_seed_box(MSG_SHOW_MPK, 'hot')
+ vbox.addLayout(vbox0)
+ seed_e0.setText(xpub_hot)
+ seed_e0.setReadOnly(True)
+ entries = []
+ for i in range(n):
+ vbox2, seed_e2 = seed_dialog.enter_seed_box(MSG_ENTER_COLD_MPK, 'cold')
+ vbox.addLayout(vbox2)
+ entries.append(seed_e2)
+ vbox.addStretch(1)
+ hbox, button = ok_cancel_buttons2(self, _('Next'))
+ vbox.addLayout(hbox)
+ button.setEnabled(False)
+ f = lambda: button.setEnabled( map(lambda e: self.is_xpub(e), entries) == [True]*len(entries))
+ for e in entries:
+ e.textChanged.connect(f)
+ self.set_layout(vbox)
if not self.exec_():
return
+ return map(lambda e: self.get_seed_text(e), entries)
+
+
+ def multi_seed_dialog(self, n):
+ vbox = QVBoxLayout()
+ vbox1, seed_e1 = seed_dialog.enter_seed_box(MSG_ENTER_SEED_OR_MPK, 'hot')
+ vbox.addLayout(vbox1)
+ entries = [seed_e1]
+ for i in range(n):
+ vbox2, seed_e2 = seed_dialog.enter_seed_box(MSG_ENTER_SEED_OR_MPK, 'cold')
+ vbox.addLayout(vbox2)
+ entries.append(seed_e2)
+ vbox.addStretch(1)
+ hbox, button = ok_cancel_buttons2(self, _('Next'))
+ vbox.addLayout(hbox)
+ button.setEnabled(False)
- 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
+ f = lambda: button.setEnabled( map(lambda e: self.is_any(e), entries) == [True]*len(entries))
+ for e in entries:
+ e.textChanged.connect(f)
+
+ self.set_layout(vbox)
+ if not self.exec_():
+ return
+ return map(lambda e: self.get_seed_text(e), entries)
- if not seed:
- QMessageBox.warning(None, _('Error'), _('No seed'), _('OK'))
- return
- 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.show()
+ 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)
- msg = _("Please enter your master public key.")
-
- label=QLabel(msg)
- label.setWordWrap(True)
- vbox.addWidget(label)
-
- mpk_e = QTextEdit()
- mpk_e.setMaximumHeight(100)
- vbox.addWidget(mpk_e)
-
- grid = QGridLayout()
- grid.setSpacing(8)
- vbox.addLayout(grid)
-
- vbox.addStretch(1)
- vbox.addLayout(ok_cancel_buttons(self, _('Next')))
-
- if not self.exec_(): return
-
- mpk = str(mpk_e.toPlainText())
- return mpk
-
def network_dialog(self):
- if self.layout(): QWidget().setLayout(self.layout())
-
grid = QGridLayout()
grid.setSpacing(5)
- label = QLabel(_("Electrum communicates with Electrum 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 or stay offline.") + "\n\n" \
- + _("How do you want to connect to a server: "))
+ 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:")+" ")
label.setWordWrap(True)
grid.addWidget(label, 0, 0)
b2 = QRadioButton(gb)
b2.setText(_("Select server manually"))
- b3 = QRadioButton(gb)
- b3.setText(_("Stay offline"))
+ #b3 = QRadioButton(gb)
+ #b3.setText(_("Stay offline"))
grid.addWidget(b1,1,0)
grid.addWidget(b2,2,0)
- grid.addWidget(b3,3,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
- if self.layout(): QWidget().setLayout(self.layout())
- make_seed_dialog(self, wallet.seed, wallet.imported_keys)
- self.exec_()
+ def show_message(self, msg, icon=None):
+ vbox = QVBoxLayout()
+ self.set_layout(vbox)
+ if icon:
+ logo = QLabel()
+ logo.setPixmap(icon)
+ vbox.addWidget(logo)
+ vbox.addWidget(QLabel(msg))
+ vbox.addStretch(1)
+ vbox.addLayout(close_button(self, _('Next')))
+ if not self.exec_():
+ return None
+
+
+ def question(self, msg, icon=None):
+ vbox = QVBoxLayout()
+ self.set_layout(vbox)
+ if icon:
+ logo = QLabel()
+ logo.setPixmap(icon)
+ vbox.addWidget(logo)
+ vbox.addWidget(QLabel(msg))
+ vbox.addStretch(1)
+ vbox.addLayout(ok_cancel_buttons(self, _('OK')))
+ if not self.exec_():
+ return None
+ return True
- def password_dialog(self, wallet):
+ 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):
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)
-
-
- def run(self):
-
- 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)
-
- 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:
- return
-
- elif action == 'restore':
- # ask for seed and gap.
- seed = self.seed_dialog()
- if not seed:
- return
- wallet.init_seed(str(seed))
- wallet.save_seed()
+ self.set_layout( make_password_dialog(self, None, msg) )
+ return run_password_dialog(self, None, self)[2]
+
+
+ def create_cold_seed(self, wallet):
+ from electrum.bitcoin import mnemonic_to_seed, bip32_root
+ msg = _('You are about to generate the cold storage seed of your wallet.') + '\n' \
+ + _('For safety, you should do this on an offline computer.')
+ icon = QPixmap( ':icons/cold_seed.png').scaledToWidth(56)
+ if not self.question(msg, icon):
+ return
+
+ cold_seed = wallet.make_seed()
+ if not self.show_seed(cold_seed, 'cold'):
+ return
+ if not self.verify_seed(cold_seed, 'cold'):
+ return
- elif action == 'watching':
- # ask for seed and gap.
- mpk = self.mpk_dialog()
- if not mpk:
+ hex_seed = mnemonic_to_seed(cold_seed,'').encode('hex')
+ xpriv, xpub = bip32_root(hex_seed)
+ wallet.add_master_public_key('cold/', xpub)
+
+ msg = _('Your master public key was saved in your wallet file.') + '\n'\
+ + _('Your cold seed must be stored on paper; it is not in the wallet file.')+ '\n\n' \
+ + _('This program is about to close itself.') + '\n'\
+ + _('You will need to reopen your wallet on an online computer, in order to complete the creation of your wallet')
+ self.show_message(msg)
+
+
+
+ def run(self, action):
+
+ if action == 'new':
+ action, wallet_type = self.restore_or_create()
+ self.storage.put('wallet_type', wallet_type, False)
+
+ if action is None:
+ return
+
+ if action == 'restore':
+ wallet = self.restore(wallet_type)
+ if not wallet:
return
- wallet.seed = ''
+ action = None
- print eval(mpk)
- try:
- c0, K0 = eval(mpk)
- except:
- QMessageBox.warning(None, _('Error'), _('error'), _('OK'))
+ else:
+ wallet = Wallet(self.storage)
+ action = wallet.get_action()
+ # fixme: password is only needed for multiple accounts
+ password = None
+
+ while action is not None:
+
+ util.print_error("installwizard:", wallet, action)
+
+ if action == 'create_seed':
+ seed = wallet.make_seed()
+ if not self.show_seed(seed, None):
+ return
+ if not self.verify_seed(seed, None):
+ return
+ password = self.password_dialog()
+ wallet.add_seed(seed, password)
+
+ elif action == 'add_cosigner':
+ xpub_hot = wallet.master_public_keys.get("m/")
+ r = self.multi_mpk_dialog(xpub_hot, 1)
+ if not r:
+ return
+ xpub_cold = r[0]
+ wallet.add_master_public_key("cold/", xpub_cold)
+
+ elif action == 'add_two_cosigners':
+ xpub_hot = wallet.master_public_keys.get("m/")
+ r = self.multi_mpk_dialog(xpub_hot, 2)
+ if not r:
+ return
+ xpub1, xpub2 = r
+ wallet.add_master_public_key("cold/", xpub1)
+ wallet.add_master_public_key("remote/", xpub2)
+
+ elif action == 'create_accounts':
+ wallet.create_accounts(password)
+ self.waiting_dialog(wallet.synchronize)
+
+ elif action == 'create_cold_seed':
+ self.create_cold_seed(wallet)
return
- wallet.create_watching_only_wallet(c0,K0)
+ else:
+ r = run_hook('install_wizard_action', self, wallet, action)
+ if not r:
+ raise BaseException('unknown wizard action', action)
+
+ # next action
+ action = wallet.get_action()
- 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)
if action == 'restore':
-
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
+
+
+
+ def restore(self, t):
+
+ if t == 'standard':
+ text = self.enter_seed_dialog(MSG_ENTER_ANYTHING, None)
+ if not text:
+ return
+ if Wallet.is_seed(text):
+ password = self.password_dialog()
+ wallet = Wallet.from_seed(text, self.storage)
+ wallet.add_seed(text, password)
+ wallet.create_accounts(password)
+ elif Wallet.is_xprv(text):
+ password = self.password_dialog()
+ wallet = Wallet.from_xprv(text, password, self.storage)
+ elif Wallet.is_old_mpk(text):
+ wallet = Wallet.from_old_mpk(text, self.storage)
+ elif Wallet.is_xpub(text):
+ wallet = Wallet.from_xpub(text, self.storage)
+ elif Wallet.is_address(text):
+ wallet = Wallet.from_address(text, self.storage)
+ elif Wallet.is_private_key(text):
+ wallet = Wallet.from_private_key(text, self.storage)
+ else:
+ raise
+
+ elif t in ['2of2']:
+ r = self.multi_seed_dialog(1)
+ if not r:
+ return
+ text1, text2 = r
+ wallet = Wallet_2of2(self.storage)
+ if Wallet.is_seed(text1) or Wallet.is_seed(text2):
+ password = self.password_dialog()
+ else:
+ password = None
+
+ if Wallet.is_seed(text1):
+ wallet.add_seed(text1, password)
+ if Wallet.is_seed(text2):
+ wallet.add_cold_seed(text2, password)
+ else:
+ wallet.add_master_public_key("cold/", text2)
+ else:
+ assert Wallet.is_xpub(text1)
+ if Wallet.is_seed(text2):
+ wallet.add_seed(text2, password)
+ wallet.add_master_public_key("cold/", text1)
+ else:
+ wallet.add_master_public_key("m/", text1)
+ wallet.add_master_public_key("cold/", text2)
+
+ wallet.create_accounts(password)
+
+
+ elif t in ['2of3']:
+ r = self.multi_seed_dialog(2)
+ if not r:
+ return
+ text1, text2, text3 = r
+ wallet = Wallet_2of3(self.storage)
+ if Wallet.is_seed(text1) or Wallet.is_seed(text2) or Wallet.is_seed(text3):
+ password = self.password_dialog()
+ else:
+ password = None
+
+ if Wallet.is_seed(text1):
+ wallet.add_seed(text1, password)
+ if Wallet.is_seed(text2):
+ wallet.add_cold_seed(text2, password)
+ else:
+ wallet.add_master_public_key("cold/", text2)
+
+ elif Wallet.is_xpub(text1):
+ if Wallet.is_seed(text2):
+ wallet.add_seed(text2, password)
+ wallet.add_master_public_key("cold/", text1)
+ else:
+ wallet.add_master_public_key("m/", text1)
+ wallet.add_master_public_key("cold/", text2)
+
+ wallet.create_accounts(password)
+
+ else:
+ wallet = run_hook('installwizard_restore', self, self.storage)
+
+ # create first keys offline
+ self.waiting_dialog(wallet.synchronize)
+
+ return wallet
git://git.novaco.in / electrum-nvc.git / blobdiff