crowetic/mev-inspect-py
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Move DB session out

Browse Source
This commit is contained in:
Luke Van Seters 2021-09-13 19:57:31 -04:00
parent 66c22682e8
commit e92c36d30a
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mev_inspect/inspect_block.py
View File

@ -22,6 +22,7 @@ from mev_inspect.swaps import get_swaps
def inspect_block( def inspect_block(
db_session,
base_provider, base_provider,
w3: Web3, w3: Web3,
block_number: int, block_number: int,

7
scripts/inspect_commands.py
View File

@ -1,6 +1,7 @@
import click import click
from web3 import Web3 from web3 import Web3
from mev_inspect.db import get_session
from mev_inspect.inspect_block import inspect_block from mev_inspect.inspect_block import inspect_block
from mev_inspect.retry import http_retry_with_backoff_request_middleware from mev_inspect.retry import http_retry_with_backoff_request_middleware
@ -15,13 +16,14 @@ def cli():
@click.argument("rpc") @click.argument("rpc")
@click.option("--cache/--no-cache", default=True) @click.option("--cache/--no-cache", default=True)
def inspect_block_command(block_number: int, rpc: str, cache: bool): def inspect_block_command(block_number: int, rpc: str, cache: bool):
db_session = get_session()
base_provider = _get_base_provider(rpc) base_provider = _get_base_provider(rpc)
w3 = Web3(base_provider) w3 = Web3(base_provider)
if not cache: if not cache:
click.echo("Skipping cache") click.echo("Skipping cache")
inspect_block(base_provider, w3, block_number, should_cache=cache) inspect_block(db_session, base_provider, w3, block_number, should_cache=cache)
@cli.command() @cli.command()
@ -32,6 +34,8 @@ def inspect_block_command(block_number: int, rpc: str, cache: bool):
def inspect_many_blocks_command( def inspect_many_blocks_command(
after_block: int, before_block: int, rpc: str, cache: bool after_block: int, before_block: int, rpc: str, cache: bool
): ):
db_session = get_session()
base_provider = _get_base_provider(rpc) base_provider = _get_base_provider(rpc)
w3 = Web3(base_provider) w3 = Web3(base_provider)
@ -48,6 +52,7 @@ def inspect_many_blocks_command(
click.echo(dashes) click.echo(dashes)
inspect_block( inspect_block(
db_session,
base_provider, base_provider,
w3, w3,
block_number, block_number,