From d84a8ae48916f47fd5156c3d1af1e314640b63b5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fran=C3=A7ois=20De=20Keersmaeker?= <francois.dekeersmaeker@uclouvain.be> Date: Thu, 7 Nov 2024 21:53:12 +0000 Subject: [PATCH] Fixed type hints --- profile_translator_blocklist/translator.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/profile_translator_blocklist/translator.py b/profile_translator_blocklist/translator.py index 5499764..28880b9 100644 --- a/profile_translator_blocklist/translator.py +++ b/profile_translator_blocklist/translator.py @@ -5,7 +5,7 @@ of NFTables firewall script and NFQueue C source code. ## Imports # Libraries -from typing import Iterator +from typing import Iterable import os import importlib import yaml @@ -301,7 +301,7 @@ def translate_policy( def translate_policies( device: dict, - policies: Iterator[dict], + policies: Iterable[dict], nfqueue_name: str = None, nfqueue_id: int = 0, output_dir: str = os.getcwd(), @@ -316,7 +316,7 @@ def translate_policies( Args: device (dict): Device metadata - policies (Iterator[dict]): iterator over policies + policies (Iterable[dict]): iterable containing the policies to translate nfqueue_id (int): NFQueue start index for this profile's policies (must be an integer between 0 and 65535) output_dir (str): Output directory for the generated files rate (int): Rate limit, in packets/second, to apply to matched traffic, instead of a binary verdict -- GitLab