Write miner payments to the DB in script run

This commit is contained in:
Luke Van Seters 2021-08-30 18:02:36 -04:00
parent 9769498c9d
commit 1aa5c5b465

View File

@ -3,7 +3,10 @@ import json
import click import click
from web3 import Web3 from web3 import Web3
from mev_inspect.arbitrages import get_arbitrages
from mev_inspect.block import create_from_block_number from mev_inspect.block import create_from_block_number
from mev_inspect.classifiers.specs import ALL_CLASSIFIER_SPECS
from mev_inspect.classifiers.trace import TraceClassifier
from mev_inspect.crud.arbitrages import ( from mev_inspect.crud.arbitrages import (
delete_arbitrages_for_block, delete_arbitrages_for_block,
write_arbitrages, write_arbitrages,
@ -12,10 +15,10 @@ from mev_inspect.crud.classified_traces import (
delete_classified_traces_for_block, delete_classified_traces_for_block,
write_classified_traces, write_classified_traces,
) )
from mev_inspect.crud.miner_payments import (
from mev_inspect.arbitrages import get_arbitrages delete_miner_payments_for_block,
from mev_inspect.classifiers.specs import ALL_CLASSIFIER_SPECS write_miner_payments,
from mev_inspect.classifiers.trace import TraceClassifier )
from mev_inspect.crud.swaps import delete_swaps_for_block, write_swaps from mev_inspect.crud.swaps import delete_swaps_for_block, write_swaps
from mev_inspect.db import get_session from mev_inspect.db import get_session
from mev_inspect.miner_payments import get_miner_payments from mev_inspect.miner_payments import get_miner_payments
@ -78,10 +81,11 @@ def _inspect_block(
block_number: int, block_number: int,
should_cache: bool, should_cache: bool,
should_print_stats: bool = True, should_print_stats: bool = True,
should_print_miner_payments: bool = True,
should_write_classified_traces: bool = True, should_write_classified_traces: bool = True,
should_write_swaps: bool = True, should_write_swaps: bool = True,
should_write_arbitrages: bool = True, should_write_arbitrages: bool = True,
should_print_miner_payments: bool = True, should_write_miner_payments: bool = True,
): ):
block = create_from_block_number(base_provider, w3, block_number, should_cache) block = create_from_block_number(base_provider, w3, block_number, should_cache)
@ -127,6 +131,10 @@ def _inspect_block(
if should_print_miner_payments: if should_print_miner_payments:
click.echo(json.dumps([p.dict() for p in miner_payments], indent=4)) click.echo(json.dumps([p.dict() for p in miner_payments], indent=4))
if should_write_miner_payments:
delete_miner_payments_for_block(db_session, block_number)
write_miner_payments(db_session, miner_payments)
def get_stats(classified_traces) -> dict: def get_stats(classified_traces) -> dict:
stats: dict = {} stats: dict = {}