Fix get_proxies method (#54)

EthTx · Oct 5, 2021 · 69ecc46 · 69ecc46
1 parent 024e5a8
commit 69ecc46
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -108,7 +108,7 @@ abi_decoded_call: DecodedCall = ethtx.decoders.abi_decoder.decode_call(raw_call,
 from ethtx.models.decoded_model import DecodedTransactionMetadata
 
 # get proxies used in the transaction
-proxies = ethtx.decoders.get_proxies(transaction.root_call)
+proxies = ethtx.decoders.get_proxies(transaction.root_call, chain_id)
 
 # semantically decode transaction components
 decoded_metadata: DecodedTransactionMetadata = ethtx.decoders.semantic_decoder.decode_metadata(block.metadata, transaction.metadata, chain_id)

diff --git a/ethtx/ethtx.py b/ethtx/ethtx.py
@@ -18,6 +18,7 @@
 from .decoders.abi.decoder import ABIDecoder
 from .decoders.decoder_service import DecoderService
 from .decoders.semantic.decoder import SemanticDecoder
+from .models.decoded_model import Proxy, DecodedTransaction
 from .models.objects_model import Call
 from .providers import EtherscanProvider, Web3Provider
 from .providers.semantic_providers.semantics_database import (
@@ -59,16 +60,18 @@ class EthTxDecoders:
 
     def __init__(self, decoder_service: DecoderService):
         self._decoder_service = decoder_service
-        self.abi_decoder = decoder_service.abi_decoder
-        self.semantic_decoder = decoder_service.semantic_decoder
+        self.abi_decoder: ABIDecoder = decoder_service.abi_decoder
+        self.semantic_decoder: SemanticDecoder = decoder_service.semantic_decoder
 
-    def decode_transaction(self, tx_hash: str, chain_id: str = None):
+    def decode_transaction(
+        self, tx_hash: str, chain_id: str = None
+    ) -> DecodedTransaction:
         assert_tx_hash(tx_hash)
         return self._decoder_service.decode_transaction(chain_id, tx_hash)
 
-    def get_proxies(self, call_tree: Call):
+    def get_proxies(self, call_tree: Call, chain_id: str) -> Dict[str, Proxy]:
         delegations = self._decoder_service.get_delegations(call_tree)
-        return self._decoder_service.get_proxies(delegations)
+        return self._decoder_service.get_proxies(delegations, chain_id)
 
 
 class EthTxProviders: