forked from LBRYCommunity/lbry-sdk
return total instead of percent so frontend can display percentage precisely
This commit is contained in:
parent
abebc0d878
commit
daed032bb7
7 changed files with 50 additions and 39 deletions
|
@ -2,7 +2,7 @@ import os
|
||||||
import asyncio
|
import asyncio
|
||||||
import tempfile
|
import tempfile
|
||||||
import multiprocessing as mp
|
import multiprocessing as mp
|
||||||
from typing import List, Optional, Iterable, Iterator, TypeVar, Generic, TYPE_CHECKING, Dict
|
from typing import List, Optional, Iterable, Iterator, TypeVar, Generic, TYPE_CHECKING, Dict, Tuple
|
||||||
from concurrent.futures import Executor, ThreadPoolExecutor, ProcessPoolExecutor
|
from concurrent.futures import Executor, ThreadPoolExecutor, ProcessPoolExecutor
|
||||||
from functools import partial
|
from functools import partial
|
||||||
|
|
||||||
|
@ -288,7 +288,8 @@ class Database:
|
||||||
async def search_supports(self, **constraints) -> Result[Output]:
|
async def search_supports(self, **constraints) -> Result[Output]:
|
||||||
return await self.fetch_result(q.search_supports, **constraints)
|
return await self.fetch_result(q.search_supports, **constraints)
|
||||||
|
|
||||||
async def sum_supports(self, claim_hash, include_channel_content=False, exclude_own_supports=False) -> List[Dict]:
|
async def sum_supports(self, claim_hash, include_channel_content=False, exclude_own_supports=False) \
|
||||||
|
-> Tuple[List[Dict], int]:
|
||||||
return await self.run(q.sum_supports, claim_hash, include_channel_content, exclude_own_supports)
|
return await self.run(q.sum_supports, claim_hash, include_channel_content, exclude_own_supports)
|
||||||
|
|
||||||
async def resolve(self, urls, **kwargs) -> Dict[str, Output]:
|
async def resolve(self, urls, **kwargs) -> Dict[str, Output]:
|
||||||
|
|
|
@ -62,7 +62,7 @@ def search_supports(**constraints) -> Tuple[List[Output], Optional[int]]:
|
||||||
return txos, total
|
return txos, total
|
||||||
|
|
||||||
|
|
||||||
def sum_supports(claim_hash, include_channel_content=False, exclude_own_supports=False) -> List[Dict]:
|
def sum_supports(claim_hash, include_channel_content=False, exclude_own_supports=False) -> Tuple[List[Dict], int]:
|
||||||
supporter = Claim.alias("supporter")
|
supporter = Claim.alias("supporter")
|
||||||
content = Claim.alias("content")
|
content = Claim.alias("content")
|
||||||
where_condition = (content.c.claim_hash == claim_hash)
|
where_condition = (content.c.claim_hash == claim_hash)
|
||||||
|
@ -91,10 +91,7 @@ def sum_supports(claim_hash, include_channel_content=False, exclude_own_supports
|
||||||
|
|
||||||
result = context().fetchall(q)
|
result = context().fetchall(q)
|
||||||
total = sum([row['staked'] for row in result])
|
total = sum([row['staked'] for row in result])
|
||||||
for row in result:
|
return result, total
|
||||||
row['percent'] = round(row['staked']/total*100, 4)
|
|
||||||
|
|
||||||
return result
|
|
||||||
|
|
||||||
|
|
||||||
def search_support_count(**constraints) -> int:
|
def search_support_count(**constraints) -> int:
|
||||||
|
|
|
@ -2594,7 +2594,11 @@ class API:
|
||||||
support sum <claim_id> [--inculde_channel_content]
|
support sum <claim_id> [--inculde_channel_content]
|
||||||
{kwargs}
|
{kwargs}
|
||||||
"""
|
"""
|
||||||
return await self.service.sum_supports(hex_str_to_hash(claim_id), include_channel_content, exclude_own_supports)
|
items, total = await self.service.sum_supports(hex_str_to_hash(claim_id), include_channel_content, exclude_own_supports)
|
||||||
|
return {
|
||||||
|
'items': dict_values_to_lbc(items),
|
||||||
|
'total': dewies_to_lbc(total),
|
||||||
|
}
|
||||||
|
|
||||||
async def support_abandon(
|
async def support_abandon(
|
||||||
self,
|
self,
|
||||||
|
|
|
@ -153,7 +153,7 @@ class Service:
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
async def sum_supports(self, claim_hash: bytes, include_channel_content=False, exclude_own_supports=False) \
|
async def sum_supports(self, claim_hash: bytes, include_channel_content=False, exclude_own_supports=False) \
|
||||||
-> List[Dict]:
|
-> Tuple[List[Dict], int]:
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
async def announce_addresses(self, address_manager, addresses: List[str]):
|
async def announce_addresses(self, address_manager, addresses: List[str]):
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
import logging
|
import logging
|
||||||
from binascii import hexlify, unhexlify
|
from binascii import hexlify, unhexlify
|
||||||
from typing import List, Dict
|
from typing import List, Dict, Tuple
|
||||||
|
|
||||||
from lbry.blockchain.lbrycrd import Lbrycrd
|
from lbry.blockchain.lbrycrd import Lbrycrd
|
||||||
from lbry.blockchain.sync import BlockchainSync
|
from lbry.blockchain.sync import BlockchainSync
|
||||||
|
@ -70,5 +70,5 @@ class FullNode(Service):
|
||||||
return await self.db.protobuf_resolve(urls, **kwargs)
|
return await self.db.protobuf_resolve(urls, **kwargs)
|
||||||
|
|
||||||
async def sum_supports(self, claim_hash: bytes, include_channel_content=False, exclude_own_supports=False) \
|
async def sum_supports(self, claim_hash: bytes, include_channel_content=False, exclude_own_supports=False) \
|
||||||
-> List[Dict]:
|
-> Tuple[List[Dict], int]:
|
||||||
return await self.db.sum_supports(claim_hash, include_channel_content, exclude_own_supports)
|
return await self.db.sum_supports(claim_hash, include_channel_content, exclude_own_supports)
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
import logging
|
import logging
|
||||||
from typing import List, Dict
|
from typing import List, Dict, Tuple
|
||||||
|
|
||||||
from lbry.conf import Config
|
from lbry.conf import Config
|
||||||
from lbry.blockchain import Ledger, Transaction
|
from lbry.blockchain import Ledger, Transaction
|
||||||
|
@ -47,5 +47,5 @@ class LightClient(Service):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
async def sum_supports(self, claim_hash: bytes, include_channel_content=False, exclude_own_supports=False) \
|
async def sum_supports(self, claim_hash: bytes, include_channel_content=False, exclude_own_supports=False) \
|
||||||
-> List[Dict]:
|
-> Tuple[List[Dict], int]:
|
||||||
return await self.client.sum_supports(claim_hash, include_channel_content, exclude_own_supports)
|
return await self.client.sum_supports(claim_hash, include_channel_content, exclude_own_supports)
|
||||||
|
|
|
@ -967,11 +967,12 @@ class TestGeneralBlockchainSync(SyncingBlockchainTestCase):
|
||||||
await self.generate(1)
|
await self.generate(1)
|
||||||
|
|
||||||
# check that supports sum correctly
|
# check that supports sum correctly
|
||||||
results = await self.db.sum_supports(channel_a.claim_hash)
|
results, total = await self.db.sum_supports(channel_a.claim_hash)
|
||||||
self.assertEqual(results, [
|
self.assertEqual(results, [
|
||||||
{'supporter': ch_b.meta['short_url'], 'staked': 1000000000, 'percent': 62.5},
|
{'supporter': ch_b.meta['short_url'], 'staked': 1000000000},
|
||||||
{'supporter': ch_c.meta['short_url'], 'staked': 600000000, 'percent': 37.5},
|
{'supporter': ch_c.meta['short_url'], 'staked': 600000000},
|
||||||
])
|
])
|
||||||
|
self.assertEqual(total, 1600000000)
|
||||||
|
|
||||||
# create a claim in channel A and have channel B support that claim
|
# create a claim in channel A and have channel B support that claim
|
||||||
claim_a = await self.get_claim(await self.create_claim(name="bob", amount='2.0', sign=channel_a))
|
claim_a = await self.get_claim(await self.create_claim(name="bob", amount='2.0', sign=channel_a))
|
||||||
|
@ -979,57 +980,65 @@ class TestGeneralBlockchainSync(SyncingBlockchainTestCase):
|
||||||
await self.generate(1)
|
await self.generate(1)
|
||||||
|
|
||||||
# supports for just the channel claim should be unaffected ...
|
# supports for just the channel claim should be unaffected ...
|
||||||
results = await self.db.sum_supports(channel_a.claim_hash)
|
results, total = await self.db.sum_supports(channel_a.claim_hash)
|
||||||
self.assertEqual(results, [
|
self.assertEqual(results, [
|
||||||
{'supporter': ch_b.meta['short_url'], 'staked': 1000000000, 'percent': 62.5},
|
{'supporter': ch_b.meta['short_url'], 'staked': 1000000000},
|
||||||
{'supporter': ch_c.meta['short_url'], 'staked': 600000000, 'percent': 37.5},
|
{'supporter': ch_c.meta['short_url'], 'staked': 600000000},
|
||||||
])
|
])
|
||||||
|
self.assertEqual(total, 1600000000)
|
||||||
# ... but when you include supports for content in the channel, the support for claim_a is added in
|
# ... but when you include supports for content in the channel, the support for claim_a is added in
|
||||||
results = await self.db.sum_supports(channel_a.claim_hash, include_channel_content=True)
|
results, total = await self.db.sum_supports(channel_a.claim_hash, include_channel_content=True)
|
||||||
self.assertEqual(results, [
|
self.assertEqual(results, [
|
||||||
{'supporter': ch_b.meta['short_url'], 'staked': 1100000000, 'percent': 64.7059},
|
{'supporter': ch_b.meta['short_url'], 'staked': 1100000000},
|
||||||
{'supporter': ch_c.meta['short_url'], 'staked': 600000000, 'percent': 35.2941},
|
{'supporter': ch_c.meta['short_url'], 'staked': 600000000},
|
||||||
])
|
])
|
||||||
|
self.assertEqual(total, 1700000000)
|
||||||
|
|
||||||
# check that sum_supports works as expected for a non-channel claim (with and without including channel content)
|
# check that sum_supports works as expected for a non-channel claim (with and without including channel content)
|
||||||
results = await self.db.sum_supports(claim_a.claim_hash, include_channel_content=False)
|
results, total = await self.db.sum_supports(claim_a.claim_hash, include_channel_content=False)
|
||||||
self.assertEqual(results, [{'supporter': ch_b.meta['short_url'], 'staked': 100000000, 'percent': 100}])
|
self.assertEqual(results, [{'supporter': ch_b.meta['short_url'], 'staked': 100000000}])
|
||||||
results = await self.db.sum_supports(claim_a.claim_hash, include_channel_content=True)
|
self.assertEqual(total, 100000000)
|
||||||
self.assertEqual(results, [{'supporter': ch_b.meta['short_url'], 'staked': 100000000, 'percent': 100}])
|
results, total = await self.db.sum_supports(claim_a.claim_hash, include_channel_content=True)
|
||||||
|
self.assertEqual(results, [{'supporter': ch_b.meta['short_url'], 'staked': 100000000}])
|
||||||
|
self.assertEqual(total, 100000000)
|
||||||
|
|
||||||
# if a support is abandoned, it stops counting
|
# if a support is abandoned, it stops counting
|
||||||
await self.abandon_support(support_b)
|
await self.abandon_support(support_b)
|
||||||
await self.generate(1)
|
await self.generate(1)
|
||||||
results = await self.db.sum_supports(channel_a.claim_hash)
|
results, total = await self.db.sum_supports(channel_a.claim_hash)
|
||||||
self.assertEqual(results, [
|
self.assertEqual(results, [
|
||||||
{'supporter': ch_c.meta['short_url'], 'staked': 600000000, 'percent': 54.5455},
|
{'supporter': ch_c.meta['short_url'], 'staked': 600000000},
|
||||||
{'supporter': ch_b.meta['short_url'], 'staked': 500000000, 'percent': 45.4545},
|
{'supporter': ch_b.meta['short_url'], 'staked': 500000000},
|
||||||
])
|
])
|
||||||
|
self.assertEqual(total, 1100000000)
|
||||||
|
|
||||||
# but if a creator unlocks a tip, that still counts as the tipping channel's contribution
|
# but if a creator unlocks a tip, that still counts as the tipping channel's contribution
|
||||||
await self.abandon_support(tip_b)
|
await self.abandon_support(tip_b)
|
||||||
await self.abandon_support(tip_c)
|
await self.abandon_support(tip_c)
|
||||||
await self.generate(1)
|
await self.generate(1)
|
||||||
results = await self.db.sum_supports(channel_a.claim_hash)
|
results, total = await self.db.sum_supports(channel_a.claim_hash)
|
||||||
self.assertEqual(results, [
|
self.assertEqual(results, [
|
||||||
{'supporter': ch_c.meta['short_url'], 'staked': 600000000, 'percent': 54.5455},
|
{'supporter': ch_c.meta['short_url'], 'staked': 600000000},
|
||||||
{'supporter': ch_b.meta['short_url'], 'staked': 500000000, 'percent': 45.4545},
|
{'supporter': ch_b.meta['short_url'], 'staked': 500000000},
|
||||||
])
|
])
|
||||||
|
self.assertEqual(total, 1100000000)
|
||||||
|
|
||||||
# a channel's own supports don't count if you exclude them
|
# a channel's own supports don't count if you exclude them
|
||||||
await self.support_claim(channel_a, '10.0', sign=channel_a)
|
await self.support_claim(channel_a, '10.0', sign=channel_a)
|
||||||
await self.generate(1)
|
await self.generate(1)
|
||||||
results = await self.db.sum_supports(channel_a.claim_hash, exclude_own_supports=False)
|
results, total = await self.db.sum_supports(channel_a.claim_hash, exclude_own_supports=False)
|
||||||
self.assertEqual(results, [
|
self.assertEqual(results, [
|
||||||
{'supporter': ch_a.meta['short_url'], 'staked': 1000000000, 'percent': 47.6190},
|
{'supporter': ch_a.meta['short_url'], 'staked': 1000000000},
|
||||||
{'supporter': ch_c.meta['short_url'], 'staked': 600000000, 'percent': 28.5714},
|
{'supporter': ch_c.meta['short_url'], 'staked': 600000000},
|
||||||
{'supporter': ch_b.meta['short_url'], 'staked': 500000000, 'percent': 23.8095},
|
{'supporter': ch_b.meta['short_url'], 'staked': 500000000},
|
||||||
])
|
])
|
||||||
results = await self.db.sum_supports(channel_a.claim_hash, exclude_own_supports=True)
|
self.assertEqual(total, 2100000000)
|
||||||
|
results, total = await self.db.sum_supports(channel_a.claim_hash, exclude_own_supports=True)
|
||||||
self.assertEqual(results, [
|
self.assertEqual(results, [
|
||||||
{'supporter': ch_c.meta['short_url'], 'staked': 600000000, 'percent': 54.5455},
|
{'supporter': ch_c.meta['short_url'], 'staked': 600000000},
|
||||||
{'supporter': ch_b.meta['short_url'], 'staked': 500000000, 'percent': 45.4545},
|
{'supporter': ch_b.meta['short_url'], 'staked': 500000000},
|
||||||
])
|
])
|
||||||
|
self.assertEqual(total, 1100000000)
|
||||||
|
|
||||||
async def test_meta_fields_are_translated_to_protobuf(self):
|
async def test_meta_fields_are_translated_to_protobuf(self):
|
||||||
chan_ab = await self.get_claim(
|
chan_ab = await self.get_claim(
|
||||||
|
|
Loading…
Reference in a new issue