lbry-sdk/lbrynet/blob/reader.py

import logging
from twisted.internet import interfaces
from zope.interface import implements

log = logging.getLogger(__name__)


class HashBlobReader_v0(object):
    """
    This is a class that is only used in StreamBlobDecryptor
    and should be deprecated
    """
    implements(interfaces.IConsumer)

    def __init__(self, write_func):
        self.write_func = write_func

    def registerProducer(self, producer, streaming):
        from twisted.internet import reactor

        self.producer = producer
        self.streaming = streaming
        if self.streaming is False:
            reactor.callLater(0, self.producer.resumeProducing)

    def unregisterProducer(self):
        pass

    def write(self, data):
        from twisted.internet import reactor

        self.write_func(data)
        if self.streaming is False:
            reactor.callLater(0, self.producer.resumeProducing)

class HashBlobReader(object):
    """
    This is a file like reader class that supports
    read(size) and close()
    """
    def __init__(self, file_path, finished_cb):
        self.finished_cb = finished_cb
        self.finished_cb_d = None
        self.read_handle = open(file_path, 'rb')

    def __del__(self):
        self.close()

    def read(self, size=-1):
        return self.read_handle.read(size)

    def close(self):
        # if we've already closed and called finished_cb, do nothing
        if self.finished_cb_d is not None:
            return
        self.read_handle.close()
        self.finished_cb_d = self.finished_cb(self)