lightning: march 2018 rebase, without integration

gui/qt/lightning_invoice_list.py

@@ -0,0 +1,147 @@
+# -*- coding: utf-8 -*-
+import base64
+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):
+        if "settled" not in di:
+            di["settled"] = False
+        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 KeyError:
+            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
+
+def clickHandler(numInput, treeView, lightningRpc):
+    amt = numInput.value()
+    if amt < 1:
+        print("value too small")
+        return
+    print("creating invoice with value {}".format(amt))
+    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, "addinvoice")("--amt=" + str(amt))
+
+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))
+    def lightningWorkerHandler(self, sourceClassName, obj):
+        new = {}
+        for k, v in obj.items():
+            try:
+                v = binascii.hexlify(base64.b64decode(v)).decode("ascii")
+            except:
+                pass
+            new[k] = v
+        try:
+            obj = datatable[new["r_hash"]]
+        except KeyError:
+            print("lightning payment invoice r_hash {} unknown!".format(new["r_hash"]))
+        else:
+            for k, v in new.items():
+                try:
+                    if obj[k] != v: obj[k] = v
+                except KeyError:
+                    obj[k] = v
+    def lightningRpcHandler(self, methodName, obj):
+        if methodName != "addinvoice":
+            print("ignoring reply {} to {}".format(obj, methodName))
+            return
+        self._tv.insertTopLevelItem(0, addInvoiceRow(obj))
+        
+    def __init__(self, parent, lightningWorker, lightningRpc):
+        QtWidgets.QWidget.__init__(self, parent)
+
+        lightningWorker.subscribe(self.lightningWorkerHandler)
+        lightningRpc.subscribe(self.lightningRpcHandler)
+
+        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)
+
+        class SatoshiCountSpinBox(QtWidgets.QSpinBox):
+            def keyPressEvent(self2, e):
+                super(SatoshiCountSpinBox, self2).keyPressEvent(e)
+                if QtCore.Qt.Key_Return == e.key():
+                    clickHandler(self2, self._tv, lightningRpc)
+
+        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_())