120 lines
3.9 KiB
Python
120 lines
3.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
import binascii
|
|
from PyQt5 import QtCore, QtWidgets
|
|
from collections import OrderedDict
|
|
import logging
|
|
from electrum.lightning import lightningCall
|
|
|
|
mapping = {0: "r_hash", 1: "pay_req", 2: "settled"}
|
|
revMapp = {"r_hash": 0, "pay_req": 1, "settled": 2}
|
|
datatable = OrderedDict([])
|
|
idx = 0
|
|
|
|
class MyTableRow(QtWidgets.QTreeWidgetItem):
|
|
def __init__(self, di):
|
|
strs = [str(di[mapping[key]]) for key in range(len(mapping))]
|
|
print(strs)
|
|
super(MyTableRow, self).__init__(strs)
|
|
assert isinstance(di, dict)
|
|
self.di = di
|
|
def __getitem__(self, idx):
|
|
return self.di[idx]
|
|
def __setitem__(self, idx, val):
|
|
self.di[idx] = val
|
|
try:
|
|
self.setData(revMapp[idx], QtCore.Qt.DisplayRole, '{0}'.format(val))
|
|
except IndexError:
|
|
logging.warning("Lightning Invoice field %s unknown", idx)
|
|
def __str__(self):
|
|
return str(self.di)
|
|
|
|
def addInvoiceRow(new):
|
|
made = MyTableRow(new)
|
|
datatable[new["r_hash"]] = made
|
|
datatable.move_to_end(new["r_hash"], last=False)
|
|
return made
|
|
|
|
class SatoshiCountSpinBox(QtWidgets.QSpinBox):
|
|
def keyPressEvent(self, e):
|
|
super(SatoshiCountSpinBox, self).keyPressEvent(e)
|
|
if QtCore.Qt.Key_Return == e.key():
|
|
clickHandler(self)
|
|
|
|
def clickHandler(numInput, treeView, lightningRpc):
|
|
print(numInput.value())
|
|
global idx
|
|
obj = {
|
|
"r_hash": binascii.hexlify((int.from_bytes(bytearray.fromhex("9500edb0994b7bc23349193486b25c82097045db641f35fa988c0e849acdec29"), "big")+idx).to_bytes(byteorder="big", length=32)).decode("ascii"),
|
|
"pay_req": "lntb81920n1pdf258s" + str(idx),
|
|
"settled": False
|
|
}
|
|
treeView.insertTopLevelItem(0, addInvoiceRow(obj))
|
|
idx += 1
|
|
lightningCall(lightningRpc, "getinfo")()
|
|
|
|
class LightningInvoiceList(QtWidgets.QWidget):
|
|
def create_menu(self, position):
|
|
menu = QtWidgets.QMenu()
|
|
pay_req = self._tv.currentItem()["pay_req"]
|
|
cb = QtWidgets.QApplication.instance().clipboard()
|
|
def copy():
|
|
print(pay_req)
|
|
cb.setText(pay_req)
|
|
menu.addAction("Copy payment request", copy)
|
|
menu.exec_(self._tv.viewport().mapToGlobal(position))
|
|
"""
|
|
A simple test widget to contain and own the model and table.
|
|
"""
|
|
def __init__(self, parent, lightningWorker, lightningRpc):
|
|
QtWidgets.QWidget.__init__(self, parent)
|
|
|
|
lightningWorker.subscribe(lambda *args: print(args))
|
|
lightningRpc.subscribe(lambda *args: print(args))
|
|
|
|
self._tv=QtWidgets.QTreeWidget(self)
|
|
self._tv.setHeaderLabels([mapping[i] for i in range(len(mapping))])
|
|
self._tv.setColumnCount(len(mapping))
|
|
self._tv.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
|
|
self._tv.customContextMenuRequested.connect(self.create_menu)
|
|
|
|
numInput = SatoshiCountSpinBox(self)
|
|
|
|
button = QtWidgets.QPushButton('Add invoice', self)
|
|
button.clicked.connect(lambda: clickHandler(numInput, self._tv, lightningRpc))
|
|
|
|
l=QtWidgets.QVBoxLayout(self)
|
|
h=QtWidgets.QGridLayout(self)
|
|
h.addWidget(numInput, 0, 0)
|
|
h.addWidget(button, 0, 1)
|
|
#h.addItem(QtWidgets.QSpacerItem(100, 200, QtWidgets.QSizePolicy.Preferred, QtWidgets.QSizePolicy.Preferred), 0, 2)
|
|
#h.setSizePolicy(
|
|
h.setColumnStretch(0, 1)
|
|
h.setColumnStretch(1, 1)
|
|
h.setColumnStretch(2, 2)
|
|
l.addLayout(h)
|
|
l.addWidget(self._tv)
|
|
|
|
self.resize(2500,1000)
|
|
|
|
def tick():
|
|
key = "9500edb0994b7bc23349193486b25c82097045db641f35fa988c0e849acdec29"
|
|
if not key in datatable:
|
|
return
|
|
row = datatable[key]
|
|
row["settled"] = not row["settled"]
|
|
print("data changed")
|
|
|
|
if __name__=="__main__":
|
|
from sys import argv, exit
|
|
|
|
a=QtWidgets.QApplication(argv)
|
|
|
|
w=LightningInvoiceList()
|
|
w.show()
|
|
w.raise_()
|
|
|
|
timer = QtCore.QTimer()
|
|
timer.timeout.connect(tick)
|
|
timer.start(1000)
|
|
exit(a.exec_())
|