gui_qt.py

   1 #!/usr/bin/env python
   2 #
   3 # Electrum - lightweight Bitcoin client
   4 # Copyright (C) 2012 thomasv@gitorious
   5 #
   6 # This program is free software: you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation, either version 3 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program. If not, see <http://www.gnu.org/licenses/>.
  18
  19 import sys, time, datetime, re
  20
  21 # todo: see PySide
  22
  23 from PyQt4.QtGui import *
  24 from PyQt4.QtCore import *
  25 import PyQt4.QtCore as QtCore
  26 import PyQt4.QtGui as QtGui
  27 from interface import DEFAULT_SERVERS
  28
  29 try:
  30     import icons_rc
  31 except:
  32     print "Could not import icons_rp.py"
  33     print "Please generate it with: 'pyrcc4 icons.qrc -o icons_rc.py'"
  34     sys.exit(1)
  35
  36 from wallet import format_satoshis
  37 from decimal import Decimal
  38
  39 import platform
  40 MONOSPACE_FONT = 'Lucida Console' if platform.system() == 'Windows' else 'monospace'
  41
  42
  43 def numbify(entry, is_int = False):
  44     text = unicode(entry.text()).strip()
  45     chars = '0123456789'
  46     if not is_int: chars +='.'
  47     s = ''.join([i for i in text if i in chars])
  48     if not is_int:
  49         if '.' in s:
  50             p = s.find('.')
  51             s = s.replace('.','')
  52             s = s[:p] + '.' + s[p:p+8]
  53         try:
  54             amount = int( Decimal(s) * 100000000 )
  55         except:
  56             amount = None
  57     else:
  58         try:
  59             amount = int( s )
  60         except:
  61             amount = None
  62     entry.setText(s)
  63     return amount
  64
  65
  66 class Timer(QtCore.QThread):
  67     def run(self):
  68         while True:
  69             self.emit(QtCore.SIGNAL('timersignal'))
  70             time.sleep(0.5)
  71
  72 class EnterButton(QPushButton):
  73     def __init__(self, text, func):
  74         QPushButton.__init__(self, text)
  75         self.func = func
  76         self.clicked.connect(func)
  77
  78     def keyPressEvent(self, e):
  79         if e.key() == QtCore.Qt.Key_Return:
  80             apply(self.func,())
  81
  82 class StatusBarButton(QPushButton):
  83     def __init__(self, icon, tooltip, func):
  84         QPushButton.__init__(self, icon, '')
  85         self.setToolTip(tooltip)
  86         self.setFlat(True)
  87         self.setMaximumWidth(25)
  88         self.clicked.connect(func)
  89         self.func = func
  90
  91     def keyPressEvent(self, e):
  92         if e.key() == QtCore.Qt.Key_Return:
  93             apply(self.func,())
  94
  95
  96 class QRCodeWidget(QWidget):
  97
  98     def __init__(self, addr):
  99         super(QRCodeWidget, self).__init__()
 100         self.setGeometry(300, 300, 350, 350)
 101         self.set_addr(addr)
 102
 103     def set_addr(self, addr):
 104         import pyqrnative
 105         self.addr = addr
 106         self.qr = pyqrnative.QRCode(4, pyqrnative.QRErrorCorrectLevel.L)
 107         self.qr.addData(addr)
 108         self.qr.make()
 109
 110     def paintEvent(self, e):
 111         qp = QtGui.QPainter()
 112         qp.begin(self)
 113         boxsize = 7
 114         size = self.qr.getModuleCount()*boxsize
 115         k = self.qr.getModuleCount()
 116         black = QColor(0, 0, 0, 255)
 117         white = QColor(255, 255, 255, 255)
 118         for r in range(k):
 119             for c in range(k):
 120                 if self.qr.isDark(r, c):
 121                     qp.setBrush(black)
 122                     qp.setPen(black)
 123                 else:
 124                     qp.setBrush(white)
 125                     qp.setPen(white)
 126                 qp.drawRect(c*boxsize, r*boxsize, boxsize, boxsize)
 127         qp.end()
 128
 129
 130
 131 def ok_cancel_buttons(dialog):
 132     hbox = QHBoxLayout()
 133     hbox.addStretch(1)
 134     b = QPushButton("OK")
 135     hbox.addWidget(b)
 136     b.clicked.connect(dialog.accept)
 137     b = QPushButton("Cancel")
 138     hbox.addWidget(b)
 139     b.clicked.connect(dialog.reject)
 140     return hbox
 141
 142
 143 class ElectrumWindow(QMainWindow):
 144
 145     def __init__(self, wallet):
 146         QMainWindow.__init__(self)
 147         self.wallet = wallet
 148         self.wallet.gui_callback = self.update_callback
 149
 150         self.funds_error = False
 151
 152         self.tabs = tabs = QTabWidget(self)
 153         tabs.addTab(self.create_history_tab(), 'History')
 154         tabs.addTab(self.create_send_tab(),    'Send')
 155         tabs.addTab(self.create_receive_tab(), 'Receive')
 156         tabs.addTab(self.create_contacts_tab(),'Contacts')
 157         tabs.addTab(self.create_wall_tab(),    'Wall')
 158         tabs.setMinimumSize(600, 400)
 159         tabs.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
 160         self.setCentralWidget(tabs)
 161         self.create_status_bar()
 162         self.setGeometry(100,100,840,400)
 163         self.setWindowTitle( 'Electrum ' + self.wallet.electrum_version )
 164         self.show()
 165
 166         QShortcut(QKeySequence("Ctrl+W"), self, self.close)
 167         QShortcut(QKeySequence("Ctrl+Q"), self, self.close)
 168         QShortcut(QKeySequence("Ctrl+PgUp"), self, lambda: tabs.setCurrentIndex( (tabs.currentIndex() - 1 )%tabs.count() ))
 169         QShortcut(QKeySequence("Ctrl+PgDown"), self, lambda: tabs.setCurrentIndex( (tabs.currentIndex() + 1 )%tabs.count() ))
 170
 171         self.connect(self, QtCore.SIGNAL('updatesignal'), self.update_wallet)
 172
 173
 174     def connect_slots(self, sender):
 175         self.connect(sender, QtCore.SIGNAL('timersignal'), self.check_recipient)
 176         self.previous_payto_e=''
 177
 178     def check_recipient(self):
 179         if self.payto_e.hasFocus():
 180             return
 181         r = unicode( self.payto_e.text() )
 182         if r != self.previous_payto_e:
 183             self.previous_payto_e = r
 184             r = r.strip()
 185             if re.match('^(|([\w\-\.]+)@)((\w[\w\-]+\.)+[\w\-]+)$', r):
 186                 try:
 187                     to_address = self.wallet.get_alias(r, True, self.show_message, self.question)
 188                 except:
 189                     return
 190                 if to_address:
 191                     s = r + ' <' + to_address + '>'
 192                     self.payto_e.setText(s)
 193
 194
 195     def update_callback(self):
 196         self.emit(QtCore.SIGNAL('updatesignal'))
 197
 198     def update_wallet(self):
 199         if self.wallet.interface.is_connected:
 200             if self.wallet.blocks == -1:
 201                 text = "Connecting..."
 202                 icon = QIcon(":icons/status_disconnected.png")
 203             elif self.wallet.blocks == 0:
 204                 text = "Server not ready"
 205                 icon = QIcon(":icons/status_disconnected.png")
 206             elif not self.wallet.up_to_date:
 207                 text = "Synchronizing..."
 208                 icon = QIcon(":icons/status_waiting.png")
 209             else:
 210                 c, u = self.wallet.get_balance()
 211                 text =  "Balance: %s "%( format_satoshis(c,False,self.wallet.num_zeros) )
 212                 if u: text +=  "[%s unconfirmed]"%( format_satoshis(u,True,self.wallet.num_zeros).strip() )
 213                 icon = QIcon(":icons/status_connected.png")
 214         else:
 215             text = "Not connected"
 216             icon = QIcon(":icons/status_disconnected.png")
 217
 218         if self.funds_error:
 219             text = "Not enough funds"
 220
 221         self.statusBar().showMessage(text)
 222         self.status_button.setIcon( icon )
 223
 224         if self.wallet.up_to_date:
 225             self.textbox.setText( self.wallet.banner )
 226             self.update_history_tab()
 227             self.update_receive_tab()
 228             self.update_contacts_tab()
 229
 230
 231     def create_history_tab(self):
 232         self.history_list = w = QTreeWidget(self)
 233         #print w.getContentsMargins()
 234         w.setColumnCount(5)
 235         w.setColumnWidth(0, 40)
 236         w.setColumnWidth(1, 140)
 237         w.setColumnWidth(2, 350)
 238         w.setColumnWidth(3, 140)
 239         w.setColumnWidth(4, 140)
 240         w.setHeaderLabels( [ '', 'Date', 'Description', 'Amount', 'Balance'] )
 241         self.connect(w, SIGNAL('itemActivated(QTreeWidgetItem*, int)'), self.tx_details)
 242         self.connect(w, SIGNAL('itemDoubleClicked(QTreeWidgetItem*, int)'), self.tx_label_clicked)
 243         self.connect(w, SIGNAL('itemChanged(QTreeWidgetItem*, int)'), self.tx_label_changed)
 244         return w
 245
 246     def tx_details(self, item, column):
 247         tx_hash = str(item.toolTip(0))
 248         tx = self.wallet.tx_history.get(tx_hash)
 249
 250         if tx['height']:
 251             conf = self.wallet.blocks - tx['height'] + 1
 252             time_str = datetime.datetime.fromtimestamp( tx['timestamp']).isoformat(' ')[:-3]
 253         else:
 254             conf = 0
 255             time_str = 'pending'
 256
 257         tx_details = "Transaction Details:\n\n" \
 258             + "Transaction ID:\n" + tx_hash + "\n\n" \
 259             + "Status: %d confirmations\n\n"%conf  \
 260             + "Date: %s\n\n"%time_str \
 261             + "Inputs:\n-"+ '\n-'.join(tx['inputs']) + "\n\n" \
 262             + "Outputs:\n-"+ '\n-'.join(tx['outputs'])
 263
 264         r = self.wallet.receipts.get(tx_hash)
 265         if r:
 266             tx_details += "\n_______________________________________" \
 267                 + '\n\nSigned URI: ' + r[2] \
 268                 + "\n\nSigned by: " + r[0] \
 269                 + '\n\nSignature: ' + r[1]
 270
 271         QMessageBox.information(self, 'Details', tx_details, 'OK')
 272
 273
 274     def tx_label_clicked(self, item, column):
 275         if column==2 and item.isSelected():
 276             tx_hash = str(item.toolTip(0))
 277             self.is_edit=True
 278             #if not self.wallet.labels.get(tx_hash): item.setText(2,'')
 279             item.setFlags(Qt.ItemIsEditable|Qt.ItemIsSelectable | Qt.ItemIsUserCheckable | Qt.ItemIsEnabled | Qt.ItemIsDragEnabled)
 280             self.history_list.editItem( item, column )
 281             item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsUserCheckable | Qt.ItemIsEnabled | Qt.ItemIsDragEnabled)
 282             self.is_edit=False
 283
 284     def tx_label_changed(self, item, column):
 285         if self.is_edit:
 286             return
 287         self.is_edit=True
 288         tx_hash = str(item.toolTip(0))
 289         tx = self.wallet.tx_history.get(tx_hash)
 290         s = self.wallet.labels.get(tx_hash)
 291         text = unicode( item.text(2) )
 292         if text:
 293             self.wallet.labels[tx_hash] = text
 294             item.setForeground(2, QBrush(QColor('black')))
 295         else:
 296             if s: self.wallet.labels.pop(tx_hash)
 297             text = tx['default_label']
 298             item.setText(2, text)
 299             item.setForeground(2, QBrush(QColor('gray')))
 300         self.is_edit=False
 301
 302     def address_label_clicked(self, item, column, l):
 303         if column==1 and item.isSelected():
 304             addr = unicode( item.text(0) )
 305             if addr in map(lambda x:x[1], self.wallet.aliases.values()):
 306                 return
 307             item.setFlags(Qt.ItemIsEditable|Qt.ItemIsSelectable | Qt.ItemIsUserCheckable | Qt.ItemIsEnabled | Qt.ItemIsDragEnabled)
 308             l.editItem( item, column )
 309             item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsUserCheckable | Qt.ItemIsEnabled | Qt.ItemIsDragEnabled)
 310
 311     def address_label_changed(self, item, column, l):
 312         addr = unicode( item.text(0) )
 313         text = unicode( item.text(1) )
 314         if text:
 315             self.wallet.labels[addr] = text
 316         else:
 317             s = self.wallet.labels.get(addr)
 318             if s: self.wallet.labels.pop(addr)
 319         self.update_history_tab()
 320
 321     def update_history_tab(self):
 322         self.history_list.clear()
 323         balance = 0
 324         for tx in self.wallet.get_tx_history():
 325             tx_hash = tx['tx_hash']
 326             if tx['height']:
 327                 conf = self.wallet.blocks - tx['height'] + 1
 328                 time_str = datetime.datetime.fromtimestamp( tx['timestamp']).isoformat(' ')[:-3]
 329                 icon = QIcon(":icons/confirmed.png")
 330             else:
 331                 conf = 0
 332                 time_str = 'pending'
 333                 icon = QIcon(":icons/unconfirmed.png")
 334             v = tx['value']
 335             balance += v
 336             label = self.wallet.labels.get(tx_hash)
 337             is_default_label = (label == '') or (label is None)
 338             if is_default_label: label = tx['default_label']
 339
 340             item = QTreeWidgetItem( [ '', time_str, label, format_satoshis(v,True,self.wallet.num_zeros), format_satoshis(balance,False,self.wallet.num_zeros)] )
 341             item.setFont(2, QFont(MONOSPACE_FONT))
 342             item.setFont(3, QFont(MONOSPACE_FONT))
 343             item.setFont(4, QFont(MONOSPACE_FONT))
 344             item.setToolTip(0, tx_hash)
 345             if is_default_label:
 346                 item.setForeground(2, QBrush(QColor('grey')))
 347
 348             item.setIcon(0, icon)
 349             self.history_list.insertTopLevelItem(0,item)
 350
 351
 352     def create_send_tab(self):
 353         w = QWidget()
 354
 355         grid = QGridLayout()
 356         grid.setSpacing(8)
 357         grid.setColumnMinimumWidth(3,300)
 358         grid.setColumnStretch(4,1)
 359
 360         self.payto_e = QLineEdit()
 361         grid.addWidget(QLabel('Pay to'), 1, 0)
 362         grid.addWidget(self.payto_e, 1, 1, 1, 3)
 363
 364         self.message_e = QLineEdit()
 365         grid.addWidget(QLabel('Description'), 2, 0)
 366         grid.addWidget(self.message_e, 2, 1, 1, 3)
 367
 368         self.amount_e = QLineEdit()
 369         grid.addWidget(QLabel('Amount'), 3, 0)
 370         grid.addWidget(self.amount_e, 3, 1, 1, 2)
 371
 372         self.fee_e = QLineEdit()
 373         grid.addWidget(QLabel('Fee'), 4, 0)
 374         grid.addWidget(self.fee_e, 4, 1, 1, 2)
 375
 376         b = EnterButton("Send", self.do_send)
 377         grid.addWidget(b, 5, 1)
 378
 379         b = EnterButton("Clear",self.do_clear)
 380         grid.addWidget(b, 5, 2)
 381
 382         self.payto_sig = QLabel('')
 383         grid.addWidget(self.payto_sig, 6, 0, 1, 4)
 384
 385         w.setLayout(grid)
 386         w.show()
 387
 388         w2 = QWidget()
 389         vbox = QVBoxLayout()
 390         vbox.addWidget(w)
 391         vbox.addStretch(1)
 392         w2.setLayout(vbox)
 393
 394         def entry_changed( is_fee ):
 395             self.funds_error = False
 396             amount = numbify(self.amount_e)
 397             fee = numbify(self.fee_e)
 398             if not is_fee: fee = None
 399             if amount is None:
 400                 return
 401             inputs, total, fee = self.wallet.choose_tx_inputs( amount, fee )
 402             if not is_fee:
 403                 self.fee_e.setText( str( Decimal( fee ) / 100000000 ) )
 404             if inputs:
 405                 palette = QPalette()
 406                 palette.setColor(self.amount_e.foregroundRole(), QColor('black'))
 407             else:
 408                 palette = QPalette()
 409                 palette.setColor(self.amount_e.foregroundRole(), QColor('red'))
 410                 self.funds_error = True
 411             self.amount_e.setPalette(palette)
 412             self.fee_e.setPalette(palette)
 413
 414         self.amount_e.textChanged.connect(lambda: entry_changed(False) )
 415         self.fee_e.textChanged.connect(lambda: entry_changed(True) )
 416
 417         return w2
 418
 419     def do_send(self):
 420
 421         label = unicode( self.message_e.text() )
 422         r = unicode( self.payto_e.text() )
 423         r = r.strip()
 424
 425         m1 = re.match('^(|([\w\-\.]+)@)((\w[\w\-]+\.)+[\w\-]+)$', r)
 426         m2 = re.match('(|([\w\-\.]+)@)((\w[\w\-]+\.)+[\w\-]+) \<([1-9A-HJ-NP-Za-km-z]{26,})\>', r)
 427
 428         if m1:
 429             to_address = self.wallet.get_alias(r, True, self.show_message, self.question)
 430             if not to_address:
 431                 return
 432         elif m2:
 433             to_address = m2.group(5)
 434         else:
 435             to_address = r
 436
 437         if not self.wallet.is_valid(to_address):
 438             QMessageBox.warning(self, 'Error', 'Invalid Bitcoin Address:\n'+to_address, 'OK')
 439             return
 440
 441         try:
 442             amount = int( Decimal( unicode( self.amount_e.text())) * 100000000 )
 443         except:
 444             QMessageBox.warning(self, 'Error', 'Invalid Amount', 'OK')
 445             return
 446         try:
 447             fee = int( Decimal( unicode( self.fee_e.text())) * 100000000 )
 448         except:
 449             QMessageBox.warning(self, 'Error', 'Invalid Fee', 'OK')
 450             return
 451
 452         if self.wallet.use_encryption:
 453             password = self.password_dialog()
 454             if not password:
 455                 return
 456         else:
 457             password = None
 458
 459         try:
 460             tx = self.wallet.mktx( to_address, amount, label, password, fee )
 461         except BaseException, e:
 462             self.show_message(e.message)
 463             return
 464
 465         status, msg = self.wallet.sendtx( tx )
 466         if status:
 467             QMessageBox.information(self, '', 'Payment sent.\n'+msg, 'OK')
 468             self.do_clear()
 469             self.update_contacts_tab()
 470         else:
 471             QMessageBox.warning(self, 'Error', msg, 'OK')
 472
 473
 474     def set_url(self, url):
 475         payto, amount, label, message, signature, identity, url = self.wallet.parse_url(url, self.show_message, self.question)
 476         self.tabs.setCurrentIndex(1)
 477         self.payto_e.setText(payto)
 478         self.message_e.setText(message)
 479         self.amount_e.setText(amount)
 480         if identity:
 481             self.set_frozen(self.payto_e,True)
 482             self.set_frozen(self.amount_e,True)
 483             self.set_frozen(self.message_e,True)
 484             self.payto_sig.setText( '      The bitcoin URI was signed by ' + identity )
 485         else:
 486             self.payto_sig.setVisible(False)
 487
 488     def do_clear(self):
 489         self.payto_sig.setVisible(False)
 490         for e in [self.payto_e, self.message_e, self.amount_e, self.fee_e]:
 491             e.setText('')
 492             self.set_frozen(e,False)
 493
 494     def set_frozen(self,entry,frozen):
 495         if frozen:
 496             entry.setReadOnly(True)
 497             entry.setFrame(False)
 498             palette = QPalette()
 499             palette.setColor(entry.backgroundRole(), QColor('lightgray'))
 500             entry.setPalette(palette)
 501         else:
 502             entry.setReadOnly(False)
 503             entry.setFrame(True)
 504             palette = QPalette()
 505             palette.setColor(entry.backgroundRole(), QColor('white'))
 506             entry.setPalette(palette)
 507
 508
 509     def make_address_list(self, is_recv):
 510
 511         l = QTreeWidget(self)
 512         l.setColumnCount(3)
 513         l.setColumnWidth(0, 350)
 514         l.setColumnWidth(1, 330)
 515         l.setColumnWidth(2, 20)
 516         l.setHeaderLabels( ['Address', 'Label','Tx'])
 517
 518         vbox = QVBoxLayout()
 519         vbox.setMargin(0)
 520         vbox.setSpacing(0)
 521         vbox.addWidget(l)
 522
 523         hbox = QHBoxLayout()
 524         hbox.setMargin(0)
 525         hbox.setSpacing(0)
 526
 527         def get_addr(l):
 528             i = l.currentItem()
 529             if not i: return
 530             addr = unicode( i.text(0) )
 531             return addr
 532
 533         qrButton = EnterButton("QR",lambda: self.show_address_qrcode(get_addr(l)))
 534
 535         def copy2clipboard(addr):
 536             self.app.clipboard().setText(addr)
 537         copyButton = EnterButton("Copy to Clipboard", lambda: copy2clipboard(get_addr(l)))
 538         hbox.addWidget(qrButton)
 539         hbox.addWidget(copyButton)
 540         if not is_recv:
 541             addButton = EnterButton("New", self.newaddress_dialog)
 542             hbox.addWidget(addButton)
 543             def payto(addr):
 544                 if not addr:return
 545                 self.tabs.setCurrentIndex(1)
 546                 self.payto_e.setText(addr)
 547                 self.amount_e.setFocus()
 548             paytoButton = EnterButton('Pay to', lambda: payto(get_addr(l)))
 549             hbox.addWidget(paytoButton)
 550         hbox.addStretch(1)
 551         buttons = QWidget()
 552         buttons.setLayout(hbox)
 553         vbox.addWidget(buttons)
 554
 555         w = QWidget()
 556         w.setLayout(vbox)
 557         return w, l
 558
 559     def create_receive_tab(self):
 560         w, l = self.make_address_list(True)
 561         self.connect(l, SIGNAL('itemDoubleClicked(QTreeWidgetItem*, int)'), lambda a, b: self.address_label_clicked(a,b,l))
 562         self.connect(l, SIGNAL('itemChanged(QTreeWidgetItem*, int)'), lambda a,b: self.address_label_changed(a,b,l))
 563         self.receive_list = l
 564         return w
 565
 566     def create_contacts_tab(self):
 567         w, l = self.make_address_list(False)
 568         self.connect(l, SIGNAL('itemDoubleClicked(QTreeWidgetItem*, int)'), lambda a, b: self.address_label_clicked(a,b,l))
 569         self.connect(l, SIGNAL('itemChanged(QTreeWidgetItem*, int)'), lambda a,b: self.address_label_changed(a,b,l))
 570         self.connect(l, SIGNAL('itemActivated(QTreeWidgetItem*, int)'), self.show_contact_details)
 571         self.contacts_list = l
 572         return w
 573
 574     def update_receive_tab(self):
 575         self.receive_list.clear()
 576         for address in self.wallet.all_addresses():
 577             if self.wallet.is_change(address):continue
 578             label = self.wallet.labels.get(address,'')
 579             n = 0
 580             h = self.wallet.history.get(address,[])
 581             for item in h:
 582                 if not item['is_input'] : n=n+1
 583             tx = "None" if n==0 else "%d"%n
 584             item = QTreeWidgetItem( [ address, label, tx] )
 585             item.setFont(0, QFont(MONOSPACE_FONT))
 586             self.receive_list.addTopLevelItem(item)
 587
 588     def show_contact_details(self, item, column):
 589         m = unicode(item.text(0))
 590         a = self.wallet.aliases.get(m)
 591         if a:
 592             if a[0] in self.wallet.authorities.keys():
 593                 s = self.wallet.authorities.get(a[0])
 594             else:
 595                 s = "self-signed"
 596             msg = 'Alias: '+ m + '\nTarget address: '+ a[1] + '\n\nSigned by: ' + s + '\nSigning address:' + a[0]
 597             QMessageBox.information(self, 'Alias', msg, 'OK')
 598
 599     def update_contacts_tab(self):
 600         self.contacts_list.clear()
 601         for alias, v in self.wallet.aliases.items():
 602             s, target = v
 603             item = QTreeWidgetItem( [ target, alias, '-'] )
 604             self.contacts_list.addTopLevelItem(item)
 605
 606         for address in self.wallet.addressbook:
 607             label = self.wallet.labels.get(address,'')
 608             n = 0
 609             for item in self.wallet.tx_history.values():
 610                 if address in item['outputs'] : n=n+1
 611             tx = "None" if n==0 else "%d"%n
 612             item = QTreeWidgetItem( [ address, label, tx] )
 613             item.setFont(0, QFont(MONOSPACE_FONT))
 614             self.contacts_list.addTopLevelItem(item)
 615
 616
 617     def create_wall_tab(self):
 618         self.textbox = textbox = QTextEdit(self)
 619         textbox.setFont(QFont(MONOSPACE_FONT))
 620         textbox.setReadOnly(True)
 621         return textbox
 622
 623     def create_status_bar(self):
 624         sb = QStatusBar()
 625         sb.setFixedHeight(35)
 626         sb.addPermanentWidget( StatusBarButton( QIcon(":icons/lock.png"), "Password", lambda: self.change_password_dialog(self.wallet, self) ) )
 627         sb.addPermanentWidget( StatusBarButton( QIcon(":icons/preferences.png"), "Preferences", self.settings_dialog ) )
 628         sb.addPermanentWidget( StatusBarButton( QIcon(":icons/seed.png"), "Seed", lambda: self.show_seed_dialog(self.wallet, self) ) )
 629         self.status_button = StatusBarButton( QIcon(":icons/status_disconnected.png"), "Network", lambda: self.network_dialog(self.wallet, self) )
 630         sb.addPermanentWidget( self.status_button )
 631         self.setStatusBar(sb)
 632
 633     def newaddress_dialog(self):
 634         text, ok = QInputDialog.getText(self, 'New Contact', 'Address:')
 635         address = unicode(text)
 636         if ok:
 637             if self.wallet.is_valid(address):
 638                 self.wallet.addressbook.append(address)
 639                 self.wallet.save()
 640                 self.update_contacts_tab()
 641             else:
 642                 QMessageBox.warning(self, 'Error', 'Invalid Address', 'OK')
 643
 644     @staticmethod
 645     def show_seed_dialog(wallet, parent=None):
 646         import mnemonic
 647         if wallet.use_encryption:
 648             password = parent.password_dialog()
 649             if not password: return
 650         else:
 651             password = None
 652
 653         try:
 654             seed = wallet.pw_decode( wallet.seed, password)
 655         except:
 656             QMessageBox.warning(parent, 'Error', 'Invalid Password', 'OK')
 657             return
 658
 659         msg = "Your wallet generation seed is:\n\n" + seed \
 660               + "\n\nPlease keep it in a safe place; if you lose it, you will not be able to restore your wallet.\n\n" \
 661               + "Equivalently, your wallet seed can be stored and recovered with the following mnemonic code:\n\n\"" \
 662               + ' '.join(mnemonic.mn_encode(seed)) + "\""
 663
 664         QMessageBox.information(parent, 'Seed', msg, 'OK')
 665         if parent: ElectrumWindow.show_seed_qrcode(seed)
 666
 667     @staticmethod
 668     def show_seed_qrcode(seed):
 669         if not seed: return
 670         d = QDialog(None)
 671         d.setModal(1)
 672         d.setWindowTitle(seed)
 673         d.setMinimumSize(270, 300)
 674         vbox = QVBoxLayout()
 675         vbox.addWidget(QRCodeWidget(seed))
 676         hbox = QHBoxLayout()
 677         hbox.addStretch(1)
 678         b = QPushButton("OK")
 679         hbox.addWidget(b)
 680         b.clicked.connect(d.accept)
 681
 682         vbox.addLayout(hbox)
 683         d.setLayout(vbox)
 684         d.exec_()
 685
 686     def show_address_qrcode(self,address):
 687         if not address: return
 688         d = QDialog(None)
 689         d.setModal(1)
 690         d.setWindowTitle(address)
 691         d.setMinimumSize(270, 350)
 692         vbox = QVBoxLayout()
 693         qrw = QRCodeWidget(address)
 694         vbox.addWidget(qrw)
 695
 696         hbox = QHBoxLayout()
 697         amount_e = QLineEdit()
 698         hbox.addWidget(QLabel('Amount'))
 699         hbox.addWidget(amount_e)
 700         vbox.addLayout(hbox)
 701
 702         #hbox = QHBoxLayout()
 703         #label_e = QLineEdit()
 704         #hbox.addWidget(QLabel('Label'))
 705         #hbox.addWidget(label_e)
 706         #vbox.addLayout(hbox)
 707
 708         def amount_changed():
 709             amount = numbify(amount_e)
 710             #label = str( label_e.getText() )
 711             if amount is not None:
 712                 qrw.set_addr('bitcoin:%s?amount=%s'%(address,str( Decimal(amount) /100000000)))
 713             else:
 714                 qrw.set_addr( address )
 715             qrw.repaint()
 716
 717         def do_save():
 718             import bmp
 719             bmp.save_qrcode(qrw.qr, "qrcode.bmp")
 720             self.show_message("QR code saved to file 'qrcode.bmp'")
 721
 722         amount_e.textChanged.connect( amount_changed )
 723
 724         hbox = QHBoxLayout()
 725         hbox.addStretch(1)
 726         b = QPushButton("Save")
 727         b.clicked.connect(do_save)
 728         hbox.addWidget(b)
 729         b = QPushButton("Close")
 730         hbox.addWidget(b)
 731         b.clicked.connect(d.accept)
 732
 733         vbox.addLayout(hbox)
 734         d.setLayout(vbox)
 735         d.exec_()
 736
 737     def question(self, msg):
 738         return QMessageBox.question(self, 'Message', msg, QMessageBox.Yes | QMessageBox.No, QMessageBox.No) == QMessageBox.Yes
 739
 740     def show_message(self, msg):
 741         QMessageBox.information(self, 'Message', msg, 'OK')
 742
 743     def password_dialog(self ):
 744         d = QDialog(self)
 745         d.setModal(1)
 746
 747         pw = QLineEdit()
 748         pw.setEchoMode(2)
 749
 750         vbox = QVBoxLayout()
 751         msg = 'Please enter your password'
 752         vbox.addWidget(QLabel(msg))
 753
 754         grid = QGridLayout()
 755         grid.setSpacing(8)
 756         grid.addWidget(QLabel('Password'), 1, 0)
 757         grid.addWidget(pw, 1, 1)
 758         vbox.addLayout(grid)
 759
 760         vbox.addLayout(ok_cancel_buttons(d))
 761         d.setLayout(vbox)
 762
 763         if not d.exec_(): return
 764         return unicode(pw.text())
 765
 766     @staticmethod
 767     def change_password_dialog( wallet, parent=None ):
 768         d = QDialog(parent)
 769         d.setModal(1)
 770
 771         pw = QLineEdit()
 772         pw.setEchoMode(2)
 773         new_pw = QLineEdit()
 774         new_pw.setEchoMode(2)
 775         conf_pw = QLineEdit()
 776         conf_pw.setEchoMode(2)
 777
 778         vbox = QVBoxLayout()
 779         if parent:
 780             msg = 'Your wallet is encrypted. Use this dialog to change your password.\nTo disable wallet encryption, enter an empty new password.' if wallet.use_encryption else 'Your wallet keys are not encrypted'
 781         else:
 782             msg = "Please choose a password to encrypt your wallet keys.\nLeave these fields empty if you want to disable encryption."
 783         vbox.addWidget(QLabel(msg))
 784
 785         grid = QGridLayout()
 786         grid.setSpacing(8)
 787
 788         if wallet.use_encryption:
 789             grid.addWidget(QLabel('Password'), 1, 0)
 790             grid.addWidget(pw, 1, 1)
 791
 792         grid.addWidget(QLabel('New Password'), 2, 0)
 793         grid.addWidget(new_pw, 2, 1)
 794
 795         grid.addWidget(QLabel('Confirm Password'), 3, 0)
 796         grid.addWidget(conf_pw, 3, 1)
 797         vbox.addLayout(grid)
 798
 799         vbox.addLayout(ok_cancel_buttons(d))
 800         d.setLayout(vbox)
 801
 802         if not d.exec_(): return
 803
 804         password = unicode(pw.text()) if wallet.use_encryption else None
 805         new_password = unicode(new_pw.text())
 806         new_password2 = unicode(conf_pw.text())
 807
 808         try:
 809             seed = wallet.pw_decode( wallet.seed, password)
 810         except:
 811             QMessageBox.warning(parent, 'Error', 'Incorrect Password', 'OK')
 812             return
 813
 814         if new_password != new_password2:
 815             QMessageBox.warning(parent, 'Error', 'Passwords do not match', 'OK')
 816             return
 817
 818         wallet.update_password(seed, new_password)
 819
 820     @staticmethod
 821     def seed_dialog(wallet, parent=None):
 822         d = QDialog(parent)
 823         d.setModal(1)
 824
 825         vbox = QVBoxLayout()
 826         msg = "Please enter your wallet seed or the corresponding mnemonic list of words, and the gap limit of your wallet."
 827         vbox.addWidget(QLabel(msg))
 828
 829         grid = QGridLayout()
 830         grid.setSpacing(8)
 831
 832         seed_e = QLineEdit()
 833         grid.addWidget(QLabel('Seed or mnemonic'), 1, 0)
 834         grid.addWidget(seed_e, 1, 1)
 835
 836         gap_e = QLineEdit()
 837         gap_e.setText("5")
 838         grid.addWidget(QLabel('Gap limit'), 2, 0)
 839         grid.addWidget(gap_e, 2, 1)
 840         gap_e.textChanged.connect(lambda: numbify(gap_e,True))
 841         vbox.addLayout(grid)
 842
 843         vbox.addLayout(ok_cancel_buttons(d))
 844         d.setLayout(vbox)
 845
 846         if not d.exec_(): return
 847
 848         try:
 849             gap = int(unicode(gap_e.text()))
 850         except:
 851             QMessageBox.warning(None, 'Error', 'error', 'OK')
 852             sys.exit(0)
 853
 854         try:
 855             seed = unicode(seed_e.text())
 856             seed.decode('hex')
 857         except:
 858             import mnemonic
 859             print "not hex, trying decode"
 860             try:
 861                 seed = mnemonic.mn_decode( seed.split(' ') )
 862             except:
 863                 QMessageBox.warning(None, 'Error', 'I cannot decode this', 'OK')
 864                 sys.exit(0)
 865         if not seed:
 866             QMessageBox.warning(None, 'Error', 'no seed', 'OK')
 867             sys.exit(0)
 868
 869         wallet.seed = str(seed)
 870         #print repr(wallet.seed)
 871         wallet.gap_limit = gap
 872         return True
 873
 874
 875     def settings_dialog(self):
 876         d = QDialog(self)
 877         d.setModal(1)
 878
 879         vbox = QVBoxLayout()
 880
 881         msg = 'Here are the settings of your wallet.'
 882         vbox.addWidget(QLabel(msg))
 883
 884         grid = QGridLayout()
 885         grid.setSpacing(8)
 886         vbox.addLayout(grid)
 887
 888         fee_e = QLineEdit()
 889         fee_e.setText("%s"% str( Decimal( self.wallet.fee)/100000000 ) )
 890         grid.addWidget(QLabel('Fee per tx. input'), 2, 0)
 891         grid.addWidget(fee_e, 2, 1)
 892         fee_e.textChanged.connect(lambda: numbify(fee_e,False))
 893
 894         nz_e = QLineEdit()
 895         nz_e.setText("%d"% self.wallet.num_zeros)
 896         grid.addWidget(QLabel('Zeros displayed after decimal point'), 3, 0)
 897         grid.addWidget(nz_e, 3, 1)
 898         nz_e.textChanged.connect(lambda: numbify(nz_e,True))
 899
 900         vbox.addLayout(ok_cancel_buttons(d))
 901         d.setLayout(vbox)
 902
 903         if not d.exec_(): return
 904
 905         fee = unicode(fee_e.text())
 906         try:
 907             fee = int( 100000000 * Decimal(fee) )
 908         except:
 909             QMessageBox.warning(self, 'Error', 'Invalid value:%s'%fee, 'OK')
 910             return
 911
 912         if self.wallet.fee != fee:
 913             self.wallet.fee = fee
 914             self.wallet.save()
 915
 916         nz = unicode(nz_e.text())
 917         try:
 918             nz = int( nz )
 919             if nz>8: nz=8
 920         except:
 921             QMessageBox.warning(self, 'Error', 'Invalid value:%s'%nz, 'OK')
 922             return
 923
 924         if self.wallet.num_zeros != nz:
 925             self.wallet.num_zeros = nz
 926             self.update_history_tab()
 927             self.wallet.save()
 928
 929     @staticmethod
 930     def network_dialog(wallet, parent=None):
 931         interface = wallet.interface
 932         if parent:
 933             if interface.is_connected:
 934                 status = "Connected to %s:%d\n%d blocks"%(interface.host, interface.port, wallet.blocks)
 935             else:
 936                 status = "Not connected"
 937             server = wallet.server
 938         else:
 939             import random
 940             status = "Please choose a server."
 941             server = random.choice( DEFAULT_SERVERS )
 942
 943         if not wallet.interface.servers:
 944             servers_list = []
 945             for x in DEFAULT_SERVERS:
 946                 h,port,protocol = x.split(':')
 947                 servers_list.append( (h,[(protocol,port)] ) )
 948         else:
 949             servers_list = wallet.interface.servers
 950
 951         plist = {}
 952         for item in servers_list:
 953             host, pp = item
 954             z = {}
 955             for item2 in pp:
 956                 protocol, port = item2
 957                 z[protocol] = port
 958             plist[host] = z
 959
 960         d = QDialog(parent)
 961         d.setModal(1)
 962         d.setWindowTitle('Server')
 963         d.setMinimumSize(375, 20)
 964
 965         vbox = QVBoxLayout()
 966         vbox.setSpacing(20)
 967
 968         hbox = QHBoxLayout()
 969         l = QLabel()
 970         l.setPixmap(QPixmap(":icons/network.png"))
 971         hbox.addWidget(l)
 972         hbox.addWidget(QLabel(status))
 973
 974         vbox.addLayout(hbox)
 975
 976         hbox = QHBoxLayout()
 977         host_line = QLineEdit()
 978         host_line.setText(server)
 979         hbox.addWidget(QLabel('Connect to:'))
 980         hbox.addWidget(host_line)
 981         vbox.addLayout(hbox)
 982
 983         hbox = QHBoxLayout()
 984
 985         buttonGroup = QGroupBox("protocol")
 986         radio1 = QRadioButton("tcp", buttonGroup)
 987         radio2 = QRadioButton("http", buttonGroup)
 988
 989         def current_line():
 990             return unicode(host_line.text()).split(':')
 991
 992         def set_button(protocol):
 993             if protocol == 't':
 994                 radio1.setChecked(1)
 995             elif protocol == 'h':
 996                 radio2.setChecked(1)
 997
 998         def set_protocol(protocol):
 999             host = current_line()[0]
