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

fixed lint issue

Browse Source
This commit is contained in:
Supragya Raj 2021-10-27 15:47:49 +02:00
parent 2f1d8262ae
commit c8413130ec
1 changed files with 34 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
mev_inspect/block.py
View File

@ -1,8 +1,8 @@
from pathlib import Path from pathlib import Path
from typing import List, Optional from typing import List, Optional
import json
import asyncio import asyncio
import aiohttp import aiohttp
import json
from sqlalchemy import orm from sqlalchemy import orm
from web3 import Web3 from web3 import Web3
@ -46,7 +46,9 @@ def _fetch_block(
block_json = w3.eth.get_block(block_number) block_json = w3.eth.get_block(block_number)
if not geth: if not geth:
receipts_json = base_provider.make_request("eth_getBlockReceipts", [block_number]) receipts_json = base_provider.make_request(
"eth_getBlockReceipts", [block_number]
)
traces_json = w3.parity.trace_block(block_number) traces_json = w3.parity.trace_block(block_number)
receipts: List[Receipt] = [ receipts: List[Receipt] = [
@ -56,17 +58,20 @@ def _fetch_block(
base_fee_per_gas = fetch_base_fee_per_gas(w3, block_number) base_fee_per_gas = fetch_base_fee_per_gas(w3, block_number)
else: else:
traces = geth_get_tx_traces_parity_format(base_provider, block_json) traces = geth_get_tx_traces_parity_format(base_provider, block_json)
geth_tx_receipts = geth_get_tx_receipts(base_provider, block_json["transactions"]) geth_tx_receipts = geth_get_tx_receipts(
base_provider, block_json["transactions"]
)
receipts = geth_receipts_translator(block_json, geth_tx_receipts) receipts = geth_receipts_translator(block_json, geth_tx_receipts)
base_fee_per_gas = 0 base_fee_per_gas = 0
return Block( return Block(
block_number=block_number, block_number=block_number,
miner=block_json["miner"], miner=block_json["miner"],
base_fee_per_gas=base_fee_per_gas, base_fee_per_gas=base_fee_per_gas,
traces=traces, traces=traces,
receipts=receipts, receipts=receipts,
) )
def _find_block( def _find_block(
trace_db_session: orm.Session, trace_db_session: orm.Session,
@ -176,20 +181,26 @@ def _get_cache_path(block_number: int) -> Path:
cache_directory_path = Path(cache_directory) cache_directory_path = Path(cache_directory)
return cache_directory_path / f"{block_number}.json" return cache_directory_path / f"{block_number}.json"
# Geth specific additions # Geth specific additions
def geth_get_tx_traces_parity_format(base_provider, block_json): def geth_get_tx_traces_parity_format(base_provider, block_json):
block_hash = block_json['hash'] block_hash = block_json["hash"]
block_trace = geth_get_tx_traces(base_provider, block_hash) block_trace = geth_get_tx_traces(base_provider, block_hash)
parity_traces = [] parity_traces = []
for idx, trace in enumerate(block_trace['result']): for idx, trace in enumerate(block_trace["result"]):
if 'result' in trace: if "result" in trace:
parity_traces.extend(unwrap_tx_trace_for_parity(block_json, idx, trace['result'])) parity_traces.extend(
unwrap_tx_trace_for_parity(block_json, idx, trace["result"])
)
return parity_traces return parity_traces
def geth_get_tx_traces(base_provider, block_hash): def geth_get_tx_traces(base_provider, block_hash):
block_trace = base_provider.make_request("debug_traceBlockByHash", [block_hash.hex(), {"tracer": "callTracer"}]) block_trace = base_provider.make_request(
"debug_traceBlockByHash", [block_hash.hex(), {"tracer": "callTracer"}]
)
return block_trace return block_trace
@ -221,7 +232,7 @@ def unwrap_tx_trace_for_parity(
response_list.append( response_list.append(
Trace( Trace(
action=action_dict, action=action_dict,
block_hash = str(block_json['hash']), block_hash=str(block_json["hash"]),
block_number=int(block_json["number"]), block_number=int(block_json["number"]),
result=result_dict, result=result_dict,
subtraces=len(tx_trace["calls"]) if "calls" in tx_trace.keys() else 0, subtraces=len(tx_trace["calls"]) if "calls" in tx_trace.keys() else 0,
@ -243,6 +254,7 @@ def unwrap_tx_trace_for_parity(
) )
return response_list return response_list
async def geth_get_tx_receipts_task(session, endpoint_uri, tx): async def geth_get_tx_receipts_task(session, endpoint_uri, tx):
data = { data = {
"jsonrpc": "2.0", "jsonrpc": "2.0",
@ -255,6 +267,7 @@ async def geth_get_tx_receipts_task(session, endpoint_uri, tx):
response.raise_for_status() response.raise_for_status()
return await response.text() return await response.text()
async def geth_get_tx_receipts_async(endpoint_uri, transactions): async def geth_get_tx_receipts_async(endpoint_uri, transactions):
geth_tx_receipts = [] geth_tx_receipts = []
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
@ -265,8 +278,11 @@ async def geth_get_tx_receipts_async(endpoint_uri, transactions):
geth_tx_receipts = await asyncio.gather(*tasks) geth_tx_receipts = await asyncio.gather(*tasks)
return [json.loads(tx_receipts) for tx_receipts in geth_tx_receipts] return [json.loads(tx_receipts) for tx_receipts in geth_tx_receipts]
def geth_get_tx_receipts(base_provider, transactions): def geth_get_tx_receipts(base_provider, transactions):
return asyncio.run(geth_get_tx_receipts_async(base_provider.endpoint_uri, transactions)) return asyncio.run(
geth_get_tx_receipts_async(base_provider.endpoint_uri, transactions)
)
def geth_receipts_translator(block_json, geth_tx_receipts) -> List[Receipt]: def geth_receipts_translator(block_json, geth_tx_receipts) -> List[Receipt]:
@ -282,6 +298,7 @@ def geth_receipts_translator(block_json, geth_tx_receipts) -> List[Receipt]:
results.append(unwrap_tx_receipt_for_parity(block_json, idx, tx_receipt)) results.append(unwrap_tx_receipt_for_parity(block_json, idx, tx_receipt))
return results return results
def unwrap_tx_receipt_for_parity(block_json, tx_pos_in_block, tx_receipt) -> Receipt: def unwrap_tx_receipt_for_parity(block_json, tx_pos_in_block, tx_receipt) -> Receipt:
try: try:
if tx_pos_in_block != int(tx_receipt["transactionIndex"], 16): if tx_pos_in_block != int(tx_receipt["transactionIndex"], 16):