Onionr/static-data/default-plugins/torgossip/clientfuncs/__init__.py

from typing import TYPE_CHECKING
import sys
import os

import logger
from blockio import store_block, subprocvalidate
import onionrblocks
sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)))
from commands import GossipCommands

if TYPE_CHECKING:
    from socket import socket


def download_blocks(sock: 'socket', offset: int, block_type: str):
    sock.sendall(
        str(int(GossipCommands.LIST_BLOCKS_BY_TYPE_OFFSET)).encode('utf-8') + str(offset).encode('utf-8') + b',' +
            block_type.encode('utf-8'))
    bl_hashs = sock.recv(600000)
    hash = None
    for i in range(len(bl_hashs)//64):
        hash = bl_hashs[:(i*64) + 64]
        sock.sendall(
            str(int(GossipCommands.GET_BLOCK)).encode('utf-8') + hash)
        bl_content = sock.recv(10**6)
        print('got block', bl_content)