forked from LBRYCommunity/lbry-sdk
72 lines
2.9 KiB
Python
72 lines
2.9 KiB
Python
import logging
|
|
import mimetypes
|
|
import os
|
|
|
|
from twisted.internet import defer
|
|
from lbrynet.core import file_utils
|
|
from lbrynet.lbryfilemanager.EncryptedFileCreator import create_lbry_file
|
|
from lbrynet.lbryfile.StreamDescriptor import publish_sd_blob
|
|
from lbrynet.metadata.Metadata import Metadata
|
|
from lbrynet.metadata.Fee import FeeValidator
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class Publisher(object):
|
|
def __init__(self, session, lbry_file_manager, wallet):
|
|
self.session = session
|
|
self.lbry_file_manager = lbry_file_manager
|
|
self.wallet = wallet
|
|
self.lbry_file = None
|
|
|
|
@defer.inlineCallbacks
|
|
def add_fee_to_metadata(self, metadata, fee):
|
|
assert len(fee) == 1, "Too many fees"
|
|
for currency in fee:
|
|
if 'address' not in fee[currency]:
|
|
new_address = yield self.session.wallet.get_new_address()
|
|
fee[currency]['address'] = new_address
|
|
metadata['fee'] = FeeValidator(fee)
|
|
defer.returnValue(metadata)
|
|
|
|
@defer.inlineCallbacks
|
|
def publish_stream(self, name, file_path, bid, metadata):
|
|
log.info('Starting publish for %s', name)
|
|
file_name = os.path.basename(file_path)
|
|
with file_utils.get_read_handle(file_path) as read_handle:
|
|
stream_hash = yield create_lbry_file(self.session, self.lbry_file_manager, file_name,
|
|
read_handle)
|
|
prm = self.session.payment_rate_manager
|
|
self.lbry_file = yield self.lbry_file_manager.add_lbry_file(stream_hash, prm)
|
|
sd_hash = yield publish_sd_blob(self.lbry_file_manager.stream_info_manager,
|
|
self.session.blob_manager, self.lbry_file.stream_hash)
|
|
if 'sources' not in metadata:
|
|
metadata['sources'] = {}
|
|
metadata['sources']['lbry_sd_hash'] = sd_hash
|
|
metadata['content_type'] = get_content_type(file_path)
|
|
metadata['ver'] = Metadata.current_version
|
|
claim_out = yield self.make_claim(name, bid, metadata)
|
|
self.lbry_file.completed = True
|
|
yield self.lbry_file.load_file_attributes()
|
|
yield self.lbry_file.save_status()
|
|
defer.returnValue(claim_out)
|
|
|
|
@defer.inlineCallbacks
|
|
def update_stream(self, name, bid, metadata):
|
|
my_claim = yield self.wallet.get_my_claim(name)
|
|
updated_metadata = my_claim['value']
|
|
for meta_key in metadata:
|
|
updated_metadata[meta_key] = metadata[meta_key]
|
|
claim_out = yield self.make_claim(name, bid, updated_metadata)
|
|
defer.returnValue(claim_out)
|
|
|
|
@defer.inlineCallbacks
|
|
def make_claim(self, name, bid, metadata):
|
|
validated_metadata = Metadata(metadata)
|
|
claim_out = yield self.wallet.claim_name(name, bid, validated_metadata)
|
|
defer.returnValue(claim_out)
|
|
|
|
|
|
def get_content_type(filename):
|
|
return mimetypes.guess_type(filename)[0] or 'application/octet-stream'
|