diff --git a/backend/src/Models/etherType.py b/backend/src/Models/etherType.py index 872d18e..8e087d6 100644 --- a/backend/src/Models/etherType.py +++ b/backend/src/Models/etherType.py @@ -131,9 +131,9 @@ ETHERTYPE_MAP: Dict[int, EtherTypeEnum] = { } -def ethertype_from_int(value: int) -> EtherTypeEnum: +def ethertype_from_int(value: int) -> str: """ Map a numeric EtherType value to the corresponding enum. Returns UNKNOWN if not found. """ - return ETHERTYPE_MAP.get(value, EtherTypeEnum.UNKNOWN) + return ETHERTYPE_MAP.get(value, EtherTypeEnum.UNKNOWN).name diff --git a/backend/src/Models/ip_protocol.py b/backend/src/Models/ip_protocol.py index 5d9465a..a50f047 100644 --- a/backend/src/Models/ip_protocol.py +++ b/backend/src/Models/ip_protocol.py @@ -1,4 +1,5 @@ from enum import Enum +from typing import Literal """ This module defines an enumeration for IP protocol types and provides a mapping @@ -310,6 +311,6 @@ IP_PROTOCOL_MAP = { 255: IPProtocolEnum.RESERVED, } -def protocol_from_number(n: int) -> IPProtocolEnum: - return IP_PROTOCOL_MAP.get(n, IPProtocolEnum.UNKNOWN) +def protocol_from_number(n: int) -> str: + return IP_PROTOCOL_MAP.get(n, IPProtocolEnum.UNKNOWN).name