1000             pp = plist[host]
1001             if protocol not in pp.keys():
1002                 protocol = pp.keys()[0]
1003                 set_button(protocol)
1004             port = pp[protocol]
1005             host_line.setText( host + ':' + port + ':' + protocol)
1006
1007         radio1.clicked.connect(lambda x: set_protocol('t') )
1008         radio2.clicked.connect(lambda x: set_protocol('h') )
1009
1010         set_button(current_line()[2])
1011
1012         hbox.addWidget(QLabel('Protocol:'))
1013         hbox.addWidget(radio1)
1014         hbox.addWidget(radio2)
1015
1016         vbox.addLayout(hbox)
1017
1018         if wallet.interface.servers:
1019             label = 'Active Servers'
1020         else:
1021             label = 'Default Servers'
1022
1023         servers_list_widget = QTreeWidget(parent)
1024         servers_list_widget.setHeaderLabels( [ label ] )
1025         servers_list_widget.setMaximumHeight(150)
1026         for host in plist.keys():
1027             servers_list_widget.addTopLevelItem(QTreeWidgetItem( [ host ] ))
1028
1029         def do_set_line(x):
1030             host = unicode(x.text(0))
1031             pp = plist[host]
1032             if 't' in pp.keys():
1033                 protocol = 't'
1034             else:
1035                 protocol = pp.keys()[0]
1036             port = pp[protocol]
1037             host_line.setText( host + ':' + port + ':' + protocol)
1038             set_button(protocol)
1039
1040         servers_list_widget.connect(servers_list_widget, SIGNAL('itemClicked(QTreeWidgetItem*, int)'), do_set_line)
1041         vbox.addWidget(servers_list_widget)
1042
1043         vbox.addLayout(ok_cancel_buttons(d))
1044         d.setLayout(vbox)
1045
1046         if not d.exec_(): return
1047         server = unicode( host_line.text() )
1048
1049         try:
1050             wallet.set_server(server)
1051         except:
1052             QMessageBox.information(None, 'Error', 'error', 'OK')
1053             if parent == None:
1054                 sys.exit(1)
1055             else:
1056                 return
1057
1058         return True
1059
1060
1061
1062 class ElectrumGui():
1063
1064     def __init__(self, wallet):
1065         self.wallet = wallet
1066         self.app = QApplication(sys.argv)
1067
1068     def waiting_dialog(self):
1069
1070         s = Timer()
1071         s.start()
1072         w = QDialog()
1073         w.resize(200, 70)
1074         w.setWindowTitle('Electrum')
1075         l = QLabel('')
1076         vbox = QVBoxLayout()
1077         vbox.addWidget(l)
1078         w.setLayout(vbox)
1079         w.show()
1080         def f():
1081             if self.wallet.up_to_date: w.close()
1082             else:
1083                 l.setText("Please wait...\nGenerating addresses: %d"%len(self.wallet.all_addresses()))
1084                 pass
1085         w.connect(s, QtCore.SIGNAL('timersignal'), f)
1086         self.wallet.interface.poke()
1087         w.exec_()
1088         w.destroy()
1089
1090
1091     def restore_or_create(self):
1092
1093         msg = "Wallet file not found.\nDo you want to create a new wallet,\n or to restore an existing one?"
1094         r = QMessageBox.question(None, 'Message', msg, 'create', 'restore', 'cancel', 0, 2)
1095         if r==2: return False
1096
1097         is_recovery = (r==1)
1098         wallet = self.wallet
1099         # ask for the server.
1100         if not ElectrumWindow.network_dialog( wallet, parent=None ): return False
1101
1102         if not is_recovery:
1103             wallet.new_seed(None)
1104             wallet.init_mpk( wallet.seed )
1105             wallet.up_to_date_event.clear()
1106             wallet.up_to_date = False
1107             self.waiting_dialog()
1108             # run a dialog indicating the seed, ask the user to remember it
1109             ElectrumWindow.show_seed_dialog(wallet)
1110             #ask for password
1111             ElectrumWindow.change_password_dialog(wallet)
1112         else:
1113             # ask for seed and gap.
1114             if not ElectrumWindow.seed_dialog( wallet ): return False
1115             wallet.init_mpk( wallet.seed )
1116             wallet.up_to_date_event.clear()
1117             wallet.up_to_date = False
1118             self.waiting_dialog()
1119             if wallet.is_found():
1120                 # history and addressbook
1121                 wallet.update_tx_history()
1122                 wallet.fill_addressbook()
1123                 print "recovery successful"
1124                 wallet.save()
1125             else:
1126                 QMessageBox.information(None, 'Message', "No transactions found for this seed", 'OK')
1127
1128         wallet.save()
1129         return True
1130
1131     def main(self,url):
1132         s = Timer()
1133         s.start()
1134         w = ElectrumWindow(self.wallet)
1135         if url: w.set_url(url)
1136         w.app = self.app
1137         w.connect_slots(s)
1138         w.update_wallet()
1139
1140         self.app.exec_()