From 5ff127a4a5d97e0d4bad582e7267e2e5315a02a0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20De=20Keersmaeker?=
<francois.dekeersmaeker@uclouvain.be>
Date: Tue, 3 Sep 2024 07:46:26 +0000
Subject: [PATCH] Updated Python package
---
Protocol.py | 2 +-
__init__.py | 57 +++++++++++++++++++++++++++++++++++++++++++++++++++++
2 files changed, 58 insertions(+), 1 deletion(-)
diff --git a/Protocol.py b/Protocol.py
index ec16fd3..81d0b59 100644
--- a/Protocol.py
+++ b/Protocol.py
@@ -48,7 +48,7 @@ class Protocol:
:param protocol_data: Dictionary containing the protocol data.
:param device: Dictionary containing the device metadata.
"""
- module = importlib.import_module(f"{protocol_name}")
+ module = importlib.import_module(f"profile_translator.protocols.{protocol_name}")
cls = getattr(module, protocol_name)
return cls(protocol_data, device)
diff --git a/__init__.py b/__init__.py
index e69de29..075ac7d 100644
--- a/__init__.py
+++ b/__init__.py
@@ -0,0 +1,57 @@
+"""
+`protocol-translators-blocklist` package.
+"""
+
+## Abstract classes
+from .Protocol import Protocol
+from .Transport import Transport
+from .Custom import Custom
+
+## Concrete classes
+# Layer 2
+from .arp import arp
+# Layer 3
+from .igmp import igmp
+from .ip import ip
+from .ipv4 import ipv4
+from .ipv6 import ipv6
+from .icmp import icmp
+from .icmpv6 import icmpv6
+# Layer 4
+from .tcp import tcp
+from .udp import udp
+# Layer 7
+from .dns import dns
+from .mdns import mdns
+from .http import http
+from .coap import coap
+from .dhcp import dhcp
+from .ssdp import ssdp
+
+
+__all__ = [
+ ## Abstract classes
+ "Protocol",
+ "Transport",
+ "Custom",
+ ## Concrete classes
+ # Layer 2
+ "arp",
+ # Layer 3
+ "ip",
+ "ipv4",
+ "ipv6",
+ "icmp",
+ "icmpv6",
+ "igmp",
+ # Layer 4
+ "tcp",
+ "udp",
+ # Layer 7
+ "coap",
+ "dhcp",
+ "dns",
+ "http",
+ "mdns",
+ "ssdp"
+]
--
GitLab