gui/installwizard.py

   1 from PyQt4.QtGui import *
   2 from PyQt4.QtCore import *
   3 import PyQt4.QtCore as QtCore
   4 from i18n import _
   5
   6 from electrum import Wallet, mnemonic
   7
   8 from seed_dialog import SeedDialog
   9 from network_dialog import NetworkDialog
  10 from qt_util import *
  11 from amountedit import AmountEdit
  12
  13 import sys
  14
  15 class InstallWizard(QDialog):
  16
  17     def __init__(self, config, interface, blockchain, storage):
  18         QDialog.__init__(self)
  19         self.config = config
  20         self.interface = interface
  21         self.blockchain = blockchain
  22         self.storage = storage
  23
  24
  25     def restore_or_create(self):
  26         msg = _("Wallet file not found.")+"\n"+_("Do you want to create a new wallet, or to restore an existing one?")
  27         r = QMessageBox.question(None, _('Message'), msg, _('Create'), _('Restore'), _('Cancel'), 0, 2)
  28         if r==2: return None
  29         return 'restore' if r==1 else 'create'
  30
  31
  32     def verify_seed(self, wallet):
  33         r = self.seed_dialog(False)
  34         if r != wallet.seed:
  35             QMessageBox.warning(None, _('Error'), 'incorrect seed', 'OK')
  36             return False
  37         else:
  38             return True
  39
  40
  41     def seed_dialog(self, is_restore=True):
  42         d = QDialog()
  43         d.setModal(1)
  44
  45         vbox = QVBoxLayout()
  46         if is_restore:
  47             msg = _("Please enter your wallet seed (or your master public key if you want to create a watching-only wallet)." + ' ')
  48         else:
  49             msg = _("Your seed is important! To make sure that you have properly saved your seed, please type it here." + ' ')
  50
  51         msg += _("Your seed can be entered as a sequence of words, or as a hexadecimal string."+ '\n')
  52
  53         label=QLabel(msg)
  54         label.setWordWrap(True)
  55         vbox.addWidget(label)
  56
  57         seed_e = QTextEdit()
  58         seed_e.setMaximumHeight(100)
  59         vbox.addWidget(seed_e)
  60
  61         if is_restore:
  62             grid = QGridLayout()
  63             grid.setSpacing(8)
  64             gap_e = AmountEdit(None, True)
  65             gap_e.setText("5")
  66             grid.addWidget(QLabel(_('Gap limit')), 2, 0)
  67             grid.addWidget(gap_e, 2, 1)
  68             grid.addWidget(HelpButton(_('Keep the default value unless you modified this parameter in your wallet.')), 2, 3)
  69             vbox.addLayout(grid)
  70
  71         vbox.addLayout(ok_cancel_buttons(d))
  72         d.setLayout(vbox)
  73
  74         if not d.exec_(): return
  75
  76         try:
  77             seed = str(seed_e.toPlainText())
  78             seed.decode('hex')
  79         except:
  80             try:
  81                 seed = mnemonic.mn_decode( seed.split() )
  82             except:
  83                 QMessageBox.warning(None, _('Error'), _('I cannot decode this'), _('OK'))
  84                 return
  85
  86         if not seed:
  87             QMessageBox.warning(None, _('Error'), _('No seed'), _('OK'))
  88             return
  89
  90         if not is_restore:
  91             return seed
  92         else:
  93             try:
  94                 gap = int(unicode(gap_e.text()))
  95             except:
  96                 QMessageBox.warning(None, _('Error'), 'error', 'OK')
  97                 return
  98             return seed, gap
  99
 100
 101     def network_dialog(self):
 102         return NetworkDialog(self.interface, self.config, None).do_exec()
 103
 104
 105     def show_seed(self, wallet):
 106         d = SeedDialog()
 107         d.show_seed(wallet.seed, wallet.imported_keys)
 108
 109
 110     def password_dialog(self, wallet):
 111         from password_dialog import PasswordDialog
 112         d = PasswordDialog(wallet)
 113         d.run()
 114
 115
 116     def restore_wallet(self, wallet):
 117
 118         # wait until we are connected, because the user might have selected another server
 119         if not wallet.interface.is_connected:
 120             waiting = lambda: False if wallet.interface.is_connected else "%s \n" % (_("Connecting..."))
 121             waiting_dialog(waiting)
 122
 123         waiting = lambda: False if wallet.is_up_to_date() else "%s\n%s %d\n%s %.1f"\
 124             %(_("Please wait..."),_("Addresses generated:"),len(wallet.addresses(True)),_("Kilobytes received:"), wallet.interface.bytes_received/1024.)
 125
 126         wallet.set_up_to_date(False)
 127         wallet.interface.poke('synchronizer')
 128         waiting_dialog(waiting)
 129
 130         # try to restore old account
 131         if not wallet.is_found():
 132             print "trying old method"
 133             wallet.create_old_account()
 134             wallet.set_up_to_date(False)
 135             wallet.interface.poke('synchronizer')
 136             waiting_dialog(waiting)
 137
 138         if wallet.is_found():
 139             QMessageBox.information(None, _('Information'), _("Recovery successful"), _('OK'))
 140         else:
 141             QMessageBox.information(None, _('Information'), _("No transactions found for this seed"), _('OK'))
 142
 143         return True
 144
 145
 146     def run(self):
 147
 148         a = self.restore_or_create()
 149         if not a: exit()
 150
 151         wallet = Wallet(self.storage)
 152
 153         if a =='create':
 154             wallet.init_seed(None)
 155             self.show_seed(wallet)
 156             if self.verify_seed(wallet):
 157                 wallet.save_seed()
 158             else:
 159                 exit()
 160         else:
 161             # ask for seed and gap.
 162             sg = self.seed_dialog()
 163             if not sg: exit()
 164             seed, gap = sg
 165             if not seed: exit()
 166             wallet.gap_limit = gap
 167             if len(seed) == 128:
 168                 wallet.seed = ''
 169                 wallet.init_sequence(str(seed))
 170             else:
 171                 wallet.init_seed(str(seed))
 172                 wallet.save_seed()
 173
 174         # select a server.
 175         s = self.network_dialog()
 176         if s is None:
 177             self.config.set_key("server", None, True)
 178             self.config.set_key('auto_cycle', False, True)
 179
 180         #self.interface.start(wait = False)
 181
 182         # generate the first addresses, in case we are offline
 183         if s is None or a == 'create':
 184             wallet.synchronize()
 185
 186         # start wallet threads
 187         wallet.start_threads(self.interface, self.blockchain)
 188
 189
 190         if a == 'restore' and s is not None:
 191             try:
 192                 keep_it = self.restore_wallet(wallet)
 193                 wallet.fill_addressbook()
 194             except:
 195                 import traceback
 196                 traceback.print_exc(file=sys.stdout)
 197                 exit()
 198
 199             if not keep_it: exit()
 200
 201
 202         self.password_dialog(wallet)