thomasv
12 years ago
11 changed files with 768 additions and 554 deletions
@ -0,0 +1,183 @@ |
|||||
|
from PyQt4.QtGui import * |
||||
|
from PyQt4.QtCore import * |
||||
|
import PyQt4.QtCore as QtCore |
||||
|
from i18n import _ |
||||
|
|
||||
|
from electrum import Wallet, mnemonic |
||||
|
from seed_dialog import SeedDialog |
||||
|
from network_dialog import NetworkDialog |
||||
|
from qt_util import * |
||||
|
|
||||
|
class InstallWizard(QDialog): |
||||
|
|
||||
|
def __init__(self, config, interface): |
||||
|
QDialog.__init__(self) |
||||
|
self.config = config |
||||
|
self.interface = interface |
||||
|
|
||||
|
|
||||
|
def restore_or_create(self): |
||||
|
msg = _("Wallet file not found.")+"\n"+_("Do you want to create a new wallet, or to restore an existing one?") |
||||
|
r = QMessageBox.question(None, _('Message'), msg, _('Create'), _('Restore'), _('Cancel'), 0, 2) |
||||
|
if r==2: return None |
||||
|
return 'restore' if r==1 else 'create' |
||||
|
|
||||
|
|
||||
|
def verify_seed(self, wallet): |
||||
|
r = self.seed_dialog(False) |
||||
|
if r != wallet.seed: |
||||
|
QMessageBox.warning(None, _('Error'), 'incorrect seed', 'OK') |
||||
|
return False |
||||
|
else: |
||||
|
return True |
||||
|
|
||||
|
|
||||
|
def seed_dialog(self, is_restore=True): |
||||
|
d = QDialog() |
||||
|
d.setModal(1) |
||||
|
|
||||
|
vbox = QVBoxLayout() |
||||
|
if is_restore: |
||||
|
msg = _("Please enter your wallet seed (or your master public key if you want to create a watching-only wallet)." + ' ') |
||||
|
else: |
||||
|
msg = _("Your seed is important! To make sure that you have properly saved your seed, please type it here." + ' ') |
||||
|
|
||||
|
msg += _("Your seed can be entered as a sequence of words, or as a hexadecimal string."+ '\n') |
||||
|
|
||||
|
label=QLabel(msg) |
||||
|
label.setWordWrap(True) |
||||
|
vbox.addWidget(label) |
||||
|
|
||||
|
seed_e = QTextEdit() |
||||
|
seed_e.setMaximumHeight(100) |
||||
|
vbox.addWidget(seed_e) |
||||
|
|
||||
|
if is_restore: |
||||
|
grid = QGridLayout() |
||||
|
grid.setSpacing(8) |
||||
|
gap_e = AmountEdit(None, True) |
||||
|
gap_e.setText("5") |
||||
|
grid.addWidget(QLabel(_('Gap limit')), 2, 0) |
||||
|
grid.addWidget(gap_e, 2, 1) |
||||
|
grid.addWidget(HelpButton(_('Keep the default value unless you modified this parameter in your wallet.')), 2, 3) |
||||
|
vbox.addLayout(grid) |
||||
|
|
||||
|
vbox.addLayout(ok_cancel_buttons(d)) |
||||
|
d.setLayout(vbox) |
||||
|
|
||||
|
if not d.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 |
||||
|
|
||||
|
if not is_restore: |
||||
|
return seed |
||||
|
else: |
||||
|
try: |
||||
|
gap = int(unicode(gap_e.text())) |
||||
|
except: |
||||
|
QMessageBox.warning(None, _('Error'), 'error', 'OK') |
||||
|
return |
||||
|
return seed, gap |
||||
|
|
||||
|
|
||||
|
def network_dialog(self): |
||||
|
return NetworkDialog(self.interface, self.config, None).do_exec() |
||||
|
|
||||
|
|
||||
|
def show_seed(self, wallet): |
||||
|
d = SeedDialog() |
||||
|
d.show_seed(wallet.seed, wallet.imported_keys) |
||||
|
|
||||
|
|
||||
|
def password_dialog(self, wallet): |
||||
|
from password_dialog import PasswordDialog |
||||
|
d = PasswordDialog(wallet) |
||||
|
d.run() |
||||
|
|
||||
|
|
||||
|
def restore_wallet(self): |
||||
|
wallet = self.wallet |
||||
|
# wait until we are connected, because the user might have selected another server |
||||
|
if not wallet.interface.is_connected: |
||||
|
waiting = lambda: False if wallet.interface.is_connected else "%s \n" % (_("Connecting...")) |
||||
|
waiting_dialog(waiting) |
||||
|
|
||||
|
waiting = lambda: False if wallet.is_up_to_date() else "%s\n%s %d\n%s %.1f"\ |
||||
|
%(_("Please wait..."),_("Addresses generated:"),len(wallet.addresses(True)),_("Kilobytes received:"), wallet.interface.bytes_received/1024.) |
||||
|
|
||||
|
wallet.set_up_to_date(False) |
||||
|
wallet.interface.poke('synchronizer') |
||||
|
waiting_dialog(waiting) |
||||
|
if wallet.is_found(): |
||||
|
print_error( "Recovery successful" ) |
||||
|
else: |
||||
|
QMessageBox.information(None, _('Error'), _("No transactions found for this seed"), _('OK')) |
||||
|
|
||||
|
return True |
||||
|
|
||||
|
|
||||
|
def run(self): |
||||
|
|
||||
|
a = self.restore_or_create() |
||||
|
if not a: exit() |
||||
|
|
||||
|
wallet = Wallet(self.config) |
||||
|
wallet.interface = self.interface |
||||
|
|
||||
|
if a =='create': |
||||
|
wallet.init_seed(None) |
||||
|
self.show_seed(wallet) |
||||
|
if self.verify_seed(wallet): |
||||
|
wallet.save_seed() |
||||
|
else: |
||||
|
exit() |
||||
|
else: |
||||
|
# ask for seed and gap. |
||||
|
sg = gui.seed_dialog() |
||||
|
if not sg: exit() |
||||
|
seed, gap = sg |
||||
|
if not seed: exit() |
||||
|
wallet.gap_limit = gap |
||||
|
if len(seed) == 128: |
||||
|
wallet.seed = '' |
||||
|
wallet.init_sequence(str(seed)) |
||||
|
else: |
||||
|
wallet.init_seed(str(seed)) |
||||
|
wallet.save_seed() |
||||
|
|
||||
|
# select a server. |
||||
|
s = self.network_dialog() |
||||
|
if s is None: |
||||
|
self.config.set_key("server", None, True) |
||||
|
self.config.set_key('auto_cycle', False, True) |
||||
|
|
||||
|
# generate the first addresses, in case we are offline |
||||
|
if s is None or a == 'create': |
||||
|
wallet.synchronize() |
||||
|
|
||||
|
|
||||
|
if a == 'restore' and s is not None: |
||||
|
try: |
||||
|
keep_it = gui.restore_wallet() |
||||
|
wallet.fill_addressbook() |
||||
|
except: |
||||
|
import traceback |
||||
|
traceback.print_exc(file=sys.stdout) |
||||
|
exit() |
||||
|
|
||||
|
if not keep_it: exit() |
||||
|
|
||||
|
|
||||
|
self.password_dialog(wallet) |
||||
@ -0,0 +1,104 @@ |
|||||
|
#!/usr/bin/env python |
||||
|
# |
||||
|
# Electrum - lightweight Bitcoin client |
||||
|
# Copyright (C) 2013 ecdsa@github |
||||
|
# |
||||
|
# This program is free software: you can redistribute it and/or modify |
||||
|
# it under the terms of the GNU General Public License as published by |
||||
|
# the Free Software Foundation, either version 3 of the License, or |
||||
|
# (at your option) any later version. |
||||
|
# |
||||
|
# This program is distributed in the hope that it will be useful, |
||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
|
# GNU General Public License for more details. |
||||
|
# |
||||
|
# You should have received a copy of the GNU General Public License |
||||
|
# along with this program. If not, see <http://www.gnu.org/licenses/>. |
||||
|
|
||||
|
from PyQt4.QtGui import * |
||||
|
from PyQt4.QtCore import * |
||||
|
from i18n import _ |
||||
|
from qt_util import * |
||||
|
|
||||
|
|
||||
|
class PasswordDialog(QDialog): |
||||
|
|
||||
|
def __init__(self, wallet, parent=None): |
||||
|
QDialog.__init__(self, parent) |
||||
|
self.setModal(1) |
||||
|
self.wallet = wallet |
||||
|
self.parent = parent |
||||
|
|
||||
|
self.pw = QLineEdit() |
||||
|
self.pw.setEchoMode(2) |
||||
|
self.new_pw = QLineEdit() |
||||
|
self.new_pw.setEchoMode(2) |
||||
|
self.conf_pw = QLineEdit() |
||||
|
self.conf_pw.setEchoMode(2) |
||||
|
|
||||
|
vbox = QVBoxLayout() |
||||
|
if parent: |
||||
|
msg = (_('Your wallet is encrypted. Use this dialog to change your password.')+'\n'\ |
||||
|
+_('To disable wallet encryption, enter an empty new password.')) \ |
||||
|
if wallet.use_encryption else _('Your wallet keys are not encrypted') |
||||
|
else: |
||||
|
msg = _("Please choose a password to encrypt your wallet keys.")+'\n'\ |
||||
|
+_("Leave these fields empty if you want to disable encryption.") |
||||
|
vbox.addWidget(QLabel(msg)) |
||||
|
|
||||
|
grid = QGridLayout() |
||||
|
grid.setSpacing(8) |
||||
|
|
||||
|
if wallet.use_encryption: |
||||
|
grid.addWidget(QLabel(_('Password')), 1, 0) |
||||
|
grid.addWidget(self.pw, 1, 1) |
||||
|
|
||||
|
grid.addWidget(QLabel(_('New Password')), 2, 0) |
||||
|
grid.addWidget(self.new_pw, 2, 1) |
||||
|
|
||||
|
grid.addWidget(QLabel(_('Confirm Password')), 3, 0) |
||||
|
grid.addWidget(self.conf_pw, 3, 1) |
||||
|
vbox.addLayout(grid) |
||||
|
|
||||
|
vbox.addLayout(ok_cancel_buttons(self)) |
||||
|
self.setLayout(vbox) |
||||
|
|
||||
|
|
||||
|
def run(self): |
||||
|
wallet = self.wallet |
||||
|
|
||||
|
if not wallet.seed: |
||||
|
QMessageBox.information(parent, _('Error'), _('No seed'), _('OK')) |
||||
|
return |
||||
|
|
||||
|
if not self.exec_(): return |
||||
|
|
||||
|
password = unicode(self.pw.text()) if wallet.use_encryption else None |
||||
|
new_password = unicode(self.new_pw.text()) |
||||
|
new_password2 = unicode(self.conf_pw.text()) |
||||
|
|
||||
|
try: |
||||
|
seed = wallet.decode_seed(password) |
||||
|
except: |
||||
|
QMessageBox.warning(self.parent, _('Error'), _('Incorrect Password'), _('OK')) |
||||
|
return |
||||
|
|
||||
|
if new_password != new_password2: |
||||
|
QMessageBox.warning(self.parent, _('Error'), _('Passwords do not match'), _('OK')) |
||||
|
self.run() # Retry |
||||
|
|
||||
|
try: |
||||
|
wallet.update_password(seed, password, new_password) |
||||
|
except: |
||||
|
QMessageBox.warning(self.parent, _('Error'), _('Failed to update password'), _('OK')) |
||||
|
return |
||||
|
|
||||
|
QMessageBox.information(self.parent, _('Success'), _('Password was updated successfully'), _('OK')) |
||||
|
|
||||
|
if self.parent: |
||||
|
icon = QIcon(":icons/lock.png") if wallet.use_encryption else QIcon(":icons/unlock.png") |
||||
|
self.parent.password_button.setIcon( icon ) |
||||
|
|
||||
|
|
||||
|
|
||||
@ -0,0 +1,82 @@ |
|||||
|
#!/usr/bin/env python |
||||
|
# |
||||
|
# Electrum - lightweight Bitcoin client |
||||
|
# Copyright (C) 2013 ecdsa@github |
||||
|
# |
||||
|
# This program is free software: you can redistribute it and/or modify |
||||
|
# it under the terms of the GNU General Public License as published by |
||||
|
# the Free Software Foundation, either version 3 of the License, or |
||||
|
# (at your option) any later version. |
||||
|
# |
||||
|
# This program is distributed in the hope that it will be useful, |
||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
|
# GNU General Public License for more details. |
||||
|
# |
||||
|
# You should have received a copy of the GNU General Public License |
||||
|
# along with this program. If not, see <http://www.gnu.org/licenses/>. |
||||
|
|
||||
|
from PyQt4.QtGui import * |
||||
|
from PyQt4.QtCore import * |
||||
|
import PyQt4.QtCore as QtCore |
||||
|
from i18n import _ |
||||
|
from electrum import mnemonic |
||||
|
from qrcodewidget import QRCodeWidget |
||||
|
|
||||
|
class SeedDialog(QDialog): |
||||
|
def __init__(self, parent=None): |
||||
|
QDialog.__init__(self, parent) |
||||
|
self.setModal(1) |
||||
|
self.setWindowTitle('Electrum' + ' - ' + _('Seed')) |
||||
|
|
||||
|
|
||||
|
def show_seed(self, seed, imported_keys, parent=None): |
||||
|
|
||||
|
brainwallet = ' '.join(mnemonic.mn_encode(seed)) |
||||
|
|
||||
|
label1 = QLabel(_("Your wallet generation seed is")+ ":") |
||||
|
|
||||
|
seed_text = QTextEdit(brainwallet) |
||||
|
seed_text.setReadOnly(True) |
||||
|
seed_text.setMaximumHeight(130) |
||||
|
|
||||
|
msg2 = _("Please write down or memorize these 12 words (order is important).") + " " \ |
||||
|
+ _("This seed will allow you to recover your wallet in case of computer failure.") + " " \ |
||||
|
+ _("Your seed is also displayed as QR code, in case you want to transfer it to a mobile phone.") + "<p>" \ |
||||
|
+ "<b>"+_("WARNING")+":</b> " + _("Never disclose your seed. Never type it on a website.") + "</b><p>" |
||||
|
if imported_keys: |
||||
|
msg2 += "<b>"+_("WARNING")+":</b> " + _("Your wallet contains imported keys. These keys cannot be recovered from seed.") + "</b><p>" |
||||
|
label2 = QLabel(msg2) |
||||
|
label2.setWordWrap(True) |
||||
|
|
||||
|
logo = QLabel() |
||||
|
logo.setPixmap(QPixmap(":icons/seed.png").scaledToWidth(56)) |
||||
|
logo.setMaximumWidth(60) |
||||
|
|
||||
|
qrw = QRCodeWidget(seed) |
||||
|
|
||||
|
ok_button = QPushButton(_("OK")) |
||||
|
ok_button.setDefault(True) |
||||
|
ok_button.clicked.connect(self.accept) |
||||
|
|
||||
|
grid = QGridLayout() |
||||
|
#main_layout.addWidget(logo, 0, 0) |
||||
|
|
||||
|
grid.addWidget(logo, 0, 0) |
||||
|
grid.addWidget(label1, 0, 1) |
||||
|
|
||||
|
grid.addWidget(seed_text, 1, 0, 1, 2) |
||||
|
|
||||
|
grid.addWidget(qrw, 0, 2, 2, 1) |
||||
|
|
||||
|
vbox = QVBoxLayout() |
||||
|
vbox.addLayout(grid) |
||||
|
vbox.addWidget(label2) |
||||
|
|
||||
|
hbox = QHBoxLayout() |
||||
|
hbox.addStretch(1) |
||||
|
hbox.addWidget(ok_button) |
||||
|
vbox.addLayout(hbox) |
||||
|
|
||||
|
self.setLayout(vbox) |
||||
|
self.exec_() |
||||
Loading…
Reference in new issue