AaronFiore/electrum
1
0
Fork 0
Code Activity

psbt_nostr: add plugin icon to txdialog button

Browse Source
This commit is contained in:
ThomasV
2025-04-15 15:28:12 +02:00
parent dff1d5b0c1
commit 9d42505eec
1 changed files with 16 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
electrum/plugins/psbt_nostr/qt.py
View File

@@ -34,6 +34,7 @@ from electrum.i18n import _
from electrum.wallet import Multisig_Wallet, Abstract_Wallet from electrum.wallet import Multisig_Wallet, Abstract_Wallet
from electrum.util import UserCancelled, event_listener, EventListener from electrum.util import UserCancelled, event_listener, EventListener
from electrum.gui.qt.transaction_dialog import show_transaction, TxDialog from electrum.gui.qt.transaction_dialog import show_transaction, TxDialog
from electrum.gui.qt.util import read_QIcon_from_bytes
from .psbt_nostr import PsbtNostrPlugin, CosignerWallet from .psbt_nostr import PsbtNostrPlugin, CosignerWallet
@@ -65,13 +66,26 @@ class Plugin(PsbtNostrPlugin):
def transaction_dialog(self, d: 'TxDialog'): def transaction_dialog(self, d: 'TxDialog'):
if cw := self.cosigner_wallets.get(d.wallet): if cw := self.cosigner_wallets.get(d.wallet):
assert isinstance(cw, QtCosignerWallet) assert isinstance(cw, QtCosignerWallet)
cw.hook_transaction_dialog(d) d.cosigner_send_button = b = QPushButton(_("Send to cosigner"))
icon = read_QIcon_from_bytes(self.read_file("nostr_multisig.png"))
b.setIcon(icon)
b.clicked.connect(lambda: cw.send_to_cosigners(d.tx))
d.buttons.insert(0, b)
b.setVisible(False)
@hook @hook
def transaction_dialog_update(self, d: 'TxDialog'): def transaction_dialog_update(self, d: 'TxDialog'):
if cw := self.cosigner_wallets.get(d.wallet): if cw := self.cosigner_wallets.get(d.wallet):
assert isinstance(cw, QtCosignerWallet) assert isinstance(cw, QtCosignerWallet)
cw.hook_transaction_dialog_update(d) if d.tx.is_complete() or d.wallet.can_sign(d.tx):
d.cosigner_send_button.setVisible(False)
return
for xpub, pubkey in cw.cosigner_list:
if cw.cosigner_can_sign(d.tx, xpub):
d.cosigner_send_button.setVisible(True)
break
else:
d.cosigner_send_button.setVisible(False)
class QtCosignerWallet(EventListener, CosignerWallet): class QtCosignerWallet(EventListener, CosignerWallet):
@@ -91,24 +105,6 @@ class QtCosignerWallet(EventListener, CosignerWallet):
if self.wallet == wallet: if self.wallet == wallet:
self.obj.cosignerReceivedPsbt.emit(*args) # put on UI thread via signal self.obj.cosignerReceivedPsbt.emit(*args) # put on UI thread via signal
def hook_transaction_dialog(self, d: 'TxDialog'):
d.cosigner_send_button = b = QPushButton(_("Send to cosigner"))
b.clicked.connect(lambda: self.send_to_cosigners(d.tx))
d.buttons.insert(0, b)
b.setVisible(False)
def hook_transaction_dialog_update(self, d: 'TxDialog'):
assert self.wallet == d.wallet
if d.tx.is_complete() or d.wallet.can_sign(d.tx):
d.cosigner_send_button.setVisible(False)
return
for xpub, pubkey in self.cosigner_list:
if self.cosigner_can_sign(d.tx, xpub):
d.cosigner_send_button.setVisible(True)
break
else:
d.cosigner_send_button.setVisible(False)
def send_to_cosigners(self, tx): def send_to_cosigners(self, tx):
self.add_transaction_to_wallet(tx, on_failure=self.on_add_fail) self.add_transaction_to_wallet(tx, on_failure=self.on_add_fail)
self.send_psbt(tx) self.send_psbt(tx)