Browse Source

qml: fix type hints in qeinvoicelistmodel

master
ThomasV 3 years ago
parent
commit
ebdebd18b4
  1. 16
      electrum/gui/qml/qeinvoicelistmodel.py

16
electrum/gui/qml/qeinvoicelistmodel.py

@ -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)

Loading…
Cancel
Save