862d914061
* refactor: aave_v1 abis * feat: aave_v2 abis * feat: aave_v2 classifiers * chore: rebase feat_aave_v2_internal * feat: discern aave from aave_v2 * feat: discern aave from aave_v3 * fix: correct aave_v3 parsing * refactor: remove superfluous lower operands
94 lines
2.1 KiB
Python
94 lines
2.1 KiB
Python
from enum import Enum
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
from .utils import CamelModel
|
|
|
|
|
|
class TraceType(Enum):
|
|
call = "call"
|
|
create = "create"
|
|
delegate_call = "delegateCall"
|
|
reward = "reward"
|
|
suicide = "suicide"
|
|
|
|
|
|
class Trace(CamelModel):
|
|
action: dict
|
|
block_hash: str
|
|
block_number: int
|
|
result: Optional[dict]
|
|
subtraces: int
|
|
trace_address: List[int]
|
|
transaction_hash: Optional[str]
|
|
transaction_position: Optional[int]
|
|
type: TraceType
|
|
error: Optional[str]
|
|
|
|
|
|
class Classification(Enum):
|
|
unknown = "unknown"
|
|
swap = "swap"
|
|
transfer = "transfer"
|
|
liquidate = "liquidate"
|
|
seize = "seize"
|
|
punk_bid = "punk_bid"
|
|
punk_accept_bid = "punk_accept_bid"
|
|
nft_trade = "nft_trade"
|
|
|
|
|
|
class Protocol(Enum):
|
|
uniswap_v2 = "uniswap_v2"
|
|
uniswap_v3 = "uniswap_v3"
|
|
sushiswap = "sushiswap"
|
|
aave = "aave"
|
|
aave_v2 = "aave_v2"
|
|
aave_v3 = "aave_v3"
|
|
weth = "weth"
|
|
curve = "curve"
|
|
zero_ex = "0x"
|
|
balancer_v1 = "balancer_v1"
|
|
compound_v2 = "compound_v2"
|
|
cream = "cream"
|
|
cryptopunks = "cryptopunks"
|
|
bancor = "bancor"
|
|
opensea = "opensea"
|
|
|
|
|
|
class ClassifiedTrace(Trace):
|
|
classification: Classification
|
|
to_address: Optional[str]
|
|
from_address: Optional[str]
|
|
gas: Optional[int]
|
|
value: Optional[int]
|
|
gas_used: Optional[int]
|
|
transaction_hash: str
|
|
transaction_position: int
|
|
protocol: Optional[Protocol]
|
|
function_name: Optional[str]
|
|
function_signature: Optional[str]
|
|
inputs: Optional[Dict[str, Any]]
|
|
abi_name: Optional[str]
|
|
|
|
class Config:
|
|
validate_assignment = True
|
|
json_encoders = {
|
|
# a little lazy but fine for now
|
|
# this is used for bytes value inputs
|
|
bytes: lambda b: b.hex(),
|
|
}
|
|
|
|
|
|
class CallTrace(ClassifiedTrace):
|
|
to_address: str
|
|
from_address: str
|
|
|
|
|
|
class DecodedCallTrace(CallTrace):
|
|
inputs: Dict[str, Any]
|
|
abi_name: str
|
|
protocol: Optional[Protocol]
|
|
gas: Optional[int]
|
|
gas_used: Optional[int]
|
|
function_name: str
|
|
function_signature: str
|