33 lines
834 B
Python
33 lines
834 B
Python
from typing import List
|
|
|
|
from mev_inspect.schemas.classified_traces import ClassifiedTrace
|
|
|
|
|
|
def is_child_trace_address(
|
|
child_trace_address: List[int],
|
|
parent_trace_address: List[int],
|
|
) -> bool:
|
|
parent_trace_length = len(parent_trace_address)
|
|
|
|
return (
|
|
len(child_trace_address) > parent_trace_length
|
|
and child_trace_address[:parent_trace_length] == parent_trace_address
|
|
)
|
|
|
|
|
|
def get_child_traces(
|
|
parent_trace_address: List[int],
|
|
traces: List[ClassifiedTrace],
|
|
) -> List[ClassifiedTrace]:
|
|
ordered_traces = sorted(traces, key=lambda t: t.trace_address)
|
|
child_traces = []
|
|
|
|
for trace in ordered_traces:
|
|
if is_child_trace_address(
|
|
trace.trace_address,
|
|
parent_trace_address,
|
|
):
|
|
child_traces.append(trace)
|
|
|
|
return child_traces
|