darkfi/script/nodetool.py

import asyncio, json, random, sys

class JsonRpc:

    async def start(self, server, port):
        reader, writer = await asyncio.open_connection(server, port)
        self.reader = reader
        self.writer = writer

    async def stop(self):
        self.writer.close()
        await self.writer.wait_closed()

    async def _make_request(self, method, params):
        ident = random.randint(0, 2**32)
        request = {
            "jsonrpc": "2.0",
            "method": method,
            "params": params,
            "id": ident
        }

        message = json.dumps(request) + "\n"
        self.writer.write(message.encode())
        await self.writer.drain()

        data = await self.reader.readline()
        message = data.decode()
        response = json.loads(message)
        return response["result"]

    async def ping(self):
        return await self._make_request("ping", [])

    async def dnet_switch(self, state):
        return await self._make_request("dnet_switch", [state])

    async def dnet_info(self):
        return await self._make_request("dnet_info", [])

async def main(argv):
    rpc = JsonRpc()
    await rpc.start("localhost", 26660)
    await rpc.dnet_switch(True)

    print(await rpc.dnet_info())

    await rpc.dnet_switch(False)
    await rpc.stop()

asyncio.run(main(sys.argv))