ThomasV
7 years ago
5 changed files with 100 additions and 316 deletions
@ -0,0 +1,76 @@ |
|||||
|
# -*- coding: utf-8 -*- |
||||
|
from PyQt5 import QtCore, QtWidgets |
||||
|
from electrum.util import inv_dict, bh2u |
||||
|
from electrum.i18n import _ |
||||
|
from .util import MyTreeWidget, SortableTreeWidgetItem |
||||
|
|
||||
|
class ChannelsList(MyTreeWidget): |
||||
|
update_rows = QtCore.pyqtSignal(list) |
||||
|
update_single_row = QtCore.pyqtSignal(dict) |
||||
|
|
||||
|
def __init__(self, parent): |
||||
|
MyTreeWidget.__init__(self, parent, self.create_menu, [_('Node ID'), _('Capacity'), _('Balance')], 0) |
||||
|
self.main_window = parent |
||||
|
self.update_rows.connect(self.do_update_rows) |
||||
|
self.update_single_row.connect(self.do_update_single_row) |
||||
|
|
||||
|
def format_fields(self, chan): |
||||
|
return [bh2u(chan.node_id), self.parent.format_amount(chan.constraints.capacity), self.parent.format_amount(chan.local_state.amount_msat//1000)] |
||||
|
|
||||
|
def create_menu(self, position): |
||||
|
menu = QtWidgets.QMenu() |
||||
|
cur = self.currentItem() |
||||
|
def close(): |
||||
|
print("closechannel result", self.parent.network.lnworker.close_channel_from_other_thread(cur.di)) |
||||
|
menu.addAction(_("Close channel"), close) |
||||
|
menu.exec_(self.viewport().mapToGlobal(position)) |
||||
|
|
||||
|
@QtCore.pyqtSlot(dict) |
||||
|
def do_update_single_row(self, chan): |
||||
|
items = self.findItems(chan.channel_id, QtCore.Qt.UserRole|QtCore.Qt.MatchContains|QtCore.Qt.MatchRecursive, column=1) |
||||
|
for item in items: |
||||
|
for i, v in enumerate(self.format_fields(chan)): |
||||
|
item.setData(i, QtCore.Qt.DisplayRole, v) |
||||
|
|
||||
|
@QtCore.pyqtSlot(list) |
||||
|
def do_update_rows(self, channels): |
||||
|
self.clear() |
||||
|
for chan in channels: |
||||
|
item = SortableTreeWidgetItem(self.format_fields(chan)) |
||||
|
item.setData(0, QtCore.Qt.UserRole, chan.channel_id) |
||||
|
self.insertTopLevelItem(0, item) |
||||
|
|
||||
|
def get_toolbar(self): |
||||
|
nodeid_inp = QtWidgets.QLineEdit(self) |
||||
|
local_amt_inp = QtWidgets.QLineEdit(self, text='200000') |
||||
|
push_amt_inp = QtWidgets.QLineEdit(self, text='0') |
||||
|
button = QtWidgets.QPushButton(_('Open channel'), self) |
||||
|
button.clicked.connect(lambda: self.main_window.protect(self.open_channel, (nodeid_inp, local_amt_inp, push_amt_inp))) |
||||
|
l=QtWidgets.QVBoxLayout(self) |
||||
|
h=QtWidgets.QGridLayout(self) |
||||
|
nodeid_label = QtWidgets.QLabel(self) |
||||
|
nodeid_label.setText(_("Node ID")) |
||||
|
local_amt_label = QtWidgets.QLabel(self) |
||||
|
local_amt_label.setText("Local amount (sat)") |
||||
|
push_amt_label = QtWidgets.QLabel(self) |
||||
|
push_amt_label.setText("Push amount (sat)") |
||||
|
h.addWidget(nodeid_label, 0, 0) |
||||
|
h.addWidget(local_amt_label, 0, 1) |
||||
|
h.addWidget(push_amt_label, 0, 2) |
||||
|
h.addWidget(nodeid_inp, 1, 0) |
||||
|
h.addWidget(local_amt_inp, 1, 1) |
||||
|
h.addWidget(push_amt_inp, 1, 2) |
||||
|
h.addWidget(button, 1, 3) |
||||
|
h.setColumnStretch(0, 3) |
||||
|
h.setColumnStretch(1, 1) |
||||
|
h.setColumnStretch(2, 1) |
||||
|
h.setColumnStretch(3, 1) |
||||
|
return h |
||||
|
|
||||
|
def open_channel(self, nodeIdInput, local_amt_inp, push_amt_inp, password): |
||||
|
node_id = str(nodeIdInput.text()) |
||||
|
local_amt = int(local_amt_inp.text()) |
||||
|
push_amt = int(push_amt_inp.text()) |
||||
|
assert local_amt >= 200000 |
||||
|
assert local_amt >= push_amt |
||||
|
obj = self.parent.network.lnworker.open_channel(node_id, local_amt, push_amt, password) |
||||
@ -1,124 +0,0 @@ |
|||||
# -*- coding: utf-8 -*- |
|
||||
import binascii, base64 |
|
||||
from PyQt5 import QtCore, QtWidgets |
|
||||
from collections import OrderedDict |
|
||||
import logging |
|
||||
import traceback |
|
||||
|
|
||||
# https://api.lightning.community/#listchannels |
|
||||
mapping = {0: "chan_id"} |
|
||||
revMapp = {"chan_id": 0} |
|
||||
datatable = OrderedDict([]) |
|
||||
|
|
||||
class MyTableRow(QtWidgets.QTreeWidgetItem): |
|
||||
def __init__(self, di): |
|
||||
strs = [str(di[mapping[key]]) for key in range(len(mapping))] |
|
||||
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 Channel field %s unknown", idx) |
|
||||
def __str__(self): |
|
||||
return str(self.di) |
|
||||
|
|
||||
def addChannelRow(new): |
|
||||
made = MyTableRow(new) |
|
||||
datatable[new["chan_id"]] = made |
|
||||
datatable.move_to_end(new["chan_id"], last=False) |
|
||||
return made |
|
||||
|
|
||||
|
|
||||
class LightningChannelsList(QtWidgets.QWidget): |
|
||||
update_rows = QtCore.pyqtSignal(dict) |
|
||||
update_single_row = QtCore.pyqtSignal(dict) |
|
||||
|
|
||||
def open_channel(self, nodeIdInput, local_amt_inp, push_amt_inp, password): |
|
||||
node_id = str(nodeIdInput.text()) |
|
||||
print("creating channel with {}".format(node_id)) |
|
||||
local_amt = int(local_amt_inp.text()) |
|
||||
push_amt = int(push_amt_inp.text()) |
|
||||
assert local_amt >= 200000 |
|
||||
assert local_amt >= push_amt |
|
||||
obj = self.lnworker.open_channel(node_id, local_amt, push_amt, password) |
|
||||
|
|
||||
def create_menu(self, position): |
|
||||
menu = QtWidgets.QMenu() |
|
||||
cur = self._tv.currentItem() |
|
||||
def close(): |
|
||||
print("closechannel result", lnworker.close_channel_from_other_thread(cur.di)) |
|
||||
menu.addAction("Close channel", close) |
|
||||
menu.exec_(self._tv.viewport().mapToGlobal(position)) |
|
||||
|
|
||||
@QtCore.pyqtSlot(dict) |
|
||||
def do_update_single_row(self, new): |
|
||||
try: |
|
||||
obj = datatable[new["chan_id"]] |
|
||||
except KeyError: |
|
||||
print("lightning chan_id {} unknown!".format(new["chan_id"])) |
|
||||
else: |
|
||||
for k, v in new.items(): |
|
||||
try: |
|
||||
if obj[k] != v: obj[k] = v |
|
||||
except KeyError: |
|
||||
obj[k] = v |
|
||||
|
|
||||
@QtCore.pyqtSlot(dict) |
|
||||
def do_update_rows(self, obj): |
|
||||
self._tv.clear() |
|
||||
for i in obj["channels"]: |
|
||||
self._tv.insertTopLevelItem(0, addChannelRow(i)) |
|
||||
|
|
||||
def __init__(self, parent, lnworker): |
|
||||
QtWidgets.QWidget.__init__(self, parent) |
|
||||
self.main_window = parent |
|
||||
|
|
||||
self.update_rows.connect(self.do_update_rows) |
|
||||
self.update_single_row.connect(self.do_update_single_row) |
|
||||
|
|
||||
self.lnworker = lnworker |
|
||||
lnworker.register_callback(self.update_rows.emit, ['channels_updated']) |
|
||||
lnworker.register_callback(self.update_single_row.emit, ['channel_updated']) |
|
||||
|
|
||||
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) |
|
||||
|
|
||||
nodeid_inp = QtWidgets.QLineEdit(self) |
|
||||
local_amt_inp = QtWidgets.QLineEdit(self, text='200000') |
|
||||
push_amt_inp = QtWidgets.QLineEdit(self, text='0') |
|
||||
button = QtWidgets.QPushButton('Open channel', self) |
|
||||
button.clicked.connect(lambda: self.main_window.protect(self.open_channel, (nodeid_inp, local_amt_inp, push_amt_inp))) |
|
||||
|
|
||||
l=QtWidgets.QVBoxLayout(self) |
|
||||
h=QtWidgets.QGridLayout(self) |
|
||||
nodeid_label = QtWidgets.QLabel(self) |
|
||||
nodeid_label.setText("Node ID") |
|
||||
local_amt_label = QtWidgets.QLabel(self) |
|
||||
local_amt_label.setText("Local amount (sat)") |
|
||||
push_amt_label = QtWidgets.QLabel(self) |
|
||||
push_amt_label.setText("Push amount (sat)") |
|
||||
h.addWidget(nodeid_label, 0, 0) |
|
||||
h.addWidget(local_amt_label, 0, 1) |
|
||||
h.addWidget(push_amt_label, 0, 2) |
|
||||
|
|
||||
h.addWidget(nodeid_inp, 1, 0) |
|
||||
h.addWidget(local_amt_inp, 1, 1) |
|
||||
h.addWidget(push_amt_inp, 1, 2) |
|
||||
h.addWidget(button, 1, 3) |
|
||||
h.setColumnStretch(0, 3) |
|
||||
h.setColumnStretch(1, 1) |
|
||||
h.setColumnStretch(2, 1) |
|
||||
h.setColumnStretch(3, 1) |
|
||||
l.addLayout(h) |
|
||||
l.addWidget(self._tv) |
|
||||
|
|
||||
self.resize(2500,1000) |
|
||||
lnworker.on_channels_updated() |
|
||||
@ -1,153 +0,0 @@ |
|||||
# -*- coding: utf-8 -*- |
|
||||
import base64 |
|
||||
import binascii |
|
||||
from PyQt5 import QtCore, QtWidgets |
|
||||
from collections import OrderedDict |
|
||||
import logging |
|
||||
from .qrcodewidget import QRDialog |
|
||||
from PyQt5.QtCore import pyqtSignal, pyqtSlot |
|
||||
|
|
||||
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 |
|
||||
|
|
||||
class LightningInvoiceList(QtWidgets.QWidget): |
|
||||
invoice_added_signal = QtCore.pyqtSignal(dict) |
|
||||
|
|
||||
@QtCore.pyqtSlot(dict) |
|
||||
def invoice_added_handler(self, di): |
|
||||
self._tv.insertTopLevelItem(0, addInvoiceRow(invoice)) |
|
||||
|
|
||||
def clickHandler(self, numInput, treeView, lnworker): |
|
||||
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 |
|
||||
lnworker.add_invoice(amt) |
|
||||
|
|
||||
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) |
|
||||
def qr(): |
|
||||
d = QRDialog(pay_req, self, "Lightning invoice") |
|
||||
d.exec_() |
|
||||
menu.addAction("Copy payment request", copy) |
|
||||
menu.addAction("Show payment request as QR code", qr) |
|
||||
menu.exec_(self._tv.viewport().mapToGlobal(position)) |
|
||||
|
|
||||
payment_received_signal = pyqtSignal(dict) |
|
||||
|
|
||||
@pyqtSlot(dict) |
|
||||
def paymentReceived(self, new): |
|
||||
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 __init__(self, parent, lnworker): |
|
||||
QtWidgets.QWidget.__init__(self, parent) |
|
||||
|
|
||||
self.payment_received_signal.connect(self.paymentReceived) |
|
||||
self.invoice_added_signal.connect(self.invoice_added_handler) |
|
||||
|
|
||||
#lnworker.subscribe_payment_received_from_other_thread(self.payment_received_signal.emit) |
|
||||
#lnworker.subscribe_invoice_added_from_other_thread(self.invoice_added_signal.emit) |
|
||||
|
|
||||
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(): |
|
||||
self.clickHandler(self2, self._tv, lnworker) |
|
||||
|
|
||||
numInput = SatoshiCountSpinBox(self) |
|
||||
|
|
||||
button = QtWidgets.QPushButton('Add invoice', self) |
|
||||
button.clicked.connect(lambda: self.clickHandler(numInput, self._tv, lnworker)) |
|
||||
|
|
||||
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_()) |
|
||||
Loading…
Reference in new issue