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

Trace naming suggestions

Browse Source
This commit is contained in:
Gui Heise 2021-08-27 14:10:17 -04:00
parent 61afdf3f88
commit c1c2a3310e
3 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mev_inspect/classifiers/trace.py
View File

@ -6,8 +6,8 @@ from mev_inspect.schemas.blocks import CallAction, CallResult, Trace, TraceType
from mev_inspect.schemas.classified_traces import ( from mev_inspect.schemas.classified_traces import (
Classification, Classification,
ClassifiedTrace, ClassifiedTrace,
Call, CallTrace,
ClassifiedCall, ClassifiedCallTrace,
ClassifierSpec, ClassifierSpec,
) )
@ -71,7 +71,7 @@ class TraceClassifier:
signature, Classification.unknown signature, Classification.unknown
) )
return ClassifiedCall( return ClassifiedCallTrace(
**trace.dict(), **trace.dict(),
trace_type=trace.type, trace_type=trace.type,
classification=classification, classification=classification,
@ -87,7 +87,7 @@ class TraceClassifier:
gas_used=result.gas_used if result is not None else None, gas_used=result.gas_used if result is not None else None,
) )
return Call( return CallTrace(
**trace.dict(), **trace.dict(),
trace_type=trace.type, trace_type=trace.type,
classification=Classification.unknown, classification=Classification.unknown,

6
mev_inspect/schemas/classified_traces.py
View File

@ -28,7 +28,7 @@ class Protocol(Enum):
curve = "curve" curve = "curve"
# -------- Trace Models ------------------------------------------------------------------------------ # -------- Traces ------------------------------------------------------------------------------
class ClassifiedTrace(Trace): class ClassifiedTrace(Trace):
@ -52,13 +52,13 @@ class ClassifiedTrace(Trace):
abi_name: Optional[str] abi_name: Optional[str]
class Call(ClassifiedTrace): class CallTrace(ClassifiedTrace):
to_address: str to_address: str
from_address: str from_address: str
class ClassifiedCall(Call): class ClassifiedCallTrace(CallTrace):
inputs: Dict[str, Any] inputs: Dict[str, Any]
abi_name: str abi_name: str

8
tests/helpers.py
View File

@ -4,8 +4,8 @@ from mev_inspect.schemas.blocks import TraceType
from mev_inspect.schemas.classified_traces import ( from mev_inspect.schemas.classified_traces import (
Classification, Classification,
ClassifiedTrace, ClassifiedTrace,
Call, CallTrace,
ClassifiedCall, ClassifiedCallTrace,
) )
@ -18,7 +18,7 @@ def make_transfer_trace(
token_address: str, token_address: str,
amount: int, amount: int,
): ):
return Call( return CallTrace(
transaction_hash=transaction_hash, transaction_hash=transaction_hash,
block_number=block_number, block_number=block_number,
type=TraceType.call, type=TraceType.call,
@ -43,7 +43,7 @@ def make_swap_trace(
recipient_address: str, recipient_address: str,
recipient_input_key: str, recipient_input_key: str,
): ):
return ClassifiedCall( return ClassifiedCallTrace(
transaction_hash=transaction_hash, transaction_hash=transaction_hash,
block_number=block_number, block_number=block_number,
type=TraceType.call, type=TraceType.call,