lbry-sdk/scripts/dht_monitor.py

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()