From ea41fe04c32e2a1f714675896a205c498aa63e85 Mon Sep 17 00:00:00 2001 From: Luke Van Seters Date: Mon, 16 Aug 2021 15:28:39 -0400 Subject: [PATCH] Add protocol to the stats output of inspect block --- scripts/inspect_block.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/scripts/inspect_block.py b/scripts/inspect_block.py index 278b06a..6ab0726 100644 --- a/scripts/inspect_block.py +++ b/scripts/inspect_block.py @@ -55,16 +55,19 @@ def get_stats(classified_traces) -> dict: stats: dict = {} for trace in classified_traces: + protocol = str(trace.protocol) abi_name = trace.abi_name classification = trace.classification.value signature = trace.function_signature - abi_name_stats = stats.get(abi_name, {}) + protocol_stats = stats.get(protocol, {}) + abi_name_stats = protocol_stats.get(abi_name, {}) class_stats = abi_name_stats.get(classification, {}) signature_count = class_stats.get(signature, 0) class_stats[signature] = signature_count + 1 abi_name_stats[classification] = class_stats - stats[abi_name] = abi_name_stats + protocol_stats[abi_name] = abi_name_stats + stats[protocol] = protocol_stats return stats