chainsyncer/cic_syncer/client/evm/websocket.py

import logging
import uuid
import json

import websocket

from .response import EVMResponse
from cic_syncer.error import RequestError

logg = logging.getLogger()


class EVMWebsocketClient:

    def __init__(self, url):
        self.url = url
        self.conn = websocket.create_connection(url)


    def __del__(self):
        self.conn.close()


    def block_number(self):
        req_id = str(uuid.uuid4())
        req = {
                'jsonrpc': '2.0',
                'method': 'eth_blockNumber',
                'id': str(req_id),
                'params': [],
                }
        self.conn.send(json.dumps(req))
        r = self.conn.recv()
        res = EVMResponse('block_number', json.loads(r))
        err = res.get_error()
        if err != None:
            raise RequestError(err)

        return res.get_result()


    def get_block_by_integer(self, n):
        req_id = str(uuid.uuid4())
        nhx = '0x' + n.to_bytes(8, 'big').hex()
        req = {
                'jsonrpc': '2.0',
                'method': 'eth_getBlockByNumber',
                'id': str(req_id),
                'params': [nhx, False],
                }
        self.conn.send(json.dumps(req))
        r = self.conn.recv()
        res = EVMResponse('get_block', json.loads(r))
        err = res.get_error()
        if err != None:
            raise RequestError(err)

        return res.get_result()