66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
from functools import partial
|
|
from typing import TYPE_CHECKING
|
|
|
|
from electrum.i18n import _
|
|
from electrum.plugin import hook
|
|
from electrum.wallet import Standard_Wallet, Abstract_Wallet
|
|
|
|
from ..hw_wallet.qt import QtHandlerBase, QtPluginBase
|
|
from ..hw_wallet.plugin import only_hook_if_libraries_available
|
|
from .digitalbitbox import DigitalBitboxPlugin
|
|
from electrum.gui.qt.wizard.wallet import WCScriptAndDerivation, WCHWXPub, WCHWUninitialized, WCHWUnlock
|
|
|
|
if TYPE_CHECKING:
|
|
from electrum.gui.qt.wizard.wallet import QENewWalletWizard
|
|
|
|
|
|
class Plugin(DigitalBitboxPlugin, QtPluginBase):
|
|
icon_unpaired = "digitalbitbox_unpaired.png"
|
|
icon_paired = "digitalbitbox.png"
|
|
|
|
def create_handler(self, window):
|
|
return DigitalBitbox_Handler(window)
|
|
|
|
@only_hook_if_libraries_available
|
|
@hook
|
|
def receive_menu(self, menu, addrs, wallet: Abstract_Wallet):
|
|
if type(wallet) is not Standard_Wallet:
|
|
return
|
|
|
|
keystore = wallet.get_keystore()
|
|
if type(keystore) is not self.keystore_class:
|
|
return
|
|
|
|
if not self.is_mobile_paired():
|
|
return
|
|
|
|
if len(addrs) == 1:
|
|
addr = addrs[0]
|
|
if wallet.get_txin_type(addr) != 'p2pkh':
|
|
return
|
|
def show_address():
|
|
keystore.thread.add(partial(self.show_address, wallet, addr, keystore))
|
|
|
|
menu.addAction(_("Show on {}").format(self.device), show_address)
|
|
|
|
@hook
|
|
def init_wallet_wizard(self, wizard: 'QENewWalletWizard'):
|
|
self.extend_wizard(wizard)
|
|
|
|
# insert digitalbitbox pages in new wallet wizard
|
|
def extend_wizard(self, wizard: 'QENewWalletWizard'):
|
|
super().extend_wizard(wizard)
|
|
views = {
|
|
'dbitbox_start': {'gui': WCScriptAndDerivation},
|
|
'dbitbox_xpub': {'gui': WCHWXPub},
|
|
'dbitbox_not_initialized': {'gui': WCHWUninitialized},
|
|
'dbitbox_unlock': {'gui': WCHWUnlock}
|
|
}
|
|
wizard.navmap_merge(views)
|
|
|
|
|
|
class DigitalBitbox_Handler(QtHandlerBase):
|
|
|
|
def __init__(self, win):
|
|
super(DigitalBitbox_Handler, self).__init__(win, 'Digital Bitbox')
|