lib/gui_lite.py

   1 import sys
   2
   3 # Let's do some dep checking and handle missing ones gracefully
   4 try:
   5     from PyQt4.QtCore import *
   6     from PyQt4.QtGui import *
   7 except ImportError:
   8     print "You need to have PyQT installed to run Electrum in graphical mode."
   9     print "If you have pip installed try 'sudo pip install pyqt' if you are on Debian/Ubuntu try 'sudo apt-get install python-qt4'."
  10     sys.exit(0)
  11
  12
  13
  14
  15 from decimal import Decimal as D
  16 from interface import DEFAULT_SERVERS
  17 from util import get_resource_path as rsrc
  18 from i18n import _
  19 import decimal
  20 import exchange_rate
  21 import os.path
  22 import random
  23 import re
  24 import time
  25 import wallet
  26 import webbrowser
  27 import history_widget
  28 import util
  29
  30 import gui_qt
  31 import shutil
  32
  33 bitcoin = lambda v: v * 100000000
  34
  35 def IconButton(filename, parent=None):
  36     pixmap = QPixmap(filename)
  37     icon = QIcon(pixmap)
  38     return QPushButton(icon, "", parent)
  39
  40 class Timer(QThread):
  41     def run(self):
  42         while True:
  43             self.emit(SIGNAL('timersignal'))
  44             time.sleep(0.5)
  45
  46 def resize_line_edit_width(line_edit, text_input):
  47     metrics = QFontMetrics(qApp.font())
  48     # Create an extra character to add some space on the end
  49     text_input += "A"
  50     line_edit.setMinimumWidth(metrics.width(text_input))
  51
  52 def load_theme_name(theme_path):
  53     try:
  54         with open(os.path.join(theme_path, "name.cfg")) as name_cfg_file:
  55             return name_cfg_file.read().rstrip("\n").strip()
  56     except IOError:
  57         return None
  58
  59
  60 def theme_dirs_from_prefix(prefix):
  61     if not os.path.exists(prefix):
  62         return []
  63     theme_paths = {}
  64     for potential_theme in os.listdir(prefix):
  65         theme_full_path = os.path.join(prefix, potential_theme)
  66         theme_css = os.path.join(theme_full_path, "style.css")
  67         if not os.path.exists(theme_css):
  68             continue
  69         theme_name = load_theme_name(theme_full_path)
  70         if theme_name is None:
  71             continue
  72         theme_paths[theme_name] = prefix, potential_theme
  73     return theme_paths
  74
  75 def load_theme_paths():
  76     theme_paths = {}
  77     prefixes = (util.local_data_dir(), util.appdata_dir())
  78     for prefix in prefixes:
  79         theme_paths.update(theme_dirs_from_prefix(prefix))
  80     return theme_paths
  81
  82
  83 class ElectrumGui(QObject):
  84
  85     def __init__(self, wallet, config):
  86         super(QObject, self).__init__()
  87
  88         self.wallet = wallet
  89         self.config = config
  90         self.check_qt_version()
  91         self.app = QApplication(sys.argv)
  92
  93
  94     def check_qt_version(self):
  95         qtVersion = qVersion()
  96         if not(int(qtVersion[0]) >= 4 and int(qtVersion[2]) >= 7):
  97             app = QApplication(sys.argv)
  98             QMessageBox.warning(None,"Could not start Lite GUI.", "Electrum was unable to load the 'Lite GUI' because it needs Qt version >= 4.7.\nChanging your config to use the 'Classic' GUI")
  99             self.config.set_key('gui','classic',True)
 100             sys.exit(0)
 101
 102
 103     def main(self, url):
 104         actuator = MiniActuator(self.wallet)
 105         self.connect(self, SIGNAL("updateservers()"),
 106                      actuator.update_servers_list)
 107         # Should probably not modify the current path but instead
 108         # change the behaviour of rsrc(...)
 109         old_path = QDir.currentPath()
 110         actuator.load_theme()
 111
 112         self.mini = MiniWindow(actuator, self.expand, self.config)
 113         driver = MiniDriver(self.wallet, self.mini)
 114
 115         # Reset path back to original value now that loading the GUI
 116         # is completed.
 117         QDir.setCurrent(old_path)
 118
 119         if url:
 120             self.set_url(url)
 121
 122         timer = Timer()
 123         timer.start()
 124         self.expert = gui_qt.ElectrumWindow(self.wallet, self.config)
 125         self.expert.app = self.app
 126         self.expert.connect_slots(timer)
 127         self.expert.update_wallet()
 128         self.app.exec_()
 129
 130     def server_list_changed(self):
 131         self.emit(SIGNAL("updateservers()"))
 132
 133     def expand(self):
 134         """Hide the lite mode window and show pro-mode."""
 135         self.mini.hide()
 136         self.expert.show()
 137
 138     def set_url(self, url):
 139         payto, amount, label, message, signature, identity, url = \
 140             self.wallet.parse_url(url, self.show_message, self.show_question)
 141         self.mini.set_payment_fields(payto, amount)
 142
 143     def show_message(self, message):
 144         QMessageBox.information(self.mini, _("Message"), message, _("OK"))
 145
 146     def show_question(self, message):
 147         choice = QMessageBox.question(self.mini, _("Message"), message,
 148                                       QMessageBox.Yes|QMessageBox.No,
 149                                       QMessageBox.No)
 150         return choice == QMessageBox.Yes
 151
 152     def restore_or_create(self):
 153         qt_gui_object = gui_qt.ElectrumGui(self.wallet, self.app)
 154         return qt_gui_object.restore_or_create()
 155
 156 class MiniWindow(QDialog):
 157
 158     def __init__(self, actuator, expand_callback, config):
 159         super(MiniWindow, self).__init__()
 160
 161         self.actuator = actuator
 162         self.config = config
 163
 164         self.btc_balance = None
 165         self.quote_currencies = ["EUR", "USD", "GBP"]
 166         self.actuator.set_configured_currency(self.set_quote_currency)
 167         self.exchanger = exchange_rate.Exchanger(self)
 168         # Needed because price discovery is done in a different thread
 169         # which needs to be sent back to this main one to update the GUI
 170         self.connect(self, SIGNAL("refresh_balance()"), self.refresh_balance)
 171
 172         self.balance_label = BalanceLabel(self.change_quote_currency)
 173         self.balance_label.setObjectName("balance_label")
 174
 175         self.receive_button = QPushButton(_("&Receive"))
 176         self.receive_button.setObjectName("receive_button")
 177         self.receive_button.setDefault(True)
 178         self.receive_button.clicked.connect(self.copy_address)
 179
 180         # Bitcoin address code
 181         self.address_input = QLineEdit()
 182         self.address_input.setPlaceholderText(_("Enter a Bitcoin address..."))
 183         self.address_input.setObjectName("address_input")
 184
 185
 186         self.address_input.textEdited.connect(self.address_field_changed)
 187         resize_line_edit_width(self.address_input,
 188                                "1BtaFUr3qVvAmwrsuDuu5zk6e4s2rxd2Gy")
 189
 190         self.address_completions = QStringListModel()
 191         address_completer = QCompleter(self.address_input)
 192         address_completer.setCaseSensitivity(False)
 193         address_completer.setModel(self.address_completions)
 194         self.address_input.setCompleter(address_completer)
 195
 196         address_layout = QHBoxLayout()
 197         address_layout.addWidget(self.address_input)
 198
 199         self.amount_input = QLineEdit()
 200         self.amount_input.setPlaceholderText(_("... and amount"))
 201         self.amount_input.setObjectName("amount_input")
 202         # This is changed according to the user's displayed balance
 203         self.amount_validator = QDoubleValidator(self.amount_input)
 204         self.amount_validator.setNotation(QDoubleValidator.StandardNotation)
 205         self.amount_validator.setDecimals(8)
 206         self.amount_input.setValidator(self.amount_validator)
 207
 208         # This removes the very ugly OSX highlighting, please leave this in :D
 209         self.address_input.setAttribute(Qt.WA_MacShowFocusRect, 0)
 210         self.amount_input.setAttribute(Qt.WA_MacShowFocusRect, 0)
 211         self.amount_input.textChanged.connect(self.amount_input_changed)
 212
 213         self.send_button = QPushButton(_("&Send"))
 214         self.send_button.setObjectName("send_button")
 215         self.send_button.setDisabled(True);
 216         self.send_button.clicked.connect(self.send)
 217
 218         main_layout = QGridLayout(self)
 219
 220         main_layout.addWidget(self.balance_label, 0, 0)
 221         main_layout.addWidget(self.receive_button, 0, 1)
 222
 223         main_layout.addWidget(self.address_input, 1, 0, 1, -1)
 224
 225         main_layout.addWidget(self.amount_input, 2, 0)
 226         main_layout.addWidget(self.send_button, 2, 1)
 227
 228         self.history_list = history_widget.HistoryWidget()
 229         self.history_list.setObjectName("history")
 230         self.history_list.hide()
 231         self.history_list.setAlternatingRowColors(True)
 232         main_layout.addWidget(self.history_list, 3, 0, 1, -1)
 233
 234         menubar = QMenuBar()
 235         electrum_menu = menubar.addMenu(_("&Bitcoin"))
 236
 237         servers_menu = electrum_menu.addMenu(_("&Servers"))
 238         servers_group = QActionGroup(self)
 239         self.actuator.set_servers_gui_stuff(servers_menu, servers_group)
 240         self.actuator.populate_servers_menu()
 241         electrum_menu.addSeparator()
 242
 243         brain_seed = electrum_menu.addAction(_("&BrainWallet Info"))
 244         brain_seed.triggered.connect(self.actuator.show_seed_dialog)
 245         quit_option = electrum_menu.addAction(_("&Quit"))
 246         quit_option.triggered.connect(self.close)
 247
 248         view_menu = menubar.addMenu(_("&View"))
 249         extra_menu = menubar.addMenu(_("&Extra"))
 250
 251         backup_wallet = extra_menu.addAction( _("&Create wallet backup"))
 252         backup_wallet.triggered.connect(self.backup_wallet)
 253
 254         expert_gui = view_menu.addAction(_("&Classic GUI"))
 255         expert_gui.triggered.connect(expand_callback)
 256         themes_menu = view_menu.addMenu(_("&Themes"))
 257         selected_theme = self.actuator.selected_theme()
 258         theme_group = QActionGroup(self)
 259         for theme_name in self.actuator.theme_names():
 260             theme_action = themes_menu.addAction(theme_name)
 261             theme_action.setCheckable(True)
 262             if selected_theme == theme_name:
 263                 theme_action.setChecked(True)
 264             class SelectThemeFunctor:
 265                 def __init__(self, theme_name, toggle_theme):
 266                     self.theme_name = theme_name
 267                     self.toggle_theme = toggle_theme
 268                 def __call__(self, checked):
 269                     if checked:
 270                         self.toggle_theme(self.theme_name)
 271             delegate = SelectThemeFunctor(theme_name, self.toggle_theme)
 272             theme_action.toggled.connect(delegate)
 273             theme_group.addAction(theme_action)
 274         view_menu.addSeparator()
 275         show_history = view_menu.addAction(_("Show History"))
 276         show_history.setCheckable(True)
 277         show_history.toggled.connect(self.show_history)
 278
 279         help_menu = menubar.addMenu(_("&Help"))
 280         the_website = help_menu.addAction(_("&Website"))
 281         the_website.triggered.connect(self.the_website)
 282         help_menu.addSeparator()
 283         report_bug = help_menu.addAction(_("&Report Bug"))
 284         report_bug.triggered.connect(self.show_report_bug)
 285         show_about = help_menu.addAction(_("&About"))
 286         show_about.triggered.connect(self.show_about)
 287         main_layout.setMenuBar(menubar)
 288
 289         quit_shortcut = QShortcut(QKeySequence("Ctrl+Q"), self)
 290         quit_shortcut.activated.connect(self.close)
 291         close_shortcut = QShortcut(QKeySequence("Ctrl+W"), self)
 292         close_shortcut.activated.connect(self.close)
 293
 294         g = self.config.get("winpos-lite",[4, 25, 351, 149])
 295         self.setGeometry(g[0], g[1], g[2], g[3])
 296
 297         show_hist = self.config.get("gui_show_history",False)
 298         show_history.setChecked(show_hist)
 299         self.show_history(show_hist)
 300
 301         self.setWindowIcon(QIcon(":electrum.png"))
 302         self.setWindowTitle("Electrum")
 303         self.setWindowFlags(Qt.Window|Qt.MSWindowsFixedSizeDialogHint)
 304         self.layout().setSizeConstraint(QLayout.SetFixedSize)
 305         self.setObjectName("main_window")
 306         self.show()
 307
 308     def toggle_theme(self, theme_name):
 309         old_path = QDir.currentPath()
 310         self.actuator.change_theme(theme_name)
 311         # Recompute style globally
 312         qApp.style().unpolish(self)
 313         qApp.style().polish(self)
 314         QDir.setCurrent(old_path)
 315
 316     def closeEvent(self, event):
 317         g = self.geometry()
 318         self.config.set_key("winpos-lite", [g.left(),g.top(),g.width(),g.height()],True)
 319         self.config.set_key("gui_show_history", self.history_list.isVisible(),True)
 320
 321         super(MiniWindow, self).closeEvent(event)
 322         qApp.quit()
 323
 324     def set_payment_fields(self, dest_address, amount):
 325         self.address_input.setText(dest_address)
 326         self.address_field_changed(dest_address)
 327         self.amount_input.setText(amount)
 328
 329     def activate(self):
 330         pass
 331
 332     def deactivate(self):
 333         pass
 334
 335     def set_quote_currency(self, currency):
 336         """Set and display the fiat currency country."""
 337         assert currency in self.quote_currencies
 338         self.quote_currencies.remove(currency)
 339         self.quote_currencies.insert(0, currency)
 340         self.refresh_balance()
 341
 342     def change_quote_currency(self):
 343         self.quote_currencies = \
 344             self.quote_currencies[1:] + self.quote_currencies[0:1]
 345         self.actuator.set_config_currency(self.quote_currencies[0])
 346         self.refresh_balance()
 347
 348     def refresh_balance(self):
 349         if self.btc_balance is None:
 350             # Price has been discovered before wallet has been loaded
 351             # and server connect... so bail.
 352             return
 353         self.set_balances(self.btc_balance)
 354         self.amount_input_changed(self.amount_input.text())
 355
 356     def set_balances(self, btc_balance):
 357         """Set the bitcoin balance and update the amount label accordingly."""
 358         self.btc_balance = btc_balance
 359         quote_text = self.create_quote_text(btc_balance)
 360         if quote_text:
 361             quote_text = "(%s)" % quote_text
 362         btc_balance = "%.2f" % (btc_balance / bitcoin(1))
 363         self.balance_label.set_balance_text(btc_balance, quote_text)
 364         self.setWindowTitle("Electrum - %s BTC" % btc_balance)
 365
 366     def amount_input_changed(self, amount_text):
 367         """Update the number of bitcoins displayed."""
 368         self.check_button_status()
 369
 370         try:
 371             amount = D(str(amount_text))
 372         except decimal.InvalidOperation:
 373             self.balance_label.show_balance()
 374         else:
 375             quote_text = self.create_quote_text(amount * bitcoin(1))
 376             if quote_text:
 377                 self.balance_label.set_amount_text(quote_text)
 378                 self.balance_label.show_amount()
 379             else:
 380                 self.balance_label.show_balance()
 381
 382     def create_quote_text(self, btc_balance):
 383         """Return a string copy of the amount fiat currency the
 384         user has in bitcoins."""
 385         quote_currency = self.quote_currencies[0]
 386         quote_balance = self.exchanger.exchange(btc_balance, quote_currency)
 387         if quote_balance is None:
 388             quote_text = ""
 389         else:
 390             quote_text = "%.2f %s" % ((quote_balance / bitcoin(1)),
 391                                       quote_currency)
 392         return quote_text
 393
 394     def send(self):
 395         if self.actuator.send(self.address_input.text(),
 396                               self.amount_input.text(), self):
 397             self.address_input.setText("")
 398             self.amount_input.setText("")
 399
 400     def check_button_status(self):
 401         """Check that the bitcoin address is valid and that something
 402         is entered in the amount before making the send button clickable."""
 403         try:
 404             value = D(str(self.amount_input.text())) * 10**8
 405         except decimal.InvalidOperation:
 406             value = None
 407         # self.address_input.property(...) returns a qVariant, not a bool.
 408         # The == is needed to properly invoke a comparison.
 409         if (self.address_input.property("isValid") == True and
 410             value is not None and 0 < value <= self.btc_balance):
 411             self.send_button.setDisabled(False)
 412         else:
 413             self.send_button.setDisabled(True)
 414
 415     def address_field_changed(self, address):
 416         if self.actuator.is_valid(address):
 417             self.check_button_status()
 418             self.address_input.setProperty("isValid", True)
 419             self.recompute_style(self.address_input)
 420         else:
 421             self.send_button.setDisabled(True)
 422             self.address_input.setProperty("isValid", False)
 423             self.recompute_style(self.address_input)
 424
 425         if len(address) == 0:
 426             self.address_input.setProperty("isValid", None)
 427             self.recompute_style(self.address_input)
 428
 429     def recompute_style(self, element):
 430         self.style().unpolish(element)
 431         self.style().polish(element)
 432
 433     def copy_address(self):
 434         receive_popup = ReceivePopup(self.receive_button)
 435         self.actuator.copy_address(receive_popup)
 436
 437     def update_completions(self, completions):
 438         self.address_completions.setStringList(completions)
 439
 440     def update_history(self, tx_history):
 441         for tx in tx_history[-10:]:
 442             address = tx["default_label"]
 443             amount = D(tx["value"]) / 10**8
 444             self.history_list.append(address, amount)
 445
 446     def acceptbit(self):
 447         self.actuator.acceptbit(self.quote_currencies[0])
 448
 449     def the_website(self):
 450         webbrowser.open("http://electrum-desktop.com")
 451
 452     def show_about(self):
 453         QMessageBox.about(self, "Electrum",
 454             _("Electrum's focus is speed, with low resource usage and simplifying Bitcoin. You do not need to perform regular backups, because your wallet can be recovered from a secret phrase that you can memorize or write on paper. Startup times are instant because it operates in conjuction with high-performance servers that handle the most complicated parts of the Bitcoin system.\n\nSend donations to 1JwTMv4GWaPdf931N6LNPJeZBfZgZJ3zX1"))
 455
 456     def show_report_bug(self):
 457         QMessageBox.information(self, "Electrum - " + _("Reporting Bugs"),
 458             _("Email bug reports to %s") % "genjix" + "@" + "riseup.net")
 459
 460     def show_history(self, toggle_state):
 461         if toggle_state:
 462             self.history_list.show()
 463         else:
 464             self.history_list.hide()
 465
 466     def backup_wallet(self):
 467         try:
 468           folderName = QFileDialog.getExistingDirectory(QWidget(), 'Select folder to save a copy of your wallet to', os.path.expanduser('~/'))
 469           if folderName:
 470             sourceFile = util.user_dir() + '/electrum.dat'
 471             shutil.copy2(sourceFile, str(folderName))
 472             QMessageBox.information(None,"Wallet backup created", "A copy of your wallet file was created in '%s'" % str(folderName))
 473         except (IOError, os.error), reason:
 474           QMessageBox.critical(None,"Unable to create backup", "Electrum was unable copy your wallet file to the specified location.\n" + str(reason))
 475
 476
 477
 478
 479 class BalanceLabel(QLabel):
 480
 481     SHOW_CONNECTING = 1
 482     SHOW_BALANCE = 2
 483     SHOW_AMOUNT = 3
 484
 485     def __init__(self, change_quote_currency, parent=None):
 486         super(QLabel, self).__init__(_("Connecting..."), parent)
 487         self.change_quote_currency = change_quote_currency
 488         self.state = self.SHOW_CONNECTING
 489         self.balance_text = ""
 490         self.amount_text = ""
 491
 492     def mousePressEvent(self, event):
 493         """Change the fiat currency selection if window background is clicked."""
 494         if self.state != self.SHOW_CONNECTING:
 495             self.change_quote_currency()
 496
 497     def set_balance_text(self, btc_balance, quote_text):
 498         """Set the amount of bitcoins in the gui."""
 499         if self.state == self.SHOW_CONNECTING:
 500             self.state = self.SHOW_BALANCE
 501         self.balance_text = "<span style='font-size: 18pt'>%s</span> <span style='font-size: 10pt'>BTC</span> <span style='font-size: 10pt'>%s</span>" % (btc_balance, quote_text)
 502         if self.state == self.SHOW_BALANCE:
 503             self.setText(self.balance_text)
 504
 505     def set_amount_text(self, quote_text):
 506         self.amount_text = "<span style='font-size: 10pt'>%s</span>" % quote_text
 507         if self.state == self.SHOW_AMOUNT:
 508             self.setText(self.amount_text)
 509
 510     def show_balance(self):
 511         if self.state == self.SHOW_AMOUNT:
 512             self.state = self.SHOW_BALANCE
 513             self.setText(self.balance_text)
 514
 515     def show_amount(self):
 516         if self.state == self.SHOW_BALANCE:
 517             self.state = self.SHOW_AMOUNT
 518             self.setText(self.amount_text)
 519
 520 def ok_cancel_buttons(dialog):
 521     row_layout = QHBoxLayout()
 522     row_layout.addStretch(1)
 523     ok_button = QPushButton(_("OK"))
 524     row_layout.addWidget(ok_button)
 525     ok_button.clicked.connect(dialog.accept)
 526     cancel_button = QPushButton(_("Cancel"))
 527     row_layout.addWidget(cancel_button)
 528     cancel_button.clicked.connect(dialog.reject)
 529     return row_layout
 530
 531 class PasswordDialog(QDialog):
 532
 533     def __init__(self, parent):
 534         super(QDialog, self).__init__(parent)
 535
 536         self.setModal(True)
 537
 538         self.password_input = QLineEdit()
 539         self.password_input.setEchoMode(QLineEdit.Password)
 540
 541         main_layout = QVBoxLayout(self)
 542         message = _('Please enter your password')
 543         main_layout.addWidget(QLabel(message))
 544
 545         grid = QGridLayout()
 546         grid.setSpacing(8)
 547         grid.addWidget(QLabel(_('Password')), 1, 0)
 548         grid.addWidget(self.password_input, 1, 1)
 549         main_layout.addLayout(grid)
 550
 551         main_layout.addLayout(ok_cancel_buttons(self))
 552         self.setLayout(main_layout)
 553
 554     def run(self):
 555         if not self.exec_():
 556             return
 557         return unicode(self.password_input.text())
 558
 559 class ReceivePopup(QDialog):
 560
 561     def leaveEvent(self, event):
 562         self.close()
 563
 564     def setup(self, address):
 565         label = QLabel(_("Copied your Bitcoin address to the clipboard!"))
 566         address_display = QLineEdit(address)
 567         address_display.setReadOnly(True)
 568         resize_line_edit_width(address_display, address)
 569
 570         main_layout = QVBoxLayout(self)
 571         main_layout.addWidget(label)
 572         main_layout.addWidget(address_display)
 573
 574         self.setMouseTracking(True)
 575         self.setWindowTitle("Electrum - " + _("Receive Bitcoin payment"))
 576         self.setWindowFlags(Qt.Window|Qt.FramelessWindowHint|
 577                             Qt.MSWindowsFixedSizeDialogHint)
 578         self.layout().setSizeConstraint(QLayout.SetFixedSize)
 579         #self.setFrameStyle(QFrame.WinPanel|QFrame.Raised)
 580         #self.setAlignment(Qt.AlignCenter)
 581
 582     def popup(self):
 583         parent = self.parent()
 584         top_left_pos = parent.mapToGlobal(parent.rect().bottomLeft())
 585         self.move(top_left_pos)
 586         center_mouse_pos = self.mapToGlobal(self.rect().center())
 587         QCursor.setPos(center_mouse_pos)
 588         self.show()
 589
 590 class MiniActuator:
 591     """Initialize the definitions relating to themes and
 592     sending/recieving bitcoins."""
 593
 594
 595     def __init__(self, wallet):
 596         """Retrieve the gui theme used in previous session."""
 597         self.wallet = wallet
 598         self.theme_name = self.wallet.config.get('litegui_theme','Cleanlook')
 599         self.themes = load_theme_paths()
 600
 601     def load_theme(self):
 602         """Load theme retrieved from wallet file."""
 603         try:
 604             theme_prefix, theme_path = self.themes[self.theme_name]
 605         except KeyError:
 606             util.print_error("Theme not found!", self.theme_name)
 607             return
 608         QDir.setCurrent(os.path.join(theme_prefix, theme_path))
 609         with open(rsrc("style.css")) as style_file:
 610             qApp.setStyleSheet(style_file.read())
 611
 612     def theme_names(self):
 613         """Sort themes."""
 614         return sorted(self.themes.keys())
 615
 616     def selected_theme(self):
 617         """Select theme."""
 618         return self.theme_name
 619
 620     def change_theme(self, theme_name):
 621         """Change theme."""
 622         self.theme_name = theme_name
 623         self.wallet.config.set_key('litegui_theme',theme_name)
 624         self.load_theme()
 625
 626     def set_configured_currency(self, set_quote_currency):
 627         """Set the inital fiat currency conversion country (USD/EUR/GBP) in
 628         the GUI to what it was set to in the wallet."""
 629         currency = self.wallet.config.get('conversion_currency')
 630         # currency can be none when Electrum is used for the first
 631         # time and no setting has been created yet.
 632         if currency is not None:
 633             set_quote_currency(currency)
 634
 635     def set_config_currency(self, conversion_currency):
 636         """Change the wallet fiat currency country."""
 637         self.wallet.config.set_key('conversion_currency',conversion_currency,True)
 638
 639     def set_servers_gui_stuff(self, servers_menu, servers_group):
 640         self.servers_menu = servers_menu
 641         self.servers_group = servers_group
 642
 643     def populate_servers_menu(self):
 644         interface = self.wallet.interface
 645         if not interface.servers:
 646             print "No servers loaded yet."
 647             self.servers_list = []
 648             for server_string in DEFAULT_SERVERS:
 649                 host, port, protocol = server_string.split(':')
 650                 transports = [(protocol,port)]
 651                 self.servers_list.append((host, transports))
 652         else:
 653             print "Servers loaded."
 654             self.servers_list = interface.servers
 655         server_names = [details[0] for details in self.servers_list]
 656         current_server = interface.server.split(":")[0]
 657         for server_name in server_names:
 658             server_action = self.servers_menu.addAction(server_name)
 659             server_action.setCheckable(True)
 660             if server_name == current_server:
 661                 server_action.setChecked(True)
 662             class SelectServerFunctor:
 663                 def __init__(self, server_name, server_selected):
 664                     self.server_name = server_name
 665                     self.server_selected = server_selected
 666                 def __call__(self, checked):
 667                     if checked:
 668                         # call server_selected
 669                         self.server_selected(self.server_name)
 670             delegate = SelectServerFunctor(server_name, self.server_selected)
 671             server_action.toggled.connect(delegate)
 672             self.servers_group.addAction(server_action)
 673
 674     def update_servers_list(self):
 675         # Clear servers_group
 676         for action in self.servers_group.actions():
 677             self.servers_group.removeAction(action)
 678         self.populate_servers_menu()
 679
 680     def server_selected(self, server_name):
 681         match = [transports for (host, transports) in self.servers_list
 682                  if host == server_name]
 683         assert len(match) == 1
 684         match = match[0]
 685         # Default to TCP if available else use anything
 686         # TODO: protocol should be selectable.
 687         tcp_port = [port for (protocol, port) in match if protocol == "t"]
 688         if len(tcp_port) == 0:
 689             protocol = match[0][0]
 690             port = match[0][1]
 691         else:
 692             protocol = "t"
 693             port = tcp_port[0]
 694         server_line = "%s:%s:%s" % (server_name, port, protocol)
 695
 696         # Should this have exception handling?
 697         self.wallet.interface.set_server(server_line, self.wallet.config.get("proxy"))
 698
 699     def copy_address(self, receive_popup):
 700         """Copy the wallet addresses into the client."""
 701         addrs = [addr for addr in self.wallet.all_addresses()
 702                  if not self.wallet.is_change(addr)]
 703         # Select most recent addresses from gap limit
 704         addrs = addrs[-self.wallet.gap_limit:]
 705         copied_address = random.choice(addrs)
 706         qApp.clipboard().setText(copied_address)
 707         receive_popup.setup(copied_address)
 708         receive_popup.popup()
 709
 710     def send(self, address, amount, parent_window):
 711         """Send bitcoins to the target address."""
 712         dest_address = self.fetch_destination(address)
 713
 714         if dest_address is None or not self.wallet.is_valid(dest_address):
 715             QMessageBox.warning(parent_window, _('Error'),
 716                 _('Invalid Bitcoin Address') + ':\n' + address, _('OK'))
 717             return False
 718
 719         convert_amount = lambda amount: \
 720             int(D(unicode(amount)) * bitcoin(1))
 721         amount = convert_amount(amount)
 722
 723         if self.wallet.use_encryption:
 724             password_dialog = PasswordDialog(parent_window)
 725             password = password_dialog.run()
 726             if not password:
 727                 return
 728         else:
 729             password = None
 730
 731         fee = 0
 732         # 0.1 BTC = 10000000
 733         if amount < bitcoin(1) / 10:
 734             # 0.001 BTC
 735             fee = bitcoin(1) / 1000
 736
 737         try:
 738             tx = self.wallet.mktx(dest_address, amount, "", password, fee)
 739         except BaseException as error:
 740             QMessageBox.warning(parent_window, _('Error'), str(error), _('OK'))
 741             return False
 742
 743         status, message = self.wallet.sendtx(tx)
 744         if not status:
 745             import tempfile
 746             dumpf = tempfile.NamedTemporaryFile(delete=False)
 747             dumpf.write(tx)
 748             dumpf.close()
 749             print "Dumped error tx to", dumpf.name
 750             QMessageBox.warning(parent_window, _('Error'), message, _('OK'))
 751             return False
 752
 753         QMessageBox.information(parent_window, '',
 754             _('Payment sent.') + '\n' + message, _('OK'))
 755         return True
 756
 757     def fetch_destination(self, address):
 758         recipient = unicode(address).strip()
 759
 760         # alias
 761         match1 = re.match("^(|([\w\-\.]+)@)((\w[\w\-]+\.)+[\w\-]+)$",
 762                           recipient)
 763
 764         # label or alias, with address in brackets
 765         match2 = re.match("(.*?)\s*\<([1-9A-HJ-NP-Za-km-z]{26,})\>",
 766                           recipient)
 767
 768         if match1:
 769             dest_address = \
 770                 self.wallet.get_alias(recipient, True,
 771                                       self.show_message, self.question)
 772             return dest_address
 773         elif match2:
 774             return match2.group(2)
 775         else:
 776             return recipient
 777
 778     def is_valid(self, address):
 779         """Check if bitcoin address is valid."""
 780         return self.wallet.is_valid(address)
 781
 782     def acceptbit(self, currency):
 783         master_pubkey = self.wallet.master_public_key
 784         url = "http://acceptbit.com/mpk/%s/%s" % (master_pubkey, currency)
 785         webbrowser.open(url)
 786
 787     def show_seed_dialog(self):
 788         gui_qt.ElectrumWindow.show_seed_dialog(self.wallet)
 789
 790 class MiniDriver(QObject):
 791
 792     INITIALIZING = 0
 793     CONNECTING = 1
 794     SYNCHRONIZING = 2
 795     READY = 3
 796
 797     def __init__(self, wallet, window):
 798         super(QObject, self).__init__()
 799
 800         self.wallet = wallet
 801         self.window = window
 802
 803         self.wallet.register_callback(self.update_callback)
 804
 805         self.state = None
 806
 807         self.initializing()
 808         self.connect(self, SIGNAL("updatesignal()"), self.update)
 809         self.update_callback()
 810
 811     # This is a hack to workaround that Qt does not like changing the
 812     # window properties from this other thread before the runloop has
 813     # been called from.
 814     def update_callback(self):
 815         self.emit(SIGNAL("updatesignal()"))
 816
 817     def update(self):
 818         if not self.wallet.interface:
 819             self.initializing()
 820         elif not self.wallet.interface.is_connected:
 821             self.connecting()
 822         elif not self.wallet.blocks == -1:
 823             self.connecting()
 824         elif not self.wallet.up_to_date:
 825             self.synchronizing()
 826         else:
 827             self.ready()
 828
 829         if self.wallet.up_to_date:
 830             self.update_balance()
 831             self.update_completions()
 832             self.update_history()
 833
 834     def initializing(self):
 835         if self.state == self.INITIALIZING:
 836             return
 837         self.state = self.INITIALIZING
 838         self.window.deactivate()
 839
 840     def connecting(self):
 841         if self.state == self.CONNECTING:
 842             return
 843         self.state = self.CONNECTING
 844         self.window.deactivate()
 845
 846     def synchronizing(self):
 847         if self.state == self.SYNCHRONIZING:
 848             return
 849         self.state = self.SYNCHRONIZING
 850         self.window.deactivate()
 851
 852     def ready(self):
 853         if self.state == self.READY:
 854             return
 855         self.state = self.READY
 856         self.window.activate()
 857
 858     def update_balance(self):
 859         conf_balance, unconf_balance = self.wallet.get_balance()
 860         balance = D(conf_balance + unconf_balance)
 861         self.window.set_balances(balance)
 862
 863     def update_completions(self):
 864         completions = []
 865         for addr, label in self.wallet.labels.items():
 866             if addr in self.wallet.addressbook:
 867                 completions.append("%s <%s>" % (label, addr))
 868         completions = completions + self.wallet.aliases.keys()
 869         self.window.update_completions(completions)
 870
 871     def update_history(self):
 872         tx_history = self.wallet.get_tx_history()
 873         self.window.update_history(tx_history)
 874
 875 if __name__ == "__main__":
 876     app = QApplication(sys.argv)
 877     with open(rsrc("style.css")) as style_file:
 878         app.setStyleSheet(style_file.read())
 879     mini = MiniWindow()
 880     sys.exit(app.exec_())
 881