From 2c1632c13dbbcb4268cd9bd6efc92ebeda01f5ee Mon Sep 17 00:00:00 2001 From: hrimlinger Date: Thu, 11 Jan 2024 11:56:24 +0100 Subject: [PATCH] [diamond_miner]: Prefix not found error --- iris/worker/inner_pipeline/diamond_miner.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/iris/worker/inner_pipeline/diamond_miner.py b/iris/worker/inner_pipeline/diamond_miner.py index 0a2310d3..a55630dd 100644 --- a/iris/worker/inner_pipeline/diamond_miner.py +++ b/iris/worker/inner_pipeline/diamond_miner.py @@ -101,8 +101,14 @@ async def diamond_miner_inner_pipeline( prefix = f"{addr_v4}/{tool_parameters.prefix_len_v4}" else: prefix = f"{addr_v6}/{tool_parameters.prefix_len_v6}" - for protocol, ttls, n_initial_flows in targets[prefix]: - prefixes.append((prefix, protocol, ttls, n_initial_flows)) + try: + for protocol, ttls, n_initial_flows in targets[prefix]: + prefixes.append((prefix, protocol, ttls, n_initial_flows)) + except KeyError: + logger.error( + f"Prefix not in initial target file {targets_filepath}:{prefix}" + ) + continue logger.info("Insert probe counts") insert_probe_counts(