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

Check whether ritual is active or expired before attempting to get ritual public key.

pull/3345/head
derekpierre 2023-11-15 09:44:45 -05:00 committed by Derek Pierre
parent 7e6bafeeb1
commit fb6a3ae800
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nucypher/blockchain/eth/agents.py
View File

@ -931,7 +931,8 @@ class CoordinatorAgent(EthereumContractAgent):
return result
def get_ritual_public_key(self, ritual_id: int) -> DkgPublicKey:
if self.get_ritual_status(ritual_id=ritual_id) != self.Ritual.Status.ACTIVE:
status = self.get_ritual_status(ritual_id=ritual_id)
if status != self.Ritual.Status.ACTIVE and status != self.Ritual.Status.EXPIRED:
# TODO should we raise here instead?
return None

3
tests/mock/coordinator.py
View File

@ -260,7 +260,8 @@ class MockCoordinatorAgent(MockContractAgent):
)
def get_ritual_public_key(self, ritual_id: int) -> DkgPublicKey:
if self.get_ritual_status(ritual_id=ritual_id) != self.RitualStatus.ACTIVE:
status = self.get_ritual_status(ritual_id=ritual_id)
if status != self.Ritual.Status.ACTIVE and status != self.Ritual.Status.EXPIRED:
# TODO should we raise here instead?
return None