forked from LBRYCommunity/lbry-sdk
DRY fix for _handle_claim_result
This commit is contained in:
parent
2728b1ed60
commit
cbd85cd7d8
1 changed files with 52 additions and 84 deletions
|
@ -687,6 +687,39 @@ class Wallet(object):
|
||||||
break
|
break
|
||||||
defer.returnValue(my_claim)
|
defer.returnValue(my_claim)
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def _decode_and_cache_claim_result(self, claim, update_caches):
|
||||||
|
if 'has_signature' in claim and claim['has_signature']:
|
||||||
|
if not claim['signature_is_valid']:
|
||||||
|
log.warning("lbry://%s#%s has an invalid signature",
|
||||||
|
claim['name'], claim['claim_id'])
|
||||||
|
try:
|
||||||
|
decoded = smart_decode(claim['value'])
|
||||||
|
claim_dict = decoded.claim_dict
|
||||||
|
outpoint = ClaimOutpoint(claim['txid'], claim['nout'])
|
||||||
|
name = claim['name']
|
||||||
|
claim['value'] = claim_dict
|
||||||
|
claim['hex'] = decoded.serialized.encode('hex')
|
||||||
|
if update_caches:
|
||||||
|
if decoded.is_stream:
|
||||||
|
yield self._save_name_metadata(name, outpoint, decoded.source_hash)
|
||||||
|
yield self._update_claimid(claim['claim_id'], name, outpoint)
|
||||||
|
yield self._storage.save_claim_to_cache(claim['claim_id'],
|
||||||
|
claim['claim_sequence'],
|
||||||
|
decoded, claim['address'],
|
||||||
|
claim['height'],
|
||||||
|
claim['amount'], claim['supports'],
|
||||||
|
claim.get('channel_name', None),
|
||||||
|
claim.get('signature_is_valid', None))
|
||||||
|
except DecodeError:
|
||||||
|
claim['hex'] = claim['value']
|
||||||
|
claim['value'] = None
|
||||||
|
claim['error'] = "Failed to decode value"
|
||||||
|
|
||||||
|
defer.returnValue(claim)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def _handle_claim_result(self, results, update_caches=True):
|
def _handle_claim_result(self, results, update_caches=True):
|
||||||
if not results:
|
if not results:
|
||||||
|
@ -702,94 +735,29 @@ class Wallet(object):
|
||||||
raise UnknownURI(results['uri'])
|
raise UnknownURI(results['uri'])
|
||||||
raise Exception(results['error'])
|
raise Exception(results['error'])
|
||||||
|
|
||||||
if 'certificate' in results:
|
# case where return value is {'certificate:{'txid', 'value',...}}
|
||||||
try:
|
elif 'certificate' in results:
|
||||||
decoded = smart_decode(results['certificate']['value'])
|
results['certificate'] = yield self._decode_and_cache_claim_result(
|
||||||
claim_dict = decoded.claim_dict
|
results['certificate'],
|
||||||
outpoint = ClaimOutpoint(results['certificate']['txid'],
|
update_caches)
|
||||||
results['certificate']['nout'])
|
|
||||||
name = results['certificate']['name']
|
|
||||||
results['certificate']['value'] = claim_dict
|
|
||||||
results['certificate']['hex'] = decoded.serialized.encode('hex')
|
|
||||||
if update_caches:
|
|
||||||
if decoded.is_stream:
|
|
||||||
yield self._save_name_metadata(name, outpoint, decoded.source_hash)
|
|
||||||
yield self._update_claimid(results['certificate']['claim_id'], name, outpoint)
|
|
||||||
yield self._storage.save_claim_to_cache(results['certificate']['claim_id'],
|
|
||||||
results['certificate']['claim_sequence'],
|
|
||||||
decoded, results['certificate']['address'],
|
|
||||||
results['certificate']['height'],
|
|
||||||
results['certificate']['amount'],
|
|
||||||
results['certificate']['supports'],
|
|
||||||
None,
|
|
||||||
None)
|
|
||||||
except DecodeError:
|
|
||||||
pass
|
|
||||||
|
|
||||||
if 'claim' in results:
|
# case where return value is {'claim':{'txid','value',...}}
|
||||||
claim = results['claim']
|
elif 'claim' in results:
|
||||||
if 'has_signature' in claim and claim['has_signature']:
|
results['claim'] = yield self._decode_and_cache_claim_result(
|
||||||
if not claim['signature_is_valid']:
|
results['claim'],
|
||||||
log.warning("lbry://%s#%s has an invalid signature",
|
update_caches)
|
||||||
claim['name'], claim['claim_id'])
|
|
||||||
try:
|
|
||||||
decoded = smart_decode(claim['value'])
|
|
||||||
claim_dict = decoded.claim_dict
|
|
||||||
outpoint = ClaimOutpoint(claim['txid'], claim['nout'])
|
|
||||||
name = claim['name']
|
|
||||||
claim['value'] = claim_dict
|
|
||||||
claim['hex'] = decoded.serialized.encode('hex')
|
|
||||||
if update_caches:
|
|
||||||
if decoded.is_stream:
|
|
||||||
yield self._save_name_metadata(name, outpoint, decoded.source_hash)
|
|
||||||
yield self._update_claimid(claim['claim_id'], name, outpoint)
|
|
||||||
yield self._storage.save_claim_to_cache(claim['claim_id'],
|
|
||||||
claim['claim_sequence'],
|
|
||||||
decoded, claim['address'],
|
|
||||||
claim['height'],
|
|
||||||
claim['amount'], claim['supports'],
|
|
||||||
claim.get('channel_name', None),
|
|
||||||
claim.get('signature_is_valid', None))
|
|
||||||
except DecodeError:
|
|
||||||
claim['hex'] = claim['value']
|
|
||||||
claim['value'] = None
|
|
||||||
claim['error'] = "Failed to decode value"
|
|
||||||
|
|
||||||
results['claim'] = claim
|
|
||||||
|
|
||||||
|
# case where return value is {'txid','value',...}
|
||||||
|
# returned by queries that are not name resolve related
|
||||||
|
# (getclaimbyoutpoint, getclaimbyid, getclaimsfromtx)
|
||||||
|
# we do not update caches here because it should be missing
|
||||||
|
# some values such as claim_sequence, and supports
|
||||||
elif 'value' in results:
|
elif 'value' in results:
|
||||||
if 'has_signature' in results and results['has_signature']:
|
results = yield self._decode_and_cache_claim_result(results, update_caches=False)
|
||||||
if not results['signature_is_valid']:
|
|
||||||
log.warning("lbry://%s#%s has an invalid signature",
|
|
||||||
results['name'], results['claim_id'])
|
|
||||||
try:
|
|
||||||
decoded = ClaimDict.load_dict(results['value'])
|
|
||||||
claim_dict = decoded.claim_dict
|
|
||||||
claim_hex = decoded.serialized.encode('hex')
|
|
||||||
claim_err = None
|
|
||||||
outpoint = ClaimOutpoint(results['txid'], results['nout'])
|
|
||||||
name = results['name']
|
|
||||||
if update_caches:
|
|
||||||
if decoded.is_stream:
|
|
||||||
yield self._save_name_metadata(name, outpoint, decoded.source_hash)
|
|
||||||
yield self._update_claimid(results['claim_id'], name, outpoint)
|
|
||||||
yield self._storage.save_claim_to_cache(results['claim_id'],
|
|
||||||
results.get('claim_sequence', None),
|
|
||||||
decoded, results['address'],
|
|
||||||
results['height'], results['amount'],
|
|
||||||
results.get('supports', '[]'),
|
|
||||||
results.get('channel_name', None),
|
|
||||||
results.get('signature_is_valid',
|
|
||||||
None))
|
|
||||||
|
|
||||||
except DecodeError:
|
else:
|
||||||
claim_dict = None
|
msg = 'result in unexpected format:{}'.format(results)
|
||||||
claim_hex = results['value']
|
assert False, msg
|
||||||
claim_err = "Failed to decode value"
|
|
||||||
if claim_err:
|
|
||||||
results['error'] = claim_err
|
|
||||||
results['hex'] = claim_hex
|
|
||||||
results['value'] = claim_dict
|
|
||||||
|
|
||||||
defer.returnValue(results)
|
defer.returnValue(results)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue