import logging import mimetypes import os from twisted.internet import defer from lbrynet.core import file_utils from lbrynet.file_manager.EncryptedFileCreator import create_lbry_file log = logging.getLogger(__name__) class Publisher(object): def __init__(self, session, lbry_file_manager, wallet, certificate_id): self.session = session self.lbry_file_manager = lbry_file_manager self.wallet = wallet self.certificate_id = certificate_id self.lbry_file = None @defer.inlineCallbacks def create_and_publish_stream(self, name, bid, claim_dict, file_path, claim_address=None, change_address=None): """Create lbry file and make claim""" log.info('Starting publish for %s', name) if not os.path.isfile(file_path): raise Exception("File {} not found".format(file_path)) if os.path.getsize(file_path) == 0: raise Exception("Cannot publish empty file {}".format(file_path)) file_name = os.path.basename(file_path) with file_utils.get_read_handle(file_path) as read_handle: self.lbry_file = yield create_lbry_file(self.session, self.lbry_file_manager, file_name, read_handle) if 'source' not in claim_dict['stream']: claim_dict['stream']['source'] = {} claim_dict['stream']['source']['source'] = self.lbry_file.sd_hash claim_dict['stream']['source']['sourceType'] = 'lbry_sd_hash' claim_dict['stream']['source']['contentType'] = get_content_type(file_path) claim_dict['stream']['source']['version'] = "_0_0_1" # need current version here claim_out = yield self.make_claim(name, bid, claim_dict, claim_address, change_address) # check if we have a file already for this claim (if this is a publish update with a new stream) old_stream_hashes = yield self.session.storage.get_old_stream_hashes_for_claim_id(claim_out['claim_id'], self.lbry_file.stream_hash) if old_stream_hashes: for lbry_file in filter(lambda l: l.stream_hash in old_stream_hashes, list(self.lbry_file_manager.lbry_files)): yield self.lbry_file_manager.delete_lbry_file(lbry_file, delete_file=False) log.info("Removed old stream for claim update: %s", lbry_file.stream_hash) yield self.session.storage.save_content_claim( self.lbry_file.stream_hash, "%s:%i" % (claim_out['txid'], claim_out['nout']) ) defer.returnValue(claim_out) @defer.inlineCallbacks def publish_stream(self, name, bid, claim_dict, stream_hash, claim_address=None, change_address=None): """Make a claim without creating a lbry file""" claim_out = yield self.make_claim(name, bid, claim_dict, claim_address, change_address) if stream_hash: # the stream_hash returned from the db will be None if this isn't a stream we have yield self.session.storage.save_content_claim(stream_hash, "%s:%i" % (claim_out['txid'], claim_out['nout'])) self.lbry_file = [f for f in self.lbry_file_manager.lbry_files if f.stream_hash == stream_hash][0] defer.returnValue(claim_out) @defer.inlineCallbacks def make_claim(self, name, bid, claim_dict, claim_address=None, change_address=None): claim_out = yield self.wallet.claim_name(name, bid, claim_dict, certificate_id=self.certificate_id, claim_address=claim_address, change_address=change_address) defer.returnValue(claim_out) def get_content_type(filename): return mimetypes.guess_type(filename)[0] or 'application/octet-stream'