LBRYFoundation/lbry-sdk
1
0
Fork 0
mirror of https://github.com/LBRYFoundation/lbry-sdk.git synced 2025-08-27 15:31:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

call restore() after claiming name

Browse source
This commit is contained in:
Jack 2016-07-26 21:24:58 -04:00
parent 3e602bce52
commit 29b5aef73a
1 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
lbrynet/lbrynet_daemon/LBRYPublisher.py
View file

@ -43,6 +43,7 @@ class Publisher(object):
self.verified = False self.verified = False
self.lbry_file = None self.lbry_file = None
self.txid = None self.txid = None
self.stream_hash = None
self.metadata = {} self.metadata = {}
def start(self, name, file_path, bid, metadata, old_txid=None): def start(self, name, file_path, bid, metadata, old_txid=None):
@ -63,6 +64,7 @@ class Publisher(object):
d.addCallback(self.add_to_lbry_files) d.addCallback(self.add_to_lbry_files)
d.addCallback(lambda _: self._create_sd_blob()) d.addCallback(lambda _: self._create_sd_blob())
d.addCallback(lambda _: self._claim_name()) d.addCallback(lambda _: self._claim_name())
d.addCallback(lambda _: self.set_status())
d.addCallbacks(lambda _: _show_result(), self._show_publish_error) d.addCallbacks(lambda _: _show_result(), self._show_publish_error)
return d return d
@ -75,16 +77,15 @@ class Publisher(object):
return True return True
return threads.deferToThread(check_file_threaded) return threads.deferToThread(check_file_threaded)
def set_status(self, lbry_file_downloader): def set_lbry_file(self, lbry_file_downloader):
self.lbry_file = lbry_file_downloader self.lbry_file = lbry_file_downloader
d = self.lbry_file_manager.change_lbry_file_status(self.lbry_file, ManagedLBRYFileDownloader.STATUS_FINISHED) return defer.succeed(None)
d.addCallback(lambda _: lbry_file_downloader.restore())
return d
def add_to_lbry_files(self, stream_hash): def add_to_lbry_files(self, stream_hash):
self.stream_hash = stream_hash
prm = PaymentRateManager(self.session.base_payment_rate_manager) prm = PaymentRateManager(self.session.base_payment_rate_manager)
d = self.lbry_file_manager.add_lbry_file(stream_hash, prm) d = self.lbry_file_manager.add_lbry_file(stream_hash, prm)
d.addCallback(self.set_status) d.addCallback(self.set_lbry_file)
return d return d
def _create_sd_blob(self): def _create_sd_blob(self):
@ -99,6 +100,11 @@ class Publisher(object):
d.addCallback(set_sd_hash) d.addCallback(set_sd_hash)
return d return d
def set_status(self):
d = self.lbry_file_manager.change_lbry_file_status(self.lbry_file, ManagedLBRYFileDownloader.STATUS_FINISHED)
d.addCallback(lambda _: self.lbry_file.restore())
return d
def _claim_name(self): def _claim_name(self):
self.metadata['content-type'] = mimetypes.guess_type(os.path.join(self.lbry_file.download_directory, self.metadata['content-type'] = mimetypes.guess_type(os.path.join(self.lbry_file.download_directory,
self.lbry_file.file_name))[0] self.lbry_file.file_name))[0]