86 lines
2.4 KiB
Python
86 lines
2.4 KiB
Python
from pathlib import Path
|
|
from typing import List
|
|
|
|
from web3 import Web3
|
|
|
|
from mev_inspect.fees import fetch_base_fee_per_gas
|
|
from mev_inspect.schemas import Block, Trace, TraceType
|
|
from mev_inspect.schemas.receipts import Receipt
|
|
|
|
|
|
cache_directory = "./cache"
|
|
|
|
|
|
def get_latest_block_number(w3: Web3) -> int:
|
|
return w3.eth.get_block("latest")["number"]
|
|
|
|
|
|
def create_from_block_number(
|
|
base_provider, w3: Web3, block_number: int, should_cache: bool
|
|
) -> Block:
|
|
if not should_cache:
|
|
return fetch_block(w3, base_provider, block_number)
|
|
|
|
cache_path = _get_cache_path(block_number)
|
|
|
|
if cache_path.is_file():
|
|
print(f"Cache for block {block_number} exists, " "loading data from cache")
|
|
|
|
return Block.parse_file(cache_path)
|
|
else:
|
|
print(f"Cache for block {block_number} did not exist, getting data")
|
|
|
|
block = fetch_block(w3, base_provider, block_number)
|
|
|
|
cache_block(cache_path, block)
|
|
|
|
return block
|
|
|
|
|
|
def fetch_block(w3, base_provider, block_number: int) -> Block:
|
|
block_json = w3.eth.get_block(block_number)
|
|
receipts_json = base_provider.make_request("eth_getBlockReceipts", [block_number])
|
|
traces_json = w3.parity.trace_block(block_number)
|
|
|
|
receipts: List[Receipt] = [
|
|
Receipt(**receipt) for receipt in receipts_json["result"]
|
|
]
|
|
traces = [Trace(**trace_json) for trace_json in traces_json]
|
|
base_fee_per_gas = fetch_base_fee_per_gas(w3, block_number)
|
|
|
|
return Block(
|
|
block_number=block_number,
|
|
miner=block_json["miner"],
|
|
base_fee_per_gas=base_fee_per_gas,
|
|
traces=traces,
|
|
receipts=receipts,
|
|
)
|
|
|
|
|
|
def get_transaction_hashes(calls: List[Trace]) -> List[str]:
|
|
result = []
|
|
|
|
for call in calls:
|
|
if call.type != TraceType.reward:
|
|
if (
|
|
call.transaction_hash is not None
|
|
and call.transaction_hash not in result
|
|
):
|
|
result.append(call.transaction_hash)
|
|
|
|
return result
|
|
|
|
|
|
def cache_block(cache_path: Path, block: Block):
|
|
write_mode = "w" if cache_path.is_file() else "x"
|
|
|
|
cache_path.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
with open(cache_path, mode=write_mode) as cache_file:
|
|
cache_file.write(block.json())
|
|
|
|
|
|
def _get_cache_path(block_number: int) -> Path:
|
|
cache_directory_path = Path(cache_directory)
|
|
return cache_directory_path / f"{block_number}-new.json"
|