ThomasV
7 years ago
committed by
SomberNight
SomberNight
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