Nucypher
/
nucypher
mirror of https://github.com/nucypher/nucypher.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Allow duration as a parameter to make_staking_provider_reservoir function call. This call is used by Porter.

pull/3467/head
derekpierre 2024-04-25 10:15:56 -04:00
parent 110633a518
commit f2f7c24bba
No known key found for this signature in database
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nucypher/policy/reservoir.py
View File

@ -13,6 +13,7 @@ def make_staking_provider_reservoir(
exclude_addresses: Optional[Iterable[ChecksumAddress]] = None,
include_addresses: Optional[Iterable[ChecksumAddress]] = None,
pagination_size: Optional[int] = None,
duration: Optional[int] = 0,
):
"""Get a sampler object containing the currently registered staking providers."""
@ -21,7 +22,9 @@ def make_staking_provider_reservoir(
include_addresses = include_addresses or ()
without_set = set(include_addresses) | set(exclude_addresses or ())
try:
reservoir = application_agent.get_staking_provider_reservoir(without=without_set, pagination_size=pagination_size)
reservoir = application_agent.get_staking_provider_reservoir(
without=without_set, pagination_size=pagination_size, duration=duration
)
except StakerSamplingApplicationAgent.NotEnoughStakingProviders:
# TODO: do that in `get_staking_provider_reservoir()`?
reservoir = StakingProvidersReservoir({})