131 lines
4.5 KiB
Python
131 lines
4.5 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
from PyQt5.QtCore import pyqtProperty, pyqtSignal, pyqtSlot, QObject
|
|
from PyQt5.QtCore import Qt, QAbstractListModel, QModelIndex
|
|
|
|
from electrum.logging import get_logger
|
|
from electrum.util import Satoshis, TxMinedInfo
|
|
|
|
from .qetypes import QEAmount
|
|
|
|
class QETransactionListModel(QAbstractListModel):
|
|
def __init__(self, wallet, parent=None):
|
|
super().__init__(parent)
|
|
self.wallet = wallet
|
|
self.tx_history = []
|
|
|
|
_logger = get_logger(__name__)
|
|
|
|
# define listmodel rolemap
|
|
_ROLE_NAMES=('txid','fee_sat','height','confirmations','timestamp','monotonic_timestamp',
|
|
'incoming','bc_value','bc_balance','date','label','txpos_in_block','fee',
|
|
'inputs','outputs','section')
|
|
_ROLE_KEYS = range(Qt.UserRole, Qt.UserRole + len(_ROLE_NAMES))
|
|
_ROLE_MAP = dict(zip(_ROLE_KEYS, [bytearray(x.encode()) for x in _ROLE_NAMES]))
|
|
_ROLE_RMAP = dict(zip(_ROLE_NAMES, _ROLE_KEYS))
|
|
|
|
def rowCount(self, index):
|
|
return len(self.tx_history)
|
|
|
|
def roleNames(self):
|
|
return self._ROLE_MAP
|
|
|
|
def data(self, index, role):
|
|
tx = self.tx_history[index.row()]
|
|
role_index = role - Qt.UserRole
|
|
value = tx[self._ROLE_NAMES[role_index]]
|
|
if isinstance(value, (bool, list, int, str, QEAmount)) or value is None:
|
|
return value
|
|
if isinstance(value, Satoshis):
|
|
return value.value
|
|
if isinstance(value, QEAmount):
|
|
return value
|
|
return str(value)
|
|
|
|
def clear(self):
|
|
self.beginResetModel()
|
|
self.tx_history = []
|
|
self.endResetModel()
|
|
|
|
def tx_to_model(self, tx):
|
|
item = tx
|
|
for output in item['outputs']:
|
|
output['value'] = output['value'].value
|
|
|
|
item['bc_value'] = QEAmount(amount_sat=item['bc_value'].value)
|
|
item['bc_balance'] = QEAmount(amount_sat=item['bc_balance'].value)
|
|
|
|
# newly arriving txs have no (block) timestamp
|
|
# TODO?
|
|
if not item['timestamp']:
|
|
item['timestamp'] = datetime.timestamp(datetime.now())
|
|
|
|
txts = datetime.fromtimestamp(item['timestamp'])
|
|
today = datetime.today().replace(hour=0, minute=0, second=0, microsecond=0)
|
|
|
|
if (txts > today):
|
|
item['section'] = 'today'
|
|
elif (txts > today - timedelta(days=1)):
|
|
item['section'] = 'yesterday'
|
|
elif (txts > today - timedelta(days=7)):
|
|
item['section'] = 'lastweek'
|
|
elif (txts > today - timedelta(days=31)):
|
|
item['section'] = 'lastmonth'
|
|
else:
|
|
item['section'] = 'older'
|
|
|
|
item['date'] = self.format_date_by_section(item['section'], datetime.fromtimestamp(item['timestamp']))
|
|
return item
|
|
|
|
def format_date_by_section(self, section, date):
|
|
#TODO: l10n
|
|
dfmt = {
|
|
'today': '%H:%M:%S',
|
|
'yesterday': '%H:%M:%S',
|
|
'lastweek': '%a, %H:%M:%S',
|
|
'lastmonth': '%a %d, %H:%M:%S',
|
|
'older': '%G-%m-%d %H:%M:%S'
|
|
}
|
|
if section not in dfmt:
|
|
section = 'older'
|
|
return date.strftime(dfmt[section])
|
|
|
|
# initial model data
|
|
def init_model(self):
|
|
history = self.wallet.get_detailed_history(show_addresses = True)
|
|
txs = []
|
|
for tx in history['transactions']:
|
|
txs.append(self.tx_to_model(tx))
|
|
|
|
self.clear()
|
|
self.beginInsertRows(QModelIndex(), 0, len(txs) - 1)
|
|
self.tx_history = txs
|
|
self.tx_history.reverse()
|
|
self.endInsertRows()
|
|
|
|
def update_tx(self, txid, info):
|
|
i = 0
|
|
for tx in self.tx_history:
|
|
if tx['txid'] == txid:
|
|
tx['height'] = info.height
|
|
tx['confirmations'] = info.conf
|
|
tx['timestamp'] = info.timestamp
|
|
tx['date'] = self.format_date_by_section(tx['section'], datetime.fromtimestamp(info.timestamp))
|
|
index = self.index(i,0)
|
|
roles = [self._ROLE_RMAP[x] for x in ['height','confirmations','timestamp','date']]
|
|
self.dataChanged.emit(index, index, roles)
|
|
return
|
|
i = i + 1
|
|
|
|
@pyqtSlot(int)
|
|
def updateBlockchainHeight(self, height):
|
|
self._logger.debug('updating height to %d' % height)
|
|
i = 0
|
|
for tx in self.tx_history:
|
|
if tx['height'] > 0:
|
|
tx['confirmations'] = height - tx['height'] + 1
|
|
index = self.index(i,0)
|
|
roles = [self._ROLE_RMAP['confirmations']]
|
|
self.dataChanged.emit(index, index, roles)
|
|
i = i + 1
|