1
0

qml: fix type hints in qeinvoicelistmodel

This commit is contained in:
ThomasV
2023-04-04 10:21:56 +02:00
parent 83ee260ab7
commit ebdebd18b4

View File

@@ -5,7 +5,7 @@ from PyQt5.QtCore import Qt, QAbstractListModel, QModelIndex
from electrum.logging import get_logger from electrum.logging import get_logger
from electrum.util import Satoshis, format_time from electrum.util import Satoshis, format_time
from electrum.invoices import Invoice, PR_EXPIRED, LN_EXPIRY_NEVER from electrum.invoices import BaseInvoice, PR_EXPIRED, LN_EXPIRY_NEVER
from .util import QtEventListener, qt_event_listener, status_update_timer_interval from .util import QtEventListener, qt_event_listener, status_update_timer_interval
from .qetypes import QEAmount from .qetypes import QEAmount
@@ -71,7 +71,7 @@ class QEAbstractInvoiceListModel(QAbstractListModel):
self.set_status_timer() self.set_status_timer()
def add_invoice(self, invoice: Invoice): def add_invoice(self, invoice: BaseInvoice):
# skip if already in list # skip if already in list
key = invoice.get_id() key = invoice.get_id()
for invoice in self.invoices: for invoice in self.invoices:
@@ -122,7 +122,7 @@ class QEAbstractInvoiceListModel(QAbstractListModel):
return return
i = i + 1 i = i + 1
def invoice_to_model(self, invoice: Invoice): def invoice_to_model(self, invoice: BaseInvoice):
item = self.get_invoice_as_dict(invoice) item = self.get_invoice_as_dict(invoice)
item['key'] = invoice.get_id() item['key'] = invoice.get_id()
item['is_lightning'] = invoice.is_lightning() item['is_lightning'] = invoice.is_lightning()
@@ -169,7 +169,7 @@ class QEAbstractInvoiceListModel(QAbstractListModel):
raise Exception('provide impl') raise Exception('provide impl')
@abstractmethod @abstractmethod
def get_invoice_as_dict(self, invoice: Invoice): def get_invoice_as_dict(self, invoice: BaseInvoice):
raise Exception('provide impl') raise Exception('provide impl')
@@ -190,7 +190,7 @@ class QEInvoiceListModel(QEAbstractInvoiceListModel, QtEventListener):
self._logger.debug(f'invoice status update for key {key} to {status}') self._logger.debug(f'invoice status update for key {key} to {status}')
self.updateInvoice(key, status) self.updateInvoice(key, status)
def invoice_to_model(self, invoice: Invoice): def invoice_to_model(self, invoice: BaseInvoice):
item = super().invoice_to_model(invoice) item = super().invoice_to_model(invoice)
item['type'] = 'invoice' item['type'] = 'invoice'
@@ -202,7 +202,7 @@ class QEInvoiceListModel(QEAbstractInvoiceListModel, QtEventListener):
def get_invoice_for_key(self, key: str): def get_invoice_for_key(self, key: str):
return self.wallet.get_invoice(key) return self.wallet.get_invoice(key)
def get_invoice_as_dict(self, invoice: Invoice): def get_invoice_as_dict(self, invoice: BaseInvoice):
return self.wallet.export_invoice(invoice) return self.wallet.export_invoice(invoice)
class QERequestListModel(QEAbstractInvoiceListModel, QtEventListener): class QERequestListModel(QEAbstractInvoiceListModel, QtEventListener):
@@ -222,7 +222,7 @@ class QERequestListModel(QEAbstractInvoiceListModel, QtEventListener):
self._logger.debug(f'request status update for key {key} to {status}') self._logger.debug(f'request status update for key {key} to {status}')
self.updateRequest(key, status) self.updateRequest(key, status)
def invoice_to_model(self, invoice: Invoice): def invoice_to_model(self, invoice: BaseInvoice):
item = super().invoice_to_model(invoice) item = super().invoice_to_model(invoice)
item['type'] = 'request' item['type'] = 'request'
@@ -234,7 +234,7 @@ class QERequestListModel(QEAbstractInvoiceListModel, QtEventListener):
def get_invoice_for_key(self, key: str): def get_invoice_for_key(self, key: str):
return self.wallet.get_request(key) return self.wallet.get_request(key)
def get_invoice_as_dict(self, invoice: Invoice): def get_invoice_as_dict(self, invoice: BaseInvoice):
return self.wallet.export_request(invoice) return self.wallet.export_request(invoice)
@pyqtSlot(str, int) @pyqtSlot(str, int)