Updated Python package

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)
 

__init__.py

+"""
+`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"
+]