support overloaded events when finding event ABI

ethereum · Aug 11, 2020 · 7a4f372 · 7a4f372
1 parent b9d7677
commit 7a4f372
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/web3/contract.py b/web3/contract.py
@@ -1122,13 +1122,18 @@ def __init__(self, *argument_names: Tuple[str]) -> None:
  else:
  self.argument_names = argument_names
 
- self.abi = self._get_event_abi()
+ self.abi = self._get_event_abi(argument_names)
 
  @classmethod
- def _get_event_abi(cls) -> ABIEvent:
+ def _get_event_abi(cls, argument_names: Optional[Tuple[str]]) -> ABIEvent:
+ if argument_names is None:
+ return find_matching_event_abi(
+ cls.contract_abi,
+ event_name=cls.event_name)
  return find_matching_event_abi(
  cls.contract_abi,
- event_name=cls.event_name)
+ event_name=cls.event_name,
+ argument_names=argument_names)
 
  @combomethod
  def processReceipt(