mirror of
https://github.com/StijnBTC/Ringtools.git
synced 2025-02-28 07:54:09 +01:00
62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
import os
|
|
import grpc
|
|
from time import sleep
|
|
|
|
from output import format_channel_error, format_error, clear_screen, format_channel
|
|
|
|
LOOP_SLEEP_TIME = 10
|
|
|
|
|
|
class Status:
|
|
def __init__(self, client, output, channels_file, keep_loop, show_fees):
|
|
self.client = client
|
|
self.output = output
|
|
self.channels_file = channels_file
|
|
self.keep_loop = keep_loop
|
|
self.show_fees = show_fees
|
|
|
|
def read_file(self, file):
|
|
if not os.path.isfile(file):
|
|
self.handle_error("File does not exist")
|
|
else:
|
|
with open(self.channels_file) as file:
|
|
return file.read().splitlines()
|
|
|
|
def run(self):
|
|
if self.keep_loop:
|
|
self.loop()
|
|
else:
|
|
self.once()
|
|
|
|
def loop(self):
|
|
while True:
|
|
clear_screen()
|
|
self.once()
|
|
sleep(LOOP_SLEEP_TIME)
|
|
|
|
def once(self):
|
|
channels = self.read_file(self.channels_file)
|
|
for channelID in channels:
|
|
# TODO: Fix this with short channel ids or convert between formats
|
|
# if len(channelID) != 18:
|
|
# continue
|
|
# if not channelID.isnumeric():
|
|
# continue
|
|
|
|
try:
|
|
response = self.client.get_edge(channelID)
|
|
node1 = self.client.get_node(response.node1_pub)
|
|
node2 = self.client.get_node(response.node2_pub)
|
|
|
|
disabled = response.node1_policy.disabled or response.node2_policy.disabled
|
|
self.print_channel(response, node1.alias, node2.alias, disabled)
|
|
except grpc.RpcError as e:
|
|
self.output.print_line(format_channel_error(channelID, e.details()))
|
|
except Exception as error:
|
|
self.output.print_line(format_channel_error(channelID, repr(error)))
|
|
|
|
def print_channel(self, channel, node1_alias, node2_alias, chan_disabled):
|
|
self.output.print_line(format_channel(channel, node1_alias, node2_alias, chan_disabled, self.show_fees))
|
|
|
|
def handle_error(self, error):
|
|
self.output.print_line(format_error(error))
|