AaronFiore/electrum
1
0
Fork 0
Code Activity

qml: add available wallets model

Browse Source
This commit is contained in:
Sander van Grieken
2021-04-06 14:13:51 +02:00
parent 599b01f009
commit 3b22ecdae4
4 changed files with 135 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
electrum/gui/qml/__init__.py
View File

@@ -15,7 +15,7 @@ try:
except Exception:
sys.exit("Error: Could not import PyQt5.QtQml on Linux systems, you may try 'sudo apt-get install python3-pyqt5.qtquick'")
from PyQt5.QtCore import pyqtProperty, pyqtSignal, QObject, QUrl, QLocale, QTimer
from PyQt5.QtCore import pyqtProperty, pyqtSignal, pyqtSlot, QObject, QUrl, QLocale, QTimer
from PyQt5.QtGui import QGuiApplication
from PyQt5.QtQml import qmlRegisterType, QQmlComponent, QQmlApplicationEngine
from PyQt5.QtQuick import QQuickView
@@ -36,8 +36,9 @@ if TYPE_CHECKING:
from electrum.simple_config import SimpleConfig
from electrum.plugin import Plugins
from .qenetwork import QENetwork, QEDaemon, QEWalletListModel
from .qewallet import *
from .qedaemon import QEDaemon, QEWalletListModel
from .qenetwork import QENetwork
from .qewallet import QEWallet
from .qeqr import QEQR
class ElectrumQmlApplication(QGuiApplication):

19
electrum/gui/qml/components/Wallets.qml
View File

