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
  74         init_plugins(self)
  75
  76
  77     def main(self, url):
  78
  79         storage = WalletStorage(self.config)
  80         if not storage.file_exists:
  81             import installwizard
  82             wizard = installwizard.InstallWizard(self.config, self.network, storage)
  83             wallet = wizard.run()
  84             if not wallet:
  85                 exit()
  86         else:
  87             wallet = Wallet(storage)
  88             wallet.start_threads(self.network)
  89
  90         self.main_window = w = ElectrumWindow(self.config, self.network)
  91
  92         # plugins that need to change the GUI do it here
  93         run_hook('init')
  94
  95         w.load_wallet(wallet)
  96
  97         s = Timer()
  98         s.start()
  99
 100         self.windows.append(w)
 101         if url: w.set_url(url)
 102         w.app = self.app
 103         w.connect_slots(s)
 104         w.update_wallet()
 105
 106         self.app.exec_()
 107
 108         wallet.stop_threads()
 109
 110