gui/qt/__init__.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, threading
  20 from electrum.i18n import _, set_language
  21 from electrum.util import print_error, print_msg, parse_url
  22 from electrum.plugins import run_hook
  23 import os.path, json, ast, traceback
  24 import shutil
  25
  26
  27 try:
  28     import PyQt4
  29 except Exception:
  30     sys.exit("Error: Could not import PyQt4 on Linux systems, you may try 'sudo apt-get install python-qt4'")
  31
  32 from PyQt4.QtGui import *
  33 from PyQt4.QtCore import *
  34 import PyQt4.QtCore as QtCore
  35
  36 from electrum import WalletStorage, Wallet
  37 from electrum.i18n import _
  38 from electrum.bitcoin import MIN_RELAY_TX_FEE
  39
  40 try:
  41     import icons_rc
  42 except Exception:
  43     sys.exit("Error: Could not import icons_rc.py, please generate it with: 'pyrcc4 icons.qrc -o gui/qt/icons_rc.py'")
  44
  45 from util import *
  46 from main_window import ElectrumWindow
  47 from electrum.plugins import init_plugins
  48
  49
  50 class OpenFileEventFilter(QObject):
  51     def __init__(self, windows):
  52         self.windows = windows
  53         super(OpenFileEventFilter, self).__init__()
  54
  55     def eventFilter(self, obj, event):
  56         if event.type() == QtCore.QEvent.FileOpen:
  57             if len(self.windows) >= 1:
  58                 self.windows[0].set_url(event.url().toEncoded())
  59                 return True
  60         return False
  61
  62
  63 class ElectrumGui:
  64
  65     def __init__(self, config, network, app=None):
  66         self.network = network
  67         self.config = config
  68         self.windows = []
  69         self.efilter = OpenFileEventFilter(self.windows)
  70         if app is None:
  71             self.app = QApplication(sys.argv)
  72         self.app.installEventFilter(self.efilter)
  73         init_plugins(self)
  74         self.payment_request = None
  75
  76
  77     def build_tray_menu(self):
  78         m = QMenu()
  79         m.addAction(_("Show/Hide"), self.show_or_hide)
  80         m.addAction(_("Dark/Light"), self.toggle_tray_icon)
  81         m.addSeparator()
  82         m.addAction(_("Exit Electrum"), self.close)
  83         self.tray.setContextMenu(m)
  84
  85     def toggle_tray_icon(self):
  86         self.dark_icon = not self.dark_icon
  87         self.config.set_key("dark_icon", self.dark_icon, True)
  88         icon = QIcon(":icons/electrum_dark_icon.png") if self.dark_icon else QIcon(':icons/electrum_light_icon.png')
  89         self.tray.setIcon(icon)
  90
  91     def show_or_hide(self):
  92         self.tray_activated(QSystemTrayIcon.DoubleClick)
  93
  94     def tray_activated(self, reason):
  95         if reason == QSystemTrayIcon.DoubleClick:
  96             if self.current_window.isMinimized() or self.current_window.isHidden():
  97                 self.current_window.show()
  98                 self.current_window.raise_()
  99             else:
 100                 self.current_window.hide()
 101
 102     def close(self):
 103         self.current_window.close()
 104
 105
 106
 107     def go_full(self):
 108         self.config.set_key('lite_mode', False, True)
 109         self.lite_window.hide()
 110         self.main_window.show()
 111         self.main_window.raise_()
 112         self.current_window = self.main_window
 113
 114     def go_lite(self):
 115         self.config.set_key('lite_mode', True, True)
 116         self.main_window.hide()
 117         self.lite_window.show()
 118         self.lite_window.raise_()
 119         self.current_window = self.lite_window
 120
 121
 122     def init_lite(self):
 123         import lite_window
 124         if not self.check_qt_version():
 125             if self.config.get('lite_mode') is True:
 126                 msg = "Electrum was unable to load the 'Lite GUI' because it needs Qt version >= 4.7.\nChanging your config to use the 'Classic' GUI"
 127                 QMessageBox.warning(None, "Could not start Lite GUI.", msg)
 128                 self.config.set_key('lite_mode', False, True)
 129                 sys.exit(0)
 130             self.lite_window = None
 131             return
 132
 133         actuator = lite_window.MiniActuator(self.main_window)
 134         actuator.load_theme()
 135         self.lite_window = lite_window.MiniWindow(actuator, self.go_full, self.config)
 136         driver = lite_window.MiniDriver(self.main_window, self.lite_window)
 137
 138
 139
 140     def check_qt_version(self):
 141         qtVersion = qVersion()
 142         return int(qtVersion[0]) >= 4 and int(qtVersion[2]) >= 7
 143
 144
 145     def set_url(self, url):
 146         from electrum import util
 147         from decimal import Decimal
 148
 149         try:
 150             address, amount, label, message, request_url, url = util.parse_url(url)
 151         except Exception:
 152             QMessageBox.warning(self.main_window, _('Error'), _('Invalid bitcoin URL'), _('OK'))
 153             return
 154
 155         if amount:
 156             try:
 157                 if self.main_window.base_unit() == 'mBTC':
 158                     amount = str( 1000* Decimal(amount))
 159                 else:
 160                     amount = str(Decimal(amount))
 161             except Exception:
 162                 amount = "0.0"
 163                 QMessageBox.warning(self.main_window, _('Error'), _('Invalid Amount'), _('OK'))
 164
 165         if request_url:
 166             from electrum import paymentrequest
 167
 168         if not request_url:
 169             self.main_window.set_send(address, amount, label, message)
 170             self.lite_window.set_payment_fields(address, amount)
 171             return
 172
 173         def payment_request():
 174             self.payment_request = paymentrequest.PaymentRequest(self.config)
 175             self.payment_request.read(request_url)
 176             if self.payment_request.verify():
 177                 self.main_window.emit(SIGNAL('payment_request_ok'))
 178             else:
 179                 self.main_window.emit(SIGNAL('payment_request_error'))
 180
 181         threading.Thread(target=payment_request).start()
 182         self.main_window.prepare_for_payment_request()
 183
 184
 185     def main(self, url):
 186
 187         storage = WalletStorage(self.config)
 188         if storage.file_exists:
 189             wallet = Wallet(storage)
 190             action = wallet.get_action()
 191         else:
 192             action = 'new'
 193
 194         if action is not None:
 195             import installwizard
 196             wizard = installwizard.InstallWizard(self.config, self.network, storage)
 197             wallet = wizard.run(action)
 198             if not wallet:
 199                 exit()
 200         else:
 201             wallet.start_threads(self.network)
 202
 203         # init tray
 204         self.dark_icon = self.config.get("dark_icon", False)
 205         icon = QIcon(":icons/electrum_dark_icon.png") if self.dark_icon else QIcon(':icons/electrum_light_icon.png')
 206         self.tray = QSystemTrayIcon(icon, None)
 207         self.tray.setToolTip('Electrum')
 208         self.tray.activated.connect(self.tray_activated)
 209         self.build_tray_menu()
 210         self.tray.show()
 211
 212         # main window
 213         self.main_window = w = ElectrumWindow(self.config, self.network, self)
 214         self.current_window = self.main_window
 215
 216         #lite window
 217         self.init_lite()
 218
 219         # initial configuration
 220         if self.config.get('hide_gui') is True and self.tray.isVisible():
 221             self.main_window.hide()
 222             self.lite_window.hide()
 223         else:
 224             if self.config.get('lite_mode') is True:
 225                 self.go_lite()
 226             else:
 227                 self.go_full()
 228
 229         # plugins that need to change the GUI do it here
 230         run_hook('init')
 231
 232         w.load_wallet(wallet)
 233
 234         s = Timer()
 235         s.start()
 236
 237         self.windows.append(w)
 238         if url:
 239             self.set_url(url)
 240
 241         w.app = self.app
 242         w.connect_slots(s)
 243         w.update_wallet()
 244
 245         self.app.exec_()
 246
 247         # clipboard persistence
 248         # see http://www.mail-archive.com/pyqt@riverbankcomputing.com/msg17328.html
 249         event = QtCore.QEvent(QtCore.QEvent.Clipboard)
 250         self.app.sendEvent(self.app.clipboard(), event)
 251
 252         wallet.stop_threads()
 253
 254