@@ -5,13 +5,16 @@ import QtQuick.Controls 2.0
Item {
property string title: 'Wallets'
anchors.fill: parent
ListView {
width: parent.width
height: 200
model: Daemon.activeWallets
height: parent.height
model: Daemon.availableWallets
delegate: Item {
width: ListView.view.width
height: 50
RowLayout {
x: 20
@@ -22,9 +25,17 @@ Item {
}
Label {
font.pointSize: 13
text: model.display
font.pointSize: model.active ? 14 : 13
font.bold: model.active
text: model.name
Layout.fillWidth: true
}
}
MouseArea {
anchors.fill: parent
onClicked: openMenu()
}
}
}

115
electrum/gui/qml/qedaemon.py Normal file
View File

@@ -0,0 +1,115 @@
import os
from PyQt5.QtCore import pyqtProperty, pyqtSignal, pyqtSlot, QObject, QUrl
from PyQt5.QtCore import Qt, QAbstractListModel, QModelIndex
from electrum.util import register_callback, get_new_wallet_name
from electrum.logging import get_logger
from electrum.wallet import Wallet, Abstract_Wallet
from .qewallet import QEWallet
# wallet list model. supports both wallet basenames (wallet file basenames)
# and whole Wallet instances (loaded wallets)
class QEWalletListModel(QAbstractListModel):
_logger = get_logger(__name__)
def __init__(self, parent=None):
QAbstractListModel.__init__(self, parent)
self.wallets = []
# define listmodel rolemap
_ROLE_NAMES= ('name','path','active')
_ROLE_KEYS = range(Qt.UserRole + 1, Qt.UserRole + 1 + len(_ROLE_NAMES))
_ROLE_MAP = dict(zip(_ROLE_KEYS, [bytearray(x.encode()) for x in _ROLE_NAMES]))
def rowCount(self, index):
return len(self.wallets)
def roleNames(self):
return self._ROLE_MAP
def data(self, index, role):
role_index = role - (Qt.UserRole + 1)
value = tx[self._ROLE_NAMES[role_index]]
if isinstance(value, bool) or isinstance(value, list) or isinstance(value, int) or value is None:
return value
if isinstance(value, Satoshis):
return value.value
return str(value)
def data(self, index, role):
(wallet_name, wallet_path, wallet) = self.wallets[index.row()]
role_index = role - (Qt.UserRole + 1)
role_name = self._ROLE_NAMES[role_index]
if role_name == 'name':
return wallet_name
if role_name == 'path':
return wallet_name
if role_name == 'active':
return wallet != None
def add_wallet(self, wallet_path = None, wallet: Abstract_Wallet = None):
if wallet_path == None and wallet == None:
return
self.beginInsertRows(QModelIndex(), len(self.wallets), len(self.wallets));
if wallet == None:
wallet_name = os.path.basename(wallet_path)
else:
wallet_name = wallet.basename()
item = (wallet_name, wallet_path, wallet)
self.wallets.append(item);
self.endInsertRows();
class QEDaemon(QObject):
def __init__(self, daemon, parent=None):
super().__init__(parent)
self.daemon = daemon
_logger = get_logger(__name__)
_loaded_wallets = QEWalletListModel()
walletLoaded = pyqtSignal()
walletRequiresPassword = pyqtSignal()
@pyqtSlot()
def load_wallet(self, path=None, password=None):
if path == None:
path = self.daemon.config.get('gui_last_wallet')
wallet = self.daemon.load_wallet(path, password)
if wallet != None:
self._loaded_wallets.add_wallet(wallet=wallet)
self._current_wallet = QEWallet(wallet)
self.walletLoaded.emit()
else:
self._logger.info('fail open wallet')
self.walletRequiresPassword.emit()
@pyqtProperty(QEWallet,notify=walletLoaded)
def currentWallet(self):
return self._current_wallet
@pyqtProperty('QString',notify=walletLoaded)
def walletName(self):
return self._current_wallet.wallet.basename()
@pyqtProperty(QEWalletListModel)
def activeWallets(self):
return self._loaded_wallets
@pyqtProperty(QEWalletListModel)
def availableWallets(self):
available = []
availableListModel = QEWalletListModel(self)
wallet_folder = os.path.dirname(self.daemon.config.get_wallet_path())
with os.scandir(wallet_folder) as it:
for i in it:
if i.is_file():
available.append(i.path)
for path in sorted(available):
wallet = self.daemon.get_wallet(path)
if wallet != None:
availableListModel.add_wallet(wallet_path = wallet.storage.path, wallet = wallet)
else:
availableListModel.add_wallet(wallet_path = path)
return availableListModel

63
electrum/gui/qml/qenetwork.py
View File

@@ -1,11 +1,7 @@
from PyQt5.QtCore import pyqtProperty, pyqtSignal, pyqtSlot, QObject, QUrl
from PyQt5.QtCore import Qt, QAbstractListModel, QModelIndex
from PyQt5.QtCore import pyqtProperty, pyqtSignal, pyqtSlot, QObject
from electrum.util import register_callback
from electrum.logging import get_logger
from electrum.wallet import Wallet, Abstract_Wallet
from .qewallet import QEWallet
class QENetwork(QObject):
def __init__(self, network, parent=None):
@@ -69,60 +65,3 @@ class QENetwork(QObject):
def status(self):
return self._status
class QEWalletListModel(QAbstractListModel):
def __init__(self, parent=None):
QAbstractListModel.__init__(self, parent)
self.wallets = []
def rowCount(self, index):
return len(self.wallets)
def data(self, index, role):
if role == Qt.DisplayRole:
return self.wallets[index.row()].basename()
def add_wallet(self, wallet: Abstract_Wallet = None):
if wallet == None:
return
self.beginInsertRows(QModelIndex(), len(self.wallets), len(self.wallets));
self.wallets.append(wallet);
self.endInsertRows();
class QEDaemon(QObject):
def __init__(self, daemon, parent=None):
super().__init__(parent)
self.daemon = daemon
_logger = get_logger(__name__)
_wallet = ''
_loaded_wallets = QEWalletListModel()
wallet_loaded = pyqtSignal()
@pyqtSlot()
def load_wallet(self, path=None, password=None):
self._logger.info(str(self.daemon.get_wallets()))
if path == None:
path = self.daemon.config.get('recently_open')[0]
wallet = self.daemon.load_wallet(path, password)
if wallet != None:
self._loaded_wallets.add_wallet(wallet)
self._wallet = wallet.basename()
self._current_wallet = QEWallet(wallet)
self.wallet_loaded.emit()
self._logger.info(str(self.daemon.get_wallets()))
else:
self._logger.info('fail open wallet')
@pyqtProperty('QString',notify=wallet_loaded)
def walletName(self):
return self._wallet
@pyqtProperty(QEWalletListModel)
def activeWallets(self):
return self._loaded_wallets
@pyqtProperty(QEWallet,notify=wallet_loaded)
def currentWallet(self):
return self._current_wallet