forked from LBRYCommunity/lbry-sdk
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import logging
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class BlobAvailabilityHandlerFactory:
|
|
# implements(IQueryHandlerFactory)
|
|
|
|
def __init__(self, blob_manager):
|
|
self.blob_manager = blob_manager
|
|
|
|
######### IQueryHandlerFactory #########
|
|
|
|
def build_query_handler(self):
|
|
q_h = BlobAvailabilityHandler(self.blob_manager)
|
|
return q_h
|
|
|
|
def get_primary_query_identifier(self):
|
|
return 'requested_blobs'
|
|
|
|
def get_description(self):
|
|
return "Blob Availability - blobs that are available to be uploaded"
|
|
|
|
|
|
class BlobAvailabilityHandler:
|
|
#implements(IQueryHandler)
|
|
|
|
def __init__(self, blob_manager):
|
|
self.blob_manager = blob_manager
|
|
self.query_identifiers = ['requested_blobs']
|
|
|
|
######### IQueryHandler #########
|
|
|
|
def register_with_request_handler(self, request_handler, peer):
|
|
request_handler.register_query_handler(self, self.query_identifiers)
|
|
|
|
def handle_queries(self, queries):
|
|
if self.query_identifiers[0] in queries:
|
|
log.info("Received the client's list of requested blobs")
|
|
available_blobs = self._get_available_blobs(queries[self.query_identifiers[0]])
|
|
log.debug("available blobs: %s", str(available_blobs))
|
|
return {'available_blobs': available_blobs}
|
|
return {}
|
|
|
|
######### internal #########
|
|
|
|
def _get_available_blobs(self, requested_blobs):
|
|
return self.blob_manager.completed_blobs(requested_blobs)
|