diff --git a/scripts/dev_tools.py b/scripts/dev_tools.py index a922f53..d03974a 100644 --- a/scripts/dev_tools.py +++ b/scripts/dev_tools.py @@ -1,29 +1,38 @@ +from subprocess import check_call import click -from subprocess import check_call, run + def lint(): - check_call(['pylint', '.']) + check_call(["pylint", "."]) + def test(): - check_call(['pytest', '--cov=mev_inspect', 'tests']) + check_call(["pytest", "--cov=mev_inspect", "tests"]) + @click.command() -@click.option('-c', required=False, is_flag=True) +@click.option("-c", required=False, is_flag=True) def isort(c: str): - '''if c is present run isort in diff mode''' + """if c is present run isort in diff mode""" if c: - check_call(['isort', '.']) + check_call(["isort", "."]) else: - check_call(['isort', '--diff', '.']) + check_call(["isort", "--diff", "."]) + def mypy(): - check_call(['mypy', '.']) + check_call(["mypy", "."]) + @click.command() -@click.option('-c', required=False, is_flag=True) +@click.option("-c", required=False, is_flag=True) def black(c: str): - '''if c is present run black in diff mode''' + """if c is present run black in diff mode""" if c: - check_call(['black', '.']) + check_call(["black", "."]) else: - check_call(['black', '--diff', '--color', '.']) + check_call(["black", "--diff", "--color", "."]) + + +def pre_commit(): + check_call(["pre-commit", "run", "--all-files"]) diff --git a/scripts/docker.py b/scripts/docker.py index 8da4c39..0776c29 100644 --- a/scripts/docker.py +++ b/scripts/docker.py @@ -1,19 +1,45 @@ +from subprocess import check_call import click -from subprocess import check_call, run -from typing import List + @click.command() -@click.option('-b', required=False, is_flag=True) +@click.option("-b", required=False, is_flag=True) def start(b: str): - '''if d is present background compose''' + """if d is present background compose""" if b: - check_call(['docker', 'compose', 'up', '-d']) - click.echo('docker running in the background...') + check_call(["docker", "compose", "up", "-d"]) + click.echo("docker running in the background...") else: - check_call(['docker', 'compose', 'up']) + check_call(["docker", "compose", "up"]) + def stop(): - check_call(['docker', 'compose', 'down']) + check_call(["docker", "compose", "down"]) + def build(): - check_call(['docker', 'compose', 'build']) + check_call(["docker", "compose", "build"]) + + +def attach(): + check_call(["docker", "exec", "-it", "mev-inspect-py_mev-inspect_1", "bash"]) + + +@click.command() +@click.option("-script", help="inspect script", default="./examples/uniswap_inspect.py") +@click.option("-block_num", help="block number to inspect", default=11931271) +@click.option("-rpc", help="rpc address", default="http://111.11.11.111:8545") +def inspect(script: str, block_num: int, rpc: str): + """Runs mev-inspect scripts through docker services""" + check_call( + [ + "docker", + "compose", + "exec", + "mev-inspect", + "python", + script, + f"-block_number {block_num}", + f"-rpc {rpc}", + ] + )