mirror of
https://github.com/LBRYFoundation/LBRY-Vault.git
synced 2025-08-27 15:31:31 +00:00
11d135b32d
Break out the workflow logic of the install wizard into a base class. This means reimplementing with full support in a new GUI is now easy; you just provide ways to request passwords, show messages etc. The API is fully documented in the base class. There are a couple of minor outstanding issues, including that the old messages shown when recovering a wallet are missing. I will come back to that. Ledger wallet might be broken. Other improvements: The install wizard code is now easy to follow and understand. Hardware wallets can now be restored without any need for their accompanying libraries. Various bits of trustedcoin were broken and have been fixed. Many plugin hooks can be removed. I have only started on this.
199 lines
6.9 KiB
Python
199 lines
6.9 KiB
Python
#!/usr/bin/env python
|
|
#
|
|
# Electrum - lightweight Bitcoin client
|
|
# Copyright (C) 2012 thomasv@gitorious
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import sys
|
|
import os
|
|
import signal
|
|
|
|
try:
|
|
import PyQt4
|
|
except Exception:
|
|
sys.exit("Error: Could not import PyQt4 on Linux systems, you may try 'sudo apt-get install python-qt4'")
|
|
|
|
from PyQt4.QtGui import *
|
|
from PyQt4.QtCore import *
|
|
import PyQt4.QtCore as QtCore
|
|
|
|
from electrum.i18n import _, set_language
|
|
from electrum.plugins import run_hook
|
|
from electrum import SimpleConfig, Wallet, WalletStorage
|
|
from electrum.paymentrequest import InvoiceStore
|
|
from electrum.contacts import Contacts
|
|
from electrum.synchronizer import Synchronizer
|
|
from electrum.verifier import SPV
|
|
from electrum.util import DebugMem
|
|
from electrum.wallet import Abstract_Wallet
|
|
from installwizard import InstallWizard
|
|
|
|
|
|
try:
|
|
import icons_rc
|
|
except Exception:
|
|
sys.exit("Error: Could not import icons_rc.py, please generate it with: 'pyrcc4 icons.qrc -o gui/qt/icons_rc.py'")
|
|
|
|
from util import * # * needed for plugins
|
|
from main_window import ElectrumWindow
|
|
|
|
|
|
class OpenFileEventFilter(QObject):
|
|
def __init__(self, windows):
|
|
self.windows = windows
|
|
super(OpenFileEventFilter, self).__init__()
|
|
|
|
def eventFilter(self, obj, event):
|
|
if event.type() == QtCore.QEvent.FileOpen:
|
|
if len(self.windows) >= 1:
|
|
self.windows[0].pay_to_URI(event.url().toEncoded())
|
|
return True
|
|
return False
|
|
|
|
|
|
|
|
class ElectrumGui(MessageBoxMixin):
|
|
|
|
def __init__(self, config, network, plugins):
|
|
set_language(config.get('language'))
|
|
# Uncomment this call to verify objects are being properly
|
|
# GC-ed when windows are closed
|
|
#network.add_jobs([DebugMem([Abstract_Wallet, SPV, Synchronizer,
|
|
# ElectrumWindow], interval=5)])
|
|
self.network = network
|
|
self.config = config
|
|
self.plugins = plugins
|
|
self.windows = []
|
|
self.efilter = OpenFileEventFilter(self.windows)
|
|
self.app = QApplication(sys.argv)
|
|
self.app.installEventFilter(self.efilter)
|
|
self.timer = Timer()
|
|
# shared objects
|
|
self.invoices = InvoiceStore(self.config)
|
|
self.contacts = Contacts(self.config)
|
|
# init tray
|
|
self.dark_icon = self.config.get("dark_icon", False)
|
|
self.tray = QSystemTrayIcon(self.tray_icon(), None)
|
|
self.tray.setToolTip('Electrum')
|
|
self.tray.activated.connect(self.tray_activated)
|
|
self.build_tray_menu()
|
|
self.tray.show()
|
|
self.app.connect(self.app, QtCore.SIGNAL('new_window'), self.start_new_window)
|
|
run_hook('init_qt', self)
|
|
|
|
def build_tray_menu(self):
|
|
# Avoid immediate GC of old menu when window closed via its action
|
|
self.old_menu = self.tray.contextMenu()
|
|
m = QMenu()
|
|
for window in self.windows:
|
|
submenu = m.addMenu(window.wallet.basename())
|
|
submenu.addAction(_("Show/Hide"), window.show_or_hide)
|
|
submenu.addAction(_("Close"), window.close)
|
|
m.addAction(_("Dark/Light"), self.toggle_tray_icon)
|
|
m.addSeparator()
|
|
m.addAction(_("Exit Electrum"), self.close)
|
|
self.tray.setContextMenu(m)
|
|
|
|
def tray_icon(self):
|
|
if self.dark_icon:
|
|
return QIcon(':icons/electrum_dark_icon.png')
|
|
else:
|
|
return QIcon(':icons/electrum_light_icon.png')
|
|
|
|
def toggle_tray_icon(self):
|
|
self.dark_icon = not self.dark_icon
|
|
self.config.set_key("dark_icon", self.dark_icon, True)
|
|
self.tray.setIcon(self.tray_icon())
|
|
|
|
def tray_activated(self, reason):
|
|
if reason == QSystemTrayIcon.DoubleClick:
|
|
if all([w.is_hidden() for w in self.windows]):
|
|
for w in self.windows:
|
|
w.bring_to_top()
|
|
else:
|
|
for w in self.windows:
|
|
w.hide()
|
|
|
|
def close(self):
|
|
for window in self.windows:
|
|
window.close()
|
|
|
|
def new_window(self, path, uri=None):
|
|
# Use a signal as can be called from daemon thread
|
|
self.app.emit(SIGNAL('new_window'), path, uri)
|
|
|
|
def create_window_for_wallet(self, wallet, task):
|
|
w = ElectrumWindow(self, wallet)
|
|
self.windows.append(w)
|
|
self.build_tray_menu()
|
|
if task:
|
|
WaitingDialog(w, task[0], task[1])
|
|
# FIXME: Remove in favour of the load_wallet hook
|
|
run_hook('on_new_window', w)
|
|
return w
|
|
|
|
def start_new_window(self, path, uri):
|
|
'''Raises the window for the wallet if it is open. Otherwise
|
|
opens the wallet and creates a new window for it.'''
|
|
for w in self.windows:
|
|
if w.wallet.storage.path == path:
|
|
w.bring_to_top()
|
|
break
|
|
else:
|
|
wizard = InstallWizard(self.config, self.app, self.plugins)
|
|
result = wizard.open_wallet(self.network, path)
|
|
if not result:
|
|
return
|
|
w = self.create_window_for_wallet(*result)
|
|
|
|
if uri:
|
|
w.pay_to_URI(uri)
|
|
|
|
return w
|
|
|
|
def close_window(self, window):
|
|
self.windows.remove(window)
|
|
self.build_tray_menu()
|
|
# save wallet path of last open window
|
|
if self.config.get('wallet_path') is None and not self.windows:
|
|
path = window.wallet.storage.path
|
|
self.config.set_key('gui_last_wallet', path)
|
|
run_hook('on_close_window', window)
|
|
|
|
def main(self):
|
|
self.timer.start()
|
|
# open last wallet
|
|
if self.config.get('wallet_path') is None:
|
|
last_wallet = self.config.get('gui_last_wallet')
|
|
if last_wallet is not None and os.path.exists(last_wallet):
|
|
self.config.cmdline_options['default_wallet_path'] = last_wallet
|
|
|
|
if not self.start_new_window(self.config.get_wallet_path(),
|
|
self.config.get('url')):
|
|
return
|
|
|
|
signal.signal(signal.SIGINT, lambda *args: self.app.quit())
|
|
|
|
# main loop
|
|
self.app.exec_()
|
|
|
|
# Shut down the timer cleanly
|
|
self.timer.stop()
|
|
|
|
# clipboard persistence. see http://www.mail-archive.com/pyqt@riverbankcomputing.com/msg17328.html
|
|
event = QtCore.QEvent(QtCore.QEvent.Clipboard)
|
|
self.app.sendEvent(self.app.clipboard(), event)
|
|
|
|
self.tray.hide()
|