From 86861662765b0c71a1a68266695c601063dce87b Mon Sep 17 00:00:00 2001 From: Luke Van Seters Date: Fri, 17 Sep 2021 09:34:06 -0400 Subject: [PATCH] Add writing transfers to inspect block --- mev_inspect/inspect_block.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/mev_inspect/inspect_block.py b/mev_inspect/inspect_block.py index 507ed3b..cbd974e 100644 --- a/mev_inspect/inspect_block.py +++ b/mev_inspect/inspect_block.py @@ -18,8 +18,10 @@ from mev_inspect.crud.miner_payments import ( write_miner_payments, ) from mev_inspect.crud.swaps import delete_swaps_for_block, write_swaps +from mev_inspect.crud.transfers import delete_transfers_for_block, write_transfers from mev_inspect.miner_payments import get_miner_payments from mev_inspect.swaps import get_swaps +from mev_inspect.transfers import get_transfers logger = logging.getLogger(__name__) @@ -33,6 +35,7 @@ def inspect_block( should_cache: bool, should_write_classified_traces: bool = True, should_write_swaps: bool = True, + should_write_transfers: bool = True, should_write_arbitrages: bool = True, should_write_miner_payments: bool = True, ): @@ -53,6 +56,11 @@ def inspect_block( delete_classified_traces_for_block(db_session, block_number) write_classified_traces(db_session, classified_traces) + transfers = get_transfers(classified_traces) + if should_write_transfers: + delete_transfers_for_block(db_session, block_number) + write_transfers(db_session, transfers) + swaps = get_swaps(classified_traces) logger.info(f"Found {len(swaps)} swaps")