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

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