#!/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 typing import Sequence, TYPE_CHECKING from PyQt5.QtCore import Qt, QItemSelectionModel from PyQt5.QtGui import QStandardItemModel, QStandardItem from PyQt5.QtWidgets import QAbstractItemView from PyQt5.QtWidgets import QMenu, QVBoxLayout, QTreeWidget, QTreeWidgetItem, QHeaderView from electrum.i18n import _ from electrum.util import format_time from electrum.invoices import Invoice, PR_UNPAID, PR_PAID, PR_INFLIGHT, PR_FAILED from electrum.lnutil import HtlcLog from .util import MyTreeView, read_QIcon, MySortModel, pr_icons from .util import CloseButton, Buttons from .util import WindowModalDialog if TYPE_CHECKING: from .main_window import ElectrumWindow from .send_tab import SendTab ROLE_REQUEST_TYPE = Qt.UserRole ROLE_REQUEST_ID = Qt.UserRole + 1 ROLE_SORT_ORDER = Qt.UserRole + 2 class InvoiceList(MyTreeView): key_role = ROLE_REQUEST_ID class Columns(IntEnum): DATE = 0 DESCRIPTION = 1 AMOUNT = 2 STATUS = 3 headers = { Columns.DATE: _('Date'), Columns.DESCRIPTION: _('Description'), Columns.AMOUNT: _('Amount'), Columns.STATUS: _('Status'), } filter_columns = [Columns.DATE, Columns.DESCRIPTION, Columns.AMOUNT] def __init__(self, send_tab: 'SendTab'): window = send_tab.window super().__init__(window, self.create_menu, stretch_column=self.Columns.DESCRIPTION) self.wallet = window.wallet self.send_tab = send_tab self.std_model = QStandardItemModel(self) self.proxy = MySortModel(self, sort_role=ROLE_SORT_ORDER) self.proxy.setSourceModel(self.std_model) self.setModel(self.proxy) self.setSortingEnabled(True) self.setSelectionMode(QAbstractItemView.ExtendedSelection) def refresh_row(self, key, row): assert row is not None invoice = self.wallet.get_invoice(key) if invoice is None: return model = self.std_model status_item = model.item(row, self.Columns.STATUS) status = self.wallet.get_invoice_status(invoice) status_str = invoice.get_status_str(status) if self.wallet.lnworker: log = self.wallet.lnworker.logs.get(key) if log and status == PR_INFLIGHT: status_str += '... (%d)'%len(log) status_item.setText(status_str) status_item.setIcon(read_QIcon(pr_icons.get(status))) def update(self): # not calling maybe_defer_update() as it interferes with conditional-visibility self.proxy.setDynamicSortFilter(False) # temp. disable re-sorting after every change self.std_model.clear() self.update_headers(self.__class__.headers) for idx, item in enumerate(self.wallet.get_unpaid_invoices()): key = item.get_id() if item.is_lightning(): icon_name = 'lightning.png' else: icon_name = 'bitcoin.png' if item.bip70: icon_name = 'seal.png' status = self.wallet.get_invoice_status(item) status_str = item.get_status_str(status) message = item.message amount = item.get_amount_sat() timestamp = item.time or 0 date_str = format_time(timestamp) if timestamp else _('Unknown') amount_str = self.parent.format_amount(amount, whitespaces=True) labels = [date_str, message, amount_str, status_str] items = [QStandardItem(e) for e in labels] self.set_editability(items) items[self.Columns.DATE].setIcon(read_QIcon(icon_name)) items[self.Columns.STATUS].setIcon(read_QIcon(pr_icons.get(status))) items[self.Columns.DATE].setData(key, role=ROLE_REQUEST_ID) #items[self.Columns.DATE].setData(item.type, role=ROLE_REQUEST_TYPE) items[self.Columns.DATE].setData(timestamp, role=ROLE_SORT_ORDER) self.std_model.insertRow(idx, items) self.filter() self.proxy.setDynamicSortFilter(True) # sort requests by date self.sortByColumn(self.Columns.DATE, Qt.DescendingOrder) self.hide_if_empty() def hide_if_empty(self): b = self.std_model.rowCount() > 0 self.setVisible(b) self.send_tab.invoices_label.setVisible(b) def create_menu(self, position): wallet = self.wallet items = self.selected_in_column(0) if len(items)>1: keys = [item.data(ROLE_REQUEST_ID) for item in items] invoices = [wallet.get_invoice(key) for key in keys] can_batch_pay = all([not i.is_lightning() and wallet.get_invoice_status(i) == PR_UNPAID for i in invoices]) menu = QMenu(self) if can_batch_pay: menu.addAction(_("Batch pay invoices") + "...", lambda: self.send_tab.pay_multiple_invoices(invoices)) menu.addAction(_("Delete invoices"), lambda: self.delete_invoices(keys)) menu.exec_(self.viewport().mapToGlobal(position)) return idx = self.indexAt(position) item = self.item_from_index(idx) item_col0 = self.item_from_index(idx.sibling(idx.row(), self.Columns.DATE)) if not item or not item_col0: return key = item_col0.data(ROLE_REQUEST_ID) invoice = self.wallet.get_invoice(key) menu = QMenu(self) copy_menu = self.add_copy_menu(menu, idx) address = invoice.get_address() if address: copy_menu.addAction(_("Address"), lambda: self.parent.do_copy(invoice.get_address(), title='Bitcoin Address')) if invoice.is_lightning(): menu.addAction(_("Details"), lambda: self.parent.show_lightning_invoice(invoice)) else: menu.addAction(_("Details"), lambda: self.parent.show_onchain_invoice(invoice)) status = wallet.get_invoice_status(invoice) if status == PR_UNPAID: menu.addAction(_("Pay") + "...", lambda: self.send_tab.do_pay_invoice(invoice)) if status == PR_FAILED: menu.addAction(_("Retry"), lambda: self.send_tab.do_pay_invoice(invoice)) if self.wallet.lnworker: log = self.wallet.lnworker.logs.get(key) if log: menu.addAction(_("View log"), lambda: self.show_log(key, log)) menu.addAction(_("Delete"), lambda: self.delete_invoices([key])) menu.exec_(self.viewport().mapToGlobal(position)) def show_log(self, key, log: Sequence[HtlcLog]): d = WindowModalDialog(self, _("Payment log")) d.setMinimumWidth(600) vbox = QVBoxLayout(d) log_w = QTreeWidget() log_w.setHeaderLabels([_('Hops'), _('Channel ID'), _('Message')]) log_w.header().setSectionResizeMode(2, QHeaderView.Stretch) log_w.header().setSectionResizeMode(1, QHeaderView.ResizeToContents) for payment_attempt_log in log: route_str, chan_str, message = payment_attempt_log.formatted_tuple() x = QTreeWidgetItem([route_str, chan_str, message]) log_w.addTopLevelItem(x) vbox.addWidget(log_w) vbox.addLayout(Buttons(CloseButton(d))) d.exec_() def delete_invoices(self, keys): for key in keys: self.wallet.delete_invoice(key, write_to_disk=False) self.delete_item(key) self.wallet.save_db()