compose-viz/compose_viz/models/port.py

from enum import Enum


class Protocol(str, Enum):
    tcp = "tcp"
    udp = "udp"
    any = "any"


class AppProtocol(str, Enum):
    rest = "REST"
    mqtt = "MQTT"
    wbsock = "WebSocket"
    http = "http"
    https = "https"
    na = "NA"


class Port:
    def __init__(
        self,
        host_port: str,
        container_port: str,
        protocol: Protocol = Protocol.any,
        app_protocol: AppProtocol = AppProtocol.na,
    ):
        self._host_port = host_port
        self._container_port = container_port
        self._protocol = protocol
        self._app_protocol = app_protocol

    @property
    def host_port(self):
        return self._host_port

    @property
    def container_port(self):
        return self._container_port

    @property
    def protocol(self):
        return self._protocol

    @property
    def app_protocol(self):
        return self._app_protocol