gui/gui_classic/installwizard.py

   1 from PyQt4.QtGui import *
   2 from PyQt4.QtCore import *
   3 import PyQt4.QtCore as QtCore
   4
   5 from electrum.i18n import _
   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 import threading
  15
  16 class InstallWizard(QDialog):
  17
  18     def __init__(self, config, network, storage):
  19         QDialog.__init__(self)
  20         self.config = config
  21         self.network = network
  22         self.storage = storage
  23         self.setMinimumSize(575, 400)
  24         self.setWindowTitle('Electrum')
  25         self.connect(self, QtCore.SIGNAL('accept'), self.accept)
  26
  27
  28     def restore_or_create(self):
  29
  30         grid = QGridLayout()
  31         grid.setSpacing(5)
  32
  33         msg = _("Wallet file not found.")+"\n\n"+_("Do you want to create a new wallet, or to restore an existing one?"+"\n")
  34         label = QLabel(msg)
  35         label.setWordWrap(True)
  36         grid.addWidget(label, 0, 0)
  37
  38         gb = QGroupBox()
  39
  40         b1 = QRadioButton(gb)
  41         b1.setText(_("Create new wallet"))
  42         b1.setChecked(True)
  43
  44         b2 = QRadioButton(gb)
  45         b2.setText(_("Restore wallet from seed"))
  46
  47         b3 = QRadioButton(gb)
  48         b3.setText(_("Restore wallet from master public key"))
  49
  50         grid.addWidget(b1,1,0)
  51         grid.addWidget(b2,2,0)
  52         grid.addWidget(b3,3,0)
  53
  54         vbox = QVBoxLayout(self)
  55         vbox.addLayout(grid)
  56
  57         vbox.addStretch(1)
  58         vbox.addLayout(ok_cancel_buttons(self, _('Next')))
  59
  60         if not self.exec_():
  61             return
  62
  63         if b1.isChecked():
  64             answer = 'create'
  65         elif b2.isChecked():
  66             answer = 'restore'
  67         else:
  68             answer = 'watching'
  69
  70         return answer
  71
  72
  73     def verify_seed(self, wallet):
  74         r = self.seed_dialog(False)
  75         if not r:
  76             return
  77
  78         if r != wallet.seed:
  79             QMessageBox.warning(None, _('Error'), _('Incorrect seed'), _('OK'))
  80             return False
  81         else:
  82             return True
  83
  84
  85     def seed_dialog(self, is_restore=True):
  86
  87         if self.layout(): QWidget().setLayout(self.layout())
  88
  89         vbox = QVBoxLayout(self)
  90         if is_restore:
  91             msg = _("Please enter your wallet seed." + ' ')
  92             msg += _("Your seed can be entered as a sequence of words, or as a hexadecimal string."+ ' \n')
  93         else:
  94             msg = _("Your seed is important!") \
  95                   + "\n" + _("To make sure that you have properly saved your seed, please retype it here." + ' ')
  96
  97         logo = QLabel()
  98         logo.setPixmap(QPixmap(":icons/seed.png").scaledToWidth(56))
  99         logo.setMaximumWidth(60)
 100
 101         label = QLabel(msg)
 102         label.setWordWrap(True)
 103
 104         seed_e = QTextEdit()
 105         seed_e.setMaximumHeight(100)
 106
 107         vbox.addWidget(label)
 108
 109         grid = QGridLayout()
 110         grid.addWidget(logo, 0, 0)
 111         grid.addWidget(seed_e, 0, 1)
 112
 113         vbox.addLayout(grid)
 114
 115
 116         vbox.addStretch(1)
 117         vbox.addLayout(ok_cancel_buttons(self, _('Next')))
 118
 119         if not self.exec_():
 120             return
 121
 122         try:
 123             seed = str(seed_e.toPlainText())
 124             seed.decode('hex')
 125         except:
 126             try:
 127                 seed = mnemonic.mn_decode( seed.split() )
 128             except:
 129                 QMessageBox.warning(None, _('Error'), _('I cannot decode this'), _('OK'))
 130                 return
 131
 132         if not seed:
 133             QMessageBox.warning(None, _('Error'), _('No seed'), _('OK'))
 134             return
 135
 136         return seed
 137
 138
 139
 140     def waiting_dialog(self, task, msg= _("Please wait...")):
 141         def target():
 142             task()
 143             self.emit(QtCore.SIGNAL('accept'))
 144
 145         if self.layout(): QWidget().setLayout(self.layout())
 146         vbox = QVBoxLayout(self)
 147         self.waiting_label = QLabel(msg)
 148         vbox.addWidget(self.waiting_label)
 149         self.show()
 150         t = threading.Thread(target = target)
 151         t.start()
 152         self.exec_()
 153
 154
 155
 156     def mpk_dialog(self):
 157
 158         if self.layout(): QWidget().setLayout(self.layout())
 159
 160         vbox = QVBoxLayout(self)
 161         msg = _("Please enter your master public key.")
 162
 163         label=QLabel(msg)
 164         label.setWordWrap(True)
 165         vbox.addWidget(label)
 166
 167         mpk_e = QTextEdit()
 168         mpk_e.setMaximumHeight(100)
 169         vbox.addWidget(mpk_e)
 170
 171         grid = QGridLayout()
 172         grid.setSpacing(8)
 173         vbox.addLayout(grid)
 174
 175         vbox.addStretch(1)
 176         vbox.addLayout(ok_cancel_buttons(self, _('Next')))
 177
 178         if not self.exec_(): return
 179
 180         mpk = str(mpk_e.toPlainText())
 181         return mpk
 182
 183
 184     def network_dialog(self):
 185
 186         if self.layout(): QWidget().setLayout(self.layout())
 187
 188         grid = QGridLayout()
 189         grid.setSpacing(5)
 190
 191         label = QLabel(_("Network") + ":")
 192         grid.addWidget(label, 0, 0)
 193
 194         gb = QGroupBox()
 195
 196         b1 = QRadioButton(gb)
 197         b1.setText(_("Auto connect"))
 198         b1.setChecked(True)
 199
 200         b2 = QRadioButton(gb)
 201         b2.setText(_("Select server manually"))
 202
 203         b3 = QRadioButton(gb)
 204         b3.setText(_("Stay offline"))
 205
 206         grid.addWidget(b1,1,0)
 207         grid.addWidget(b2,2,0)
 208         grid.addWidget(b3,3,0)
 209
 210         vbox = QVBoxLayout(self)
 211         vbox.addLayout(grid)
 212
 213         vbox.addStretch(1)
 214         vbox.addLayout(ok_cancel_buttons(self, _('Next')))
 215
 216         if not self.exec_():
 217             return
 218
 219         if b2.isChecked():
 220             return NetworkDialog(self.network, self.config, None).do_exec()
 221
 222         elif b1.isChecked():
 223             self.config.set_key('auto_cycle', True, True)
 224             return
 225
 226         else:
 227             self.config.set_key("server", None, True)
 228             self.config.set_key('auto_cycle', False, True)
 229             return
 230
 231
 232
 233     def show_seed(self, wallet):
 234         from seed_dialog import make_seed_dialog
 235         if self.layout(): QWidget().setLayout(self.layout())
 236         make_seed_dialog(self, wallet.seed, wallet.imported_keys)
 237         self.exec_()
 238
 239
 240     def password_dialog(self, wallet):
 241         msg = _("Please choose a password to encrypt your wallet keys.")+'\n'\
 242               +_("Leave these fields empty if you want to disable encryption.")
 243         from password_dialog import make_password_dialog, run_password_dialog
 244         if self.layout(): QWidget().setLayout(self.layout())
 245         make_password_dialog(self, wallet, msg)
 246         run_password_dialog(self, wallet, self)
 247
 248
 249     def run(self):
 250
 251         action = self.restore_or_create()
 252         if not action: exit()
 253
 254         wallet = Wallet(self.storage)
 255         gap = self.config.get('gap_limit', 5)
 256         if gap != 5:
 257             wallet.gap_limit = gap
 258             wallet.storage.put('gap_limit', gap, True)
 259
 260         if action == 'create':
 261             wallet.init_seed(None)
 262             self.show_seed(wallet)
 263             if self.verify_seed(wallet):
 264                 def create():
 265                     wallet.save_seed()
 266                     wallet.create_accounts()
 267                     wallet.synchronize()  # generate first addresses offline
 268                 self.waiting_dialog(create)
 269             else:
 270                 return
 271
 272         elif action == 'restore':
 273             # ask for seed and gap.
 274             seed = self.seed_dialog()
 275             if not seed:
 276                 return
 277             wallet.init_seed(str(seed))
 278             wallet.save_seed()
 279
 280         elif action == 'watching':
 281             # ask for seed and gap.
 282             mpk = self.mpk_dialog()
 283             if not mpk:
 284                 return
 285             wallet.seed = ''
 286
 287             print eval(mpk)
 288             try:
 289                 c0, K0 = eval(mpk)
 290             except:
 291                 QMessageBox.warning(None, _('Error'), _('error'), _('OK'))
 292                 return
 293             wallet.create_watching_only_wallet(c0,K0)
 294
 295
 296         else: raise
 297
 298         #if not self.config.get('server'):
 299         self.network_dialog()
 300
 301         # start wallet threads
 302         wallet.start_threads(self.network)
 303
 304         if action == 'restore':
 305
 306             def wait_for_wallet():
 307                 wallet.set_up_to_date(False)
 308                 while not wallet.is_up_to_date():
 309                     msg = "%s\n%s %d\n%s %.1f"%(_("Please wait..."),_("Addresses generated:"),len(wallet.addresses(True)),_("Kilobytes received:"), self.network.interface.bytes_received/1024.)
 310                     self.waiting_label.setText(msg)
 311                     time.sleep(0.1)
 312
 313             def wait_for_network():
 314                 while not self.network.interface.is_connected:
 315                     msg = "%s \n" % (_("Connecting..."))
 316                     self.waiting_label.setText(msg)
 317                     time.sleep(0.1)
 318
 319             def restore():
 320                 # wait until we are connected, because the user might have selected another server
 321                 wait_for_network()
 322
 323                 # try to restore old account
 324                 wallet.create_old_account()
 325                 wait_for_wallet()
 326
 327                 if wallet.is_found():
 328                     wallet.seed_version = 4
 329                     wallet.storage.put('seed_version', wallet.seed_version, True)
 330                 else:
 331                     wallet.accounts.pop(0)
 332                     wallet.create_accounts()
 333                     wait_for_wallet()
 334
 335             self.waiting_dialog(restore)
 336
 337             if wallet.is_found():
 338                 QMessageBox.information(None, _('Information'), _("Recovery successful"), _('OK'))
 339             else:
 340                 QMessageBox.information(None, _('Information'), _("No transactions found for this seed"), _('OK'))
 341
 342             wallet.fill_addressbook()
 343
 344         self.password_dialog(wallet)
 345
 346         return wallet