removing KeyError cases from _get_bot_address

This commit is contained in:
elicb 2022-05-02 11:37:56 -07:00 committed by Eli Barbieri
parent a3a7cee9c7
commit 6472f0422d

View File

@ -135,7 +135,6 @@ def _get_transfer_info(
None, None,
) )
# This would be cleaner with bisect(), but creates 3.10 dependency
for index, trace in enumerate(classified_traces): for index, trace in enumerate(classified_traces):
if ( if (
mint_slice_start is None mint_slice_start is None
@ -209,8 +208,11 @@ def _get_transfer_info(
def _get_bot_address( def _get_bot_address(
mint_trace: ClassifiedTrace, classified_traces: List[ClassifiedTrace] mint_trace: ClassifiedTrace,
classified_traces: List[ClassifiedTrace],
) -> str: ) -> str:
if "from_address" in mint_trace.dict().keys():
if mint_trace.from_address in LIQUIDITY_MINT_ROUTERS: if mint_trace.from_address in LIQUIDITY_MINT_ROUTERS:
bot_trace = list( bot_trace = list(
filter( filter(
@ -226,8 +228,9 @@ def _get_bot_address(
else: else:
return "0x0000000000000000000000000000000000000000" return "0x0000000000000000000000000000000000000000"
# This case is here because from_address is optional in ClassifiedTrace elif type(mint_trace.from_address) == str:
if type(mint_trace.from_address) == str:
return mint_trace.from_address return mint_trace.from_address
else: else:
return "0x0000000000000000000000000000000000000000" return "0x0000000000000000000000000000000000000000"
else:
return "0x0000000000000000000000000000000000000000"