diff --git a/Protocol.py b/Protocol.py index ec16fd3776dee7f02c98f6923cb4c41ac1e0699f..81d0b5939413ad02dd3d74c6344fe6805d33a88a 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 e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..075ac7dd40f013df74ff73e96535d4dafd12c0da 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" +]