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

Handle UnknownContract at Agent init-time

pull/379/head
Kieran Prasch 2018-07-20 13:41:45 -07:00 committed by Kieran R. Prasch
parent 61ebd5aa5c
commit 1372323bf0
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,6 +6,7 @@ from constant_sorrow import constants
from nucypher.blockchain.eth import constants
from nucypher.blockchain.eth.chains import Blockchain
from nucypher.blockchain.eth.interfaces import EthereumContractRegistry
class EthereumContractAgent(ABC):
@ -33,9 +34,13 @@ class EthereumContractAgent(ABC):
blockchain = Blockchain.connect()
self.blockchain = blockchain
# Fetch the contract by reading address and abi from the registry and blockchain
contract = self.blockchain.interface.get_contract_by_name(name=self.principal_contract_name,
upgradeable=self._upgradeable)
try:
# Fetch the contract by reading address and abi from the registry and blockchain
contract = self.blockchain.interface.get_contract_by_name(name=self.principal_contract_name,
upgradeable=self._upgradeable)
except EthereumContractRegistry.UnknownContract:
raise self.ContractNotDeployed("There is no registry entry for {}".format(self.principal_contract_name))
self.__contract = contract
super().__init__()