Comrad/p2p/p2p.py

import logging
import asyncio


import shelve
from collections import OrderedDict
import pickle,os

# NODES_PRIME = [("128.232.229.63",8468), ("68.66.241.111",8468)]    

NODES_PRIME = [("68.66.241.111",8468)]

async def echo(msg):
    print('echo',msg)    

def boot_selfless_node(port=8468, loop=None):
    handler = logging.StreamHandler()
    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
    handler.setFormatter(formatter)
    log = logging.getLogger('kademlia')
    log.addHandler(handler)
    log.setLevel(logging.INFO)

    if not loop: loop = asyncio.get_event_loop()
    loop.set_debug(True)

    # shelf = HalfForgetfulStorage()
    shelf = None
    print('starting kad server')

    #server = Server(storage=shelf)
    try:
        from kad import KadServer,HalfForgetfulStorage
    except ImportError:
        from .kad import KadServer,HalfForgetfulStorage
    
    server = KadServer(storage=HalfForgetfulStorage())
    loop.create_task(server.listen(port))

    # try:
    #     loop.run_forever()
    # except KeyboardInterrupt:
    #     pass
    # finally:
    #     server.stop()
    #     loop.close()
    return loop


def boot_lonely_selfless_node(port=8467):
    loop = boot_selfless_node(port)
    try:
        loop.run_forever()
    except KeyboardInterrupt:
        pass
    finally:
        server.stop()
        loop.close()