Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

stream cost estimate fix for uri without size

Browse source
This commit is contained in:
Victor Shyba 2018-10-17 12:11:27 -03:00 committed by Lex Berezhny
parent 8462f56b13
commit e11bae2dda
1 changed files with 10 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
lbrynet/daemon/Daemon.py
View file

@ -433,8 +433,8 @@ class Daemon(AuthJSONRPCServer):
if blob: if blob:
return self.blob_manager.get_blob(blob[0]) return self.blob_manager.get_blob(blob[0])
return download_sd_blob( return download_sd_blob(
sd_hash, self.blob_manager, self.dht_node.peer_finder, self.rate_limiter, sd_hash.decode(), self.blob_manager, self.dht_node.peer_finder, self.rate_limiter,
self.payment_rate_manager, self.wallet_manager, timeout=conf.settings['search_timeout'], self.payment_rate_manager, self.wallet_manager, timeout=conf.settings['peer_search_timeout'],
download_mirrors=conf.settings['download_mirrors'] download_mirrors=conf.settings['download_mirrors']
) )
@ -442,7 +442,7 @@ class Daemon(AuthJSONRPCServer):
"""Return previously downloaded sd blob if already in the blob """Return previously downloaded sd blob if already in the blob
manager, otherwise download and return it manager, otherwise download and return it
""" """
d = self.blob_manager.completed_blobs([sd_hash]) d = self.blob_manager.completed_blobs([sd_hash.decode()])
d.addCallback(self._get_or_download_sd_blob, sd_hash) d.addCallback(self._get_or_download_sd_blob, sd_hash)
return d return d
@ -465,23 +465,19 @@ class Daemon(AuthJSONRPCServer):
return 0.0 return 0.0
return size / (10 ** 6) * conf.settings['data_rate'] return size / (10 ** 6) * conf.settings['data_rate']
@defer.inlineCallbacks async def get_est_cost_using_known_size(self, uri, size):
def get_est_cost_using_known_size(self, uri, size):
""" """
Calculate estimated LBC cost for a stream given its size in bytes Calculate estimated LBC cost for a stream given its size in bytes
""" """
cost = self._get_est_cost_from_stream_size(size) cost = self._get_est_cost_from_stream_size(size)
resolved = yield self.wallet_manager.resolve(uri) resolved = await self.wallet_manager.resolve(uri)
if uri in resolved and 'claim' in resolved[uri]: if uri in resolved and 'claim' in resolved[uri]:
claim = ClaimDict.load_dict(resolved[uri]['claim']['value']) claim = ClaimDict.load_dict(resolved[uri]['claim']['value'])
final_fee = self._add_key_fee_to_est_data_cost(claim.source_fee, cost) final_fee = self._add_key_fee_to_est_data_cost(claim.source_fee, cost)
result = yield self._render_response(final_fee) return final_fee
defer.returnValue(result)
else:
defer.returnValue(None)
def get_est_cost_from_sd_hash(self, sd_hash): def get_est_cost_from_sd_hash(self, sd_hash):
""" """
@ -514,13 +510,12 @@ class Daemon(AuthJSONRPCServer):
fee.amount) fee.amount)
return data_cost + fee_amount return data_cost + fee_amount
@defer.inlineCallbacks async def get_est_cost_from_uri(self, uri):
def get_est_cost_from_uri(self, uri):
""" """
Resolve a name and return the estimated stream cost Resolve a name and return the estimated stream cost
""" """
resolved = yield self.wallet_manager.resolve(uri) resolved = await self.wallet_manager.resolve(uri)
if resolved: if resolved:
claim_response = resolved[uri] claim_response = resolved[uri]
else: else:
@ -530,11 +525,11 @@ class Daemon(AuthJSONRPCServer):
if claim_response and 'claim' in claim_response: if claim_response and 'claim' in claim_response:
if 'value' in claim_response['claim'] and claim_response['claim']['value'] is not None: if 'value' in claim_response['claim'] and claim_response['claim']['value'] is not None:
claim_value = ClaimDict.load_dict(claim_response['claim']['value']) claim_value = ClaimDict.load_dict(claim_response['claim']['value'])
cost = yield self._get_est_cost_from_metadata(claim_value, uri) cost = await d2f(self._get_est_cost_from_metadata(claim_value, uri))
result = round(cost, 5) result = round(cost, 5)
else: else:
log.warning("Failed to estimate cost for %s", uri) log.warning("Failed to estimate cost for %s", uri)
defer.returnValue(result) return result
def get_est_cost(self, uri, size=None): def get_est_cost(self, uri, size=None):
"""Get a cost estimate for a lbry stream, if size is not provided the """Get a cost estimate for a lbry stream, if size is not provided the