From 1ca5be29d39c8bbabc57538f0526700135fdf94b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20De=20Keersmaeker?=
<francois.dekeersmaeker@uclouvain.be>
Date: Tue, 24 Sep 2024 10:23:57 +0000
Subject: [PATCH] translate_policies: added nfqueue_name arg
---
profile_translator_blocklist/translator.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/profile_translator_blocklist/translator.py b/profile_translator_blocklist/translator.py
index a349911..e164add 100644
--- a/profile_translator_blocklist/translator.py
+++ b/profile_translator_blocklist/translator.py
@@ -298,6 +298,7 @@ def translate_policy(
def translate_policies(
device: dict,
policies: Iterator[dict],
+ nfqueue_name: str = None,
nfqueue_id: int = 0,
output_dir: str = os.getcwd(),
rate: int = None,
@@ -324,6 +325,7 @@ def translate_policies(
args = validate_args(output_dir, nfqueue_id, rate, drop_proba)
output_dir = args["output_dir"]
drop_proba = args["drop_proba"]
+ nfqueue_name = device.get("name", nfqueue_name)
# Initialize loop variables
nfq_id_inc = 10
@@ -359,7 +361,7 @@ def translate_policies(
nfqueue_id += nfq_id_inc
# Output
- write_firewall(device, global_accs, device.get("name", policy_name), output_dir, drop_proba, log_type, log_group, test)
+ write_firewall(device, global_accs, nfqueue_name, output_dir, drop_proba, log_type, log_group, test)
--
GitLab