LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

use sets for __in constraint values

Browse source
This commit is contained in:
Lex Berezhny 2020-02-12 10:39:25 -05:00
parent dcb1b64696
commit be4515ec9a
1 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
lbry/wallet/server/db/reader.py
View file

@ -266,7 +266,7 @@ def claims_query(cols, for_count=False, **constraints) -> Tuple[str, Dict]:
else: else:
constraints['claim.claim_id__like'] = f'{claim_id[:40]}%' constraints['claim.claim_id__like'] = f'{claim_id[:40]}%'
elif 'claim_ids' in constraints: elif 'claim_ids' in constraints:
constraints['claim.claim_id__in'] = constraints.pop('claim_ids') constraints['claim.claim_id__in'] = set(constraints.pop('claim_ids'))
if 'reposted_claim_id' in constraints: if 'reposted_claim_id' in constraints:
constraints['claim.reposted_claim_hash'] = unhexlify(constraints.pop('reposted_claim_id'))[::-1] constraints['claim.reposted_claim_hash'] = unhexlify(constraints.pop('reposted_claim_id'))[::-1]
@ -282,15 +282,15 @@ def claims_query(cols, for_count=False, **constraints) -> Tuple[str, Dict]:
if 'channel_ids' in constraints: if 'channel_ids' in constraints:
channel_ids = constraints.pop('channel_ids') channel_ids = constraints.pop('channel_ids')
if channel_ids: if channel_ids:
constraints['claim.channel_hash__in'] = [ constraints['claim.channel_hash__in'] = {
unhexlify(cid)[::-1] for cid in channel_ids unhexlify(cid)[::-1] for cid in channel_ids
] }
if 'not_channel_ids' in constraints: if 'not_channel_ids' in constraints:
not_channel_ids = constraints.pop('not_channel_ids') not_channel_ids = constraints.pop('not_channel_ids')
if not_channel_ids: if not_channel_ids:
not_channel_ids_binary = [ not_channel_ids_binary = {
unhexlify(ncid)[::-1] for ncid in not_channel_ids unhexlify(ncid)[::-1] for ncid in not_channel_ids
] }
if constraints.get('has_channel_signature', False): if constraints.get('has_channel_signature', False):
constraints['claim.channel_hash__not_in'] = not_channel_ids_binary constraints['claim.channel_hash__not_in'] = not_channel_ids_binary
else: else:
@ -320,13 +320,13 @@ def claims_query(cols, for_count=False, **constraints) -> Tuple[str, Dict]:
if 'stream_types' in constraints: if 'stream_types' in constraints:
stream_types = constraints.pop('stream_types') stream_types = constraints.pop('stream_types')
if stream_types: if stream_types:
constraints['claim.stream_type__in'] = [ constraints['claim.stream_type__in'] = {
STREAM_TYPES[stream_type] for stream_type in stream_types STREAM_TYPES[stream_type] for stream_type in stream_types
] }
if 'media_types' in constraints: if 'media_types' in constraints:
media_types = constraints.pop('media_types') media_types = constraints.pop('media_types')
if media_types: if media_types:
constraints['claim.media_type__in'] = media_types constraints['claim.media_type__in'] = set(media_types)
if 'fee_currency' in constraints: if 'fee_currency' in constraints:
constraints['claim.fee_currency'] = constraints.pop('fee_currency').lower() constraints['claim.fee_currency'] = constraints.pop('fee_currency').lower()