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.
346 lines
9.6 KiB
346 lines
9.6 KiB
from PyQt4.QtGui import * |
|
from PyQt4.QtCore import * |
|
import PyQt4.QtCore as QtCore |
|
|
|
from electrum.i18n import _ |
|
from electrum import Wallet, mnemonic |
|
|
|
from seed_dialog import SeedDialog |
|
from network_dialog import NetworkDialog |
|
from qt_util import * |
|
from amountedit import AmountEdit |
|
|
|
import sys |
|
import threading |
|
|
|
class InstallWizard(QDialog): |
|
|
|
def __init__(self, config, network, storage): |
|
QDialog.__init__(self) |
|
self.config = config |
|
self.network = network |
|
self.storage = storage |
|
self.setMinimumSize(575, 400) |
|
self.setWindowTitle('Electrum') |
|
self.connect(self, QtCore.SIGNAL('accept'), self.accept) |
|
|
|
|
|
def restore_or_create(self): |
|
|
|
grid = QGridLayout() |
|
grid.setSpacing(5) |
|
|
|
msg = _("Wallet file not found.")+"\n"+_("Do you want to create a new wallet, or to restore an existing one?") |
|
label = QLabel(msg) |
|
label.setWordWrap(True) |
|
grid.addWidget(label, 0, 0) |
|
|
|
gb = QGroupBox() |
|
|
|
b1 = QRadioButton(gb) |
|
b1.setText(_("Create new wallet")) |
|
b1.setChecked(True) |
|
|
|
b2 = QRadioButton(gb) |
|
b2.setText(_("Restore wallet from seed")) |
|
|
|
b3 = QRadioButton(gb) |
|
b3.setText(_("Restore wallet from master public key")) |
|
|
|
grid.addWidget(b1,1,0) |
|
grid.addWidget(b2,2,0) |
|
grid.addWidget(b3,3,0) |
|
|
|
vbox = QVBoxLayout(self) |
|
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 answer |
|
|
|
|
|
def verify_seed(self, wallet): |
|
r = self.seed_dialog(False) |
|
if not r: |
|
return |
|
|
|
if r != wallet.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." + ' ') |
|
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) |
|
|
|
vbox.addWidget(label) |
|
|
|
grid = QGridLayout() |
|
grid.addWidget(logo, 0, 0) |
|
grid.addWidget(seed_e, 0, 1) |
|
|
|
vbox.addLayout(grid) |
|
|
|
|
|
vbox.addStretch(1) |
|
vbox.addLayout(ok_cancel_buttons(self, _('Next'))) |
|
|
|
if not self.exec_(): |
|
return |
|
|
|
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 |
|
|
|
return seed |
|
|
|
|
|
|
|
def waiting_dialog(self, task, msg= _("Please wait...")): |
|
def target(): |
|
task() |
|
self.emit(QtCore.SIGNAL('accept')) |
|
|
|
if self.layout(): QWidget().setLayout(self.layout()) |
|
vbox = QVBoxLayout(self) |
|
self.waiting_label = QLabel(msg) |
|
vbox.addWidget(self.waiting_label) |
|
self.show() |
|
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(_("Network") + ":") |
|
grid.addWidget(label, 0, 0) |
|
|
|
gb = QGroupBox() |
|
|
|
b1 = QRadioButton(gb) |
|
b1.setText(_("Auto connect")) |
|
b1.setChecked(True) |
|
|
|
b2 = QRadioButton(gb) |
|
b2.setText(_("Select server manually")) |
|
|
|
b3 = QRadioButton(gb) |
|
b3.setText(_("Stay offline")) |
|
|
|
grid.addWidget(b1,1,0) |
|
grid.addWidget(b2,2,0) |
|
grid.addWidget(b3,3,0) |
|
|
|
vbox = QVBoxLayout(self) |
|
vbox.addLayout(grid) |
|
|
|
vbox.addStretch(1) |
|
vbox.addLayout(ok_cancel_buttons(self, _('Next'))) |
|
|
|
if not self.exec_(): |
|
return |
|
|
|
if b2.isChecked(): |
|
return NetworkDialog(self.network, self.config, None).do_exec() |
|
|
|
elif b1.isChecked(): |
|
self.config.set_key('auto_cycle', True, True) |
|
return |
|
|
|
else: |
|
self.config.set_key("server", None, True) |
|
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 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) |
|
|
|
|
|
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() |
|
|
|
elif action == 'watching': |
|
# ask for seed and gap. |
|
mpk = self.mpk_dialog() |
|
if not mpk: |
|
return |
|
wallet.seed = '' |
|
|
|
print eval(mpk) |
|
try: |
|
c0, K0 = eval(mpk) |
|
except: |
|
QMessageBox.warning(None, _('Error'), _('error'), _('OK')) |
|
return |
|
wallet.create_watching_only_wallet(c0,K0) |
|
|
|
|
|
else: raise |
|
|
|
#if not self.config.get('server'): |
|
self.network_dialog() |
|
|
|
# start wallet threads |
|
wallet.start_threads(self.network) |
|
|
|
if action == 'restore': |
|
|
|
def wait_for_wallet(): |
|
wallet.set_up_to_date(False) |
|
while not wallet.is_up_to_date(): |
|
msg = "%s\n%s %d\n%s %.1f"%(_("Please wait..."),_("Addresses generated:"),len(wallet.addresses(True)),_("Kilobytes received:"), self.network.interface.bytes_received/1024.) |
|
self.waiting_label.setText(msg) |
|
time.sleep(0.1) |
|
|
|
def wait_for_network(): |
|
while not self.network.interface.is_connected: |
|
msg = "%s \n" % (_("Connecting...")) |
|
self.waiting_label.setText(msg) |
|
time.sleep(0.1) |
|
|
|
def restore(): |
|
# wait until we are connected, because the user might have selected another server |
|
wait_for_network() |
|
|
|
# try to restore old account |
|
wallet.create_old_account() |
|
wait_for_wallet() |
|
|
|
if wallet.is_found(): |
|
wallet.seed_version = 4 |
|
wallet.storage.put('seed_version', wallet.seed_version, True) |
|
else: |
|
wallet.accounts.pop(0) |
|
wallet.create_accounts() |
|
wait_for_wallet() |
|
|
|
self.waiting_dialog(restore) |
|
|
|
if wallet.is_found(): |
|
QMessageBox.information(None, _('Information'), _("Recovery successful"), _('OK')) |
|
else: |
|
QMessageBox.information(None, _('Information'), _("No transactions found for this seed"), _('OK')) |
|
|
|
wallet.fill_addressbook() |
|
|
|
self.password_dialog(wallet) |
|
|
|
return wallet
|
|
|