forked from LBRYCommunity/lbry-sdk
104 lines
2.7 KiB
Python
104 lines
2.7 KiB
Python
import curses
|
|
import time
|
|
import logging
|
|
from lbrynet.daemon import get_client
|
|
|
|
log = logging.getLogger(__name__)
|
|
log.addHandler(logging.FileHandler("dht contacts.log"))
|
|
# log.addHandler(logging.StreamHandler())
|
|
log.setLevel(logging.INFO)
|
|
stdscr = curses.initscr()
|
|
|
|
api = get_client()
|
|
|
|
|
|
def init_curses():
|
|
curses.noecho()
|
|
curses.cbreak()
|
|
stdscr.nodelay(1)
|
|
stdscr.keypad(1)
|
|
|
|
|
|
def teardown_curses():
|
|
curses.nocbreak()
|
|
stdscr.keypad(0)
|
|
curses.echo()
|
|
curses.endwin()
|
|
|
|
|
|
def refresh(last_contacts, last_blobs):
|
|
height, width = stdscr.getmaxyx()
|
|
|
|
try:
|
|
routing_table_info = api.routing_table_get()
|
|
node_id = routing_table_info['node_id']
|
|
except:
|
|
node_id = "UNKNOWN"
|
|
routing_table_info = {
|
|
'buckets': {},
|
|
'contacts': [],
|
|
'blob_hashes': []
|
|
}
|
|
for y in range(height):
|
|
stdscr.addstr(y, 0, " " * (width - 1))
|
|
|
|
buckets = routing_table_info['buckets']
|
|
stdscr.addstr(0, 0, "node id: %s" % node_id)
|
|
stdscr.addstr(1, 0, "%i buckets, %i contacts, %i blobs" %
|
|
(len(buckets), len(routing_table_info['contacts']),
|
|
len(routing_table_info['blob_hashes'])))
|
|
|
|
y = 3
|
|
for i in sorted(buckets.keys()):
|
|
stdscr.addstr(y, 0, "bucket %s" % i)
|
|
y += 1
|
|
for h in sorted(buckets[i], key=lambda x: x['node_id'].decode('hex')):
|
|
stdscr.addstr(y, 0, '%s (%s:%i) - %i blobs' % (h['node_id'], h['address'], h['port'],
|
|
len(h['blobs'])))
|
|
y += 1
|
|
y += 1
|
|
|
|
new_contacts = set(routing_table_info['contacts']) - last_contacts
|
|
lost_contacts = last_contacts - set(routing_table_info['contacts'])
|
|
|
|
if new_contacts:
|
|
for c in new_contacts:
|
|
log.debug("added contact %s", c)
|
|
if lost_contacts:
|
|
for c in lost_contacts:
|
|
log.info("lost contact %s", c)
|
|
|
|
new_blobs = set(routing_table_info['blob_hashes']) - last_blobs
|
|
lost_blobs = last_blobs - set(routing_table_info['blob_hashes'])
|
|
|
|
if new_blobs:
|
|
for c in new_blobs:
|
|
log.debug("added blob %s", c)
|
|
if lost_blobs:
|
|
for c in lost_blobs:
|
|
log.info("lost blob %s", c)
|
|
|
|
stdscr.addstr(y + 1, 0, str(time.time()))
|
|
stdscr.refresh()
|
|
return set(routing_table_info['contacts']), set(routing_table_info['blob_hashes'])
|
|
|
|
|
|
def do_main():
|
|
c = None
|
|
last_contacts, last_blobs = set(), set()
|
|
while c not in [ord('q'), ord('Q')]:
|
|
last_contacts, last_blobs = refresh(last_contacts, last_blobs)
|
|
c = stdscr.getch()
|
|
time.sleep(0.1)
|
|
|
|
|
|
def main():
|
|
try:
|
|
init_curses()
|
|
do_main()
|
|
finally:
|
|
teardown_curses()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|