You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
313 lines
14 KiB
313 lines
14 KiB
#!/usr/bin/env python |
|
# |
|
# Electrum - lightweight Bitcoin client |
|
# Copyright (C) 2015 Thomas Voegtlin |
|
# |
|
# Permission is hereby granted, free of charge, to any person |
|
# obtaining a copy of this software and associated documentation files |
|
# (the "Software"), to deal in the Software without restriction, |
|
# including without limitation the rights to use, copy, modify, merge, |
|
# publish, distribute, sublicense, and/or sell copies of the Software, |
|
# and to permit persons to whom the Software is furnished to do so, |
|
# subject to the following conditions: |
|
# |
|
# The above copyright notice and this permission notice shall be |
|
# included in all copies or substantial portions of the Software. |
|
# |
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, |
|
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF |
|
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND |
|
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS |
|
# BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN |
|
# ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN |
|
# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
|
# SOFTWARE. |
|
|
|
from enum import IntEnum |
|
|
|
from PyQt5.QtCore import Qt, QPersistentModelIndex, QModelIndex |
|
from PyQt5.QtGui import QStandardItemModel, QStandardItem, QFont |
|
from PyQt5.QtWidgets import QAbstractItemView, QComboBox, QLabel, QMenu |
|
|
|
from electrum.i18n import _ |
|
from electrum.util import block_explorer_URL, profiler |
|
from electrum.plugin import run_hook |
|
from electrum.bitcoin import is_address |
|
from electrum.wallet import InternalAddressCorruption |
|
|
|
from .util import MyTreeView, MONOSPACE_FONT, ColorScheme, webopen, MySortModel |
|
|
|
|
|
class AddressUsageStateFilter(IntEnum): |
|
ALL = 0 |
|
UNUSED = 1 |
|
FUNDED = 2 |
|
USED_AND_EMPTY = 3 |
|
FUNDED_OR_UNUSED = 4 |
|
|
|
def ui_text(self) -> str: |
|
return { |
|
self.ALL: _('All'), |
|
self.UNUSED: _('Unused'), |
|
self.FUNDED: _('Funded'), |
|
self.USED_AND_EMPTY: _('Used'), |
|
self.FUNDED_OR_UNUSED: _('Funded or Unused'), |
|
}[self] |
|
|
|
|
|
class AddressTypeFilter(IntEnum): |
|
ALL = 0 |
|
RECEIVING = 1 |
|
CHANGE = 2 |
|
|
|
def ui_text(self) -> str: |
|
return { |
|
self.ALL: _('All'), |
|
self.RECEIVING: _('Receiving'), |
|
self.CHANGE: _('Change'), |
|
}[self] |
|
|
|
|
|
class AddressList(MyTreeView): |
|
|
|
class Columns(IntEnum): |
|
TYPE = 0 |
|
ADDRESS = 1 |
|
LABEL = 2 |
|
COIN_BALANCE = 3 |
|
FIAT_BALANCE = 4 |
|
NUM_TXS = 5 |
|
|
|
filter_columns = [Columns.TYPE, Columns.ADDRESS, Columns.LABEL, Columns.COIN_BALANCE] |
|
|
|
ROLE_SORT_ORDER = Qt.UserRole + 1000 |
|
ROLE_ADDRESS_STR = Qt.UserRole + 1001 |
|
key_role = ROLE_ADDRESS_STR |
|
|
|
def __init__(self, parent): |
|
super().__init__(parent, self.create_menu, |
|
stretch_column=self.Columns.LABEL, |
|
editable_columns=[self.Columns.LABEL]) |
|
self.wallet = self.parent.wallet |
|
self.setSelectionMode(QAbstractItemView.ExtendedSelection) |
|
self.setSortingEnabled(True) |
|
self.show_change = AddressTypeFilter.ALL # type: AddressTypeFilter |
|
self.show_used = AddressUsageStateFilter.ALL # type: AddressUsageStateFilter |
|
self.change_button = QComboBox(self) |
|
self.change_button.currentIndexChanged.connect(self.toggle_change) |
|
for addr_type in AddressTypeFilter.__members__.values(): # type: AddressTypeFilter |
|
self.change_button.addItem(addr_type.ui_text()) |
|
self.used_button = QComboBox(self) |
|
self.used_button.currentIndexChanged.connect(self.toggle_used) |
|
for addr_usage_state in AddressUsageStateFilter.__members__.values(): # type: AddressUsageStateFilter |
|
self.used_button.addItem(addr_usage_state.ui_text()) |
|
self.std_model = QStandardItemModel(self) |
|
self.proxy = MySortModel(self, sort_role=self.ROLE_SORT_ORDER) |
|
self.proxy.setSourceModel(self.std_model) |
|
self.setModel(self.proxy) |
|
self.update() |
|
self.sortByColumn(self.Columns.TYPE, Qt.AscendingOrder) |
|
|
|
def get_toolbar_buttons(self): |
|
return QLabel(_("Filter:")), self.change_button, self.used_button |
|
|
|
def on_hide_toolbar(self): |
|
self.show_change = AddressTypeFilter.ALL # type: AddressTypeFilter |
|
self.show_used = AddressUsageStateFilter.ALL # type: AddressUsageStateFilter |
|
self.update() |
|
|
|
def save_toolbar_state(self, state, config): |
|
config.set_key('show_toolbar_addresses', state) |
|
|
|
def refresh_headers(self): |
|
fx = self.parent.fx |
|
if fx and fx.get_fiat_address_config(): |
|
ccy = fx.get_currency() |
|
else: |
|
ccy = _('Fiat') |
|
headers = { |
|
self.Columns.TYPE: _('Type'), |
|
self.Columns.ADDRESS: _('Address'), |
|
self.Columns.LABEL: _('Label'), |
|
self.Columns.COIN_BALANCE: _('Balance'), |
|
self.Columns.FIAT_BALANCE: ccy + ' ' + _('Balance'), |
|
self.Columns.NUM_TXS: _('Tx'), |
|
} |
|
self.update_headers(headers) |
|
|
|
def toggle_change(self, state: int): |
|
if state == self.show_change: |
|
return |
|
self.show_change = AddressTypeFilter(state) |
|
self.update() |
|
|
|
def toggle_used(self, state: int): |
|
if state == self.show_used: |
|
return |
|
self.show_used = AddressUsageStateFilter(state) |
|
self.update() |
|
|
|
@profiler |
|
def update(self): |
|
if self.maybe_defer_update(): |
|
return |
|
current_address = self.get_role_data_for_current_item(col=0, role=self.ROLE_ADDRESS_STR) |
|
if self.show_change == AddressTypeFilter.RECEIVING: |
|
addr_list = self.wallet.get_receiving_addresses() |
|
elif self.show_change == AddressTypeFilter.CHANGE: |
|
addr_list = self.wallet.get_change_addresses() |
|
else: |
|
addr_list = self.wallet.get_addresses() |
|
self.proxy.setDynamicSortFilter(False) # temp. disable re-sorting after every change |
|
self.std_model.clear() |
|
self.refresh_headers() |
|
fx = self.parent.fx |
|
set_address = None |
|
self.addresses_beyond_gap_limit = self.wallet.get_all_known_addresses_beyond_gap_limit() |
|
for address in addr_list: |
|
c, u, x = self.wallet.get_addr_balance(address) |
|
balance = c + u + x |
|
is_used_and_empty = self.wallet.adb.is_used(address) and balance == 0 |
|
if self.show_used == AddressUsageStateFilter.UNUSED and (balance or is_used_and_empty): |
|
continue |
|
if self.show_used == AddressUsageStateFilter.FUNDED and balance == 0: |
|
continue |
|
if self.show_used == AddressUsageStateFilter.USED_AND_EMPTY and not is_used_and_empty: |
|
continue |
|
if self.show_used == AddressUsageStateFilter.FUNDED_OR_UNUSED and is_used_and_empty: |
|
continue |
|
labels = ['', address, '', '', '', ''] |
|
address_item = [QStandardItem(e) for e in labels] |
|
# align text and set fonts |
|
for i, item in enumerate(address_item): |
|
item.setTextAlignment(Qt.AlignVCenter) |
|
if i not in (self.Columns.TYPE, self.Columns.LABEL): |
|
item.setFont(QFont(MONOSPACE_FONT)) |
|
self.set_editability(address_item) |
|
address_item[self.Columns.FIAT_BALANCE].setTextAlignment(Qt.AlignRight | Qt.AlignVCenter) |
|
# setup column 0 |
|
if self.wallet.is_change(address): |
|
address_item[self.Columns.TYPE].setText(_('change')) |
|
address_item[self.Columns.TYPE].setBackground(ColorScheme.YELLOW.as_color(True)) |
|
else: |
|
address_item[self.Columns.TYPE].setText(_('receiving')) |
|
address_item[self.Columns.TYPE].setBackground(ColorScheme.GREEN.as_color(True)) |
|
address_item[0].setData(address, self.ROLE_ADDRESS_STR) |
|
address_path = self.wallet.get_address_index(address) |
|
address_item[self.Columns.TYPE].setData(address_path, self.ROLE_SORT_ORDER) |
|
address_path_str = self.wallet.get_address_path_str(address) |
|
if address_path_str is not None: |
|
address_item[self.Columns.TYPE].setToolTip(address_path_str) |
|
# add item |
|
count = self.std_model.rowCount() |
|
self.std_model.insertRow(count, address_item) |
|
self.refresh_row(address, count) |
|
address_idx = self.std_model.index(count, self.Columns.LABEL) |
|
if address == current_address: |
|
set_address = QPersistentModelIndex(address_idx) |
|
self.set_current_idx(set_address) |
|
# show/hide columns |
|
if fx and fx.get_fiat_address_config(): |
|
self.showColumn(self.Columns.FIAT_BALANCE) |
|
else: |
|
self.hideColumn(self.Columns.FIAT_BALANCE) |
|
self.filter() |
|
self.proxy.setDynamicSortFilter(True) |
|
|
|
def refresh_row(self, key, row): |
|
address = key |
|
label = self.wallet.get_label(address) |
|
num = self.wallet.adb.get_address_history_len(address) |
|
c, u, x = self.wallet.get_addr_balance(address) |
|
balance = c + u + x |
|
balance_text = self.parent.format_amount(balance, whitespaces=True) |
|
# create item |
|
fx = self.parent.fx |
|
if fx and fx.get_fiat_address_config(): |
|
rate = fx.exchange_rate() |
|
fiat_balance_str = fx.value_str(balance, rate) |
|
else: |
|
fiat_balance_str = '' |
|
address_item = [self.std_model.item(row, col) for col in self.Columns] |
|
address_item[self.Columns.LABEL].setText(label) |
|
address_item[self.Columns.COIN_BALANCE].setText(balance_text) |
|
address_item[self.Columns.COIN_BALANCE].setData(balance, self.ROLE_SORT_ORDER) |
|
address_item[self.Columns.FIAT_BALANCE].setText(fiat_balance_str) |
|
address_item[self.Columns.NUM_TXS].setText("%d"%num) |
|
c = ColorScheme.BLUE.as_color(True) if self.wallet.is_frozen_address(address) else self._default_bg_brush |
|
address_item[self.Columns.ADDRESS].setBackground(c) |
|
if address in self.addresses_beyond_gap_limit: |
|
address_item[self.Columns.ADDRESS].setBackground(ColorScheme.RED.as_color(True)) |
|
|
|
def create_menu(self, position): |
|
from electrum.wallet import Multisig_Wallet |
|
is_multisig = isinstance(self.wallet, Multisig_Wallet) |
|
can_delete = self.wallet.can_delete_address() |
|
selected = self.selected_in_column(self.Columns.ADDRESS) |
|
if not selected: |
|
return |
|
multi_select = len(selected) > 1 |
|
addrs = [self.item_from_index(item).text() for item in selected] |
|
menu = QMenu() |
|
if not multi_select: |
|
idx = self.indexAt(position) |
|
if not idx.isValid(): |
|
return |
|
item = self.item_from_index(idx) |
|
if not item: |
|
return |
|
addr = addrs[0] |
|
addr_column_title = self.std_model.horizontalHeaderItem(self.Columns.LABEL).text() |
|
addr_idx = idx.sibling(idx.row(), self.Columns.LABEL) |
|
self.add_copy_menu(menu, idx) |
|
menu.addAction(_('Details'), lambda: self.parent.show_address(addr)) |
|
persistent = QPersistentModelIndex(addr_idx) |
|
menu.addAction(_("Edit {}").format(addr_column_title), lambda p=persistent: self.edit(QModelIndex(p))) |
|
#menu.addAction(_("Request payment"), lambda: self.parent.receive_at(addr)) |
|
if self.wallet.can_export(): |
|
menu.addAction(_("Private key"), lambda: self.parent.show_private_key(addr)) |
|
if not is_multisig and not self.wallet.is_watching_only(): |
|
menu.addAction(_("Sign/verify message"), lambda: self.parent.sign_verify_message(addr)) |
|
menu.addAction(_("Encrypt/decrypt message"), lambda: self.parent.encrypt_message(addr)) |
|
if can_delete: |
|
menu.addAction(_("Remove from wallet"), lambda: self.parent.remove_address(addr)) |
|
addr_URL = block_explorer_URL(self.config, 'addr', addr) |
|
if addr_URL: |
|
menu.addAction(_("View on block explorer"), lambda: webopen(addr_URL)) |
|
|
|
if not self.wallet.is_frozen_address(addr): |
|
menu.addAction(_("Freeze"), lambda: self.parent.set_frozen_state_of_addresses([addr], True)) |
|
else: |
|
menu.addAction(_("Unfreeze"), lambda: self.parent.set_frozen_state_of_addresses([addr], False)) |
|
|
|
else: |
|
# multiple items selected |
|
menu.addAction(_("Freeze"), lambda: self.parent.set_frozen_state_of_addresses(addrs, True)) |
|
menu.addAction(_("Unfreeze"), lambda: self.parent.set_frozen_state_of_addresses(addrs, False)) |
|
|
|
coins = self.wallet.get_spendable_coins(addrs) |
|
if coins: |
|
menu.addAction(_("Spend from"), lambda: self.parent.utxo_list.set_spend_list(coins)) |
|
|
|
run_hook('receive_menu', menu, addrs, self.wallet) |
|
menu.exec_(self.viewport().mapToGlobal(position)) |
|
|
|
def place_text_on_clipboard(self, text: str, *, title: str = None) -> None: |
|
if is_address(text): |
|
try: |
|
self.wallet.check_address_for_corruption(text) |
|
except InternalAddressCorruption as e: |
|
self.parent.show_error(str(e)) |
|
raise |
|
super().place_text_on_clipboard(text, title=title) |
|
|
|
def get_edit_key_from_coordinate(self, row, col): |
|
if col != self.Columns.LABEL: |
|
return None |
|
return self.get_role_data_from_coordinate(row, 0, role=self.ROLE_ADDRESS_STR) |
|
|
|
def on_edited(self, idx, edit_key, *, text): |
|
self.parent.wallet.set_label(edit_key, text) |
|
self.parent.history_model.refresh('address label edited') |
|
self.parent.utxo_list.update() |
|
self.parent.update_completions()
|
|
|