Merge pull request #174 from dioptra-io/hugo_prefix_not_found_error

[diamond_miner]: Prefix not found error
dioptra-io · Jan 11, 2024 · 045414b · 045414b
2 parents 9be8aee + 2c1632c
commit 045414b
Showing 1 changed file with 8 additions and 2 deletions.
diff --git 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(