sandbox_network_config.py

from collections.abc import Mapping
from typing import Any, TypeVar, Union, cast

from attrs import define as _attrs_define
from attrs import field as _attrs_field

from ..models.sandbox_network_rule import SandboxNetworkRule
from ..types import UNSET, Unset

T = TypeVar("T", bound="SandboxNetworkConfig")


@_attrs_define
class SandboxNetworkConfig:
    """
    Attributes:
        allow_out (Union[Unset, list[Union['SandboxNetworkRule', str]]]): List of allowed egress entries. Each entry is
            either a CIDR block/IP/host string, or a structured rule with optional per-host transforms. Allowed entries
            always take precedence over denied ones.
        allow_public_traffic (Union[Unset, bool]): Specify if the sandbox URLs should be accessible only with
            authentication. Default: True.
        deny_out (Union[Unset, list[str]]): List of denied CIDR blocks or IP addresses for egress traffic
        mask_request_host (Union[Unset, str]): Specify host mask which will be used for all sandbox requests
    """

    allow_out: Union[Unset, list[Union["SandboxNetworkRule", str]]] = UNSET
    allow_public_traffic: Union[Unset, bool] = True
    deny_out: Union[Unset, list[str]] = UNSET
    mask_request_host: Union[Unset, str] = UNSET
    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)

    def to_dict(self) -> dict[str, Any]:
        allow_out: Union[Unset, list[Union[dict[str, Any], str]]] = UNSET
        if not isinstance(self.allow_out, Unset):
            allow_out = []
            for allow_out_item_data in self.allow_out:
                allow_out_item: Union[dict[str, Any], str]
                if isinstance(allow_out_item_data, SandboxNetworkRule):
                    allow_out_item = allow_out_item_data.to_dict()
                else:
                    allow_out_item = allow_out_item_data
                allow_out.append(allow_out_item)

        allow_public_traffic = self.allow_public_traffic

        deny_out: Union[Unset, list[str]] = UNSET
        if not isinstance(self.deny_out, Unset):
            deny_out = self.deny_out

        mask_request_host = self.mask_request_host

        field_dict: dict[str, Any] = {}
        field_dict.update(self.additional_properties)
        field_dict.update({})
        if allow_out is not UNSET:
            field_dict["allowOut"] = allow_out
        if allow_public_traffic is not UNSET:
            field_dict["allowPublicTraffic"] = allow_public_traffic
        if deny_out is not UNSET:
            field_dict["denyOut"] = deny_out
        if mask_request_host is not UNSET:
            field_dict["maskRequestHost"] = mask_request_host

        return field_dict

    @classmethod
    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
        d = dict(src_dict)
        allow_out = []
        _allow_out = d.pop("allowOut", UNSET)
        for allow_out_item_data in _allow_out or []:

            def _parse_allow_out_item(data: object) -> Union["SandboxNetworkRule", str]:
                try:
                    if not isinstance(data, dict):
                        raise TypeError()
                    allow_out_item_type_1 = SandboxNetworkRule.from_dict(data)

                    return allow_out_item_type_1
                except:  # noqa: E722
                    pass
                return cast(Union["SandboxNetworkRule", str], data)

            allow_out_item = _parse_allow_out_item(allow_out_item_data)

            allow_out.append(allow_out_item)

        allow_public_traffic = d.pop("allowPublicTraffic", UNSET)

        deny_out = cast(list[str], d.pop("denyOut", UNSET))

        mask_request_host = d.pop("maskRequestHost", UNSET)

        sandbox_network_config = cls(
            allow_out=allow_out,
            allow_public_traffic=allow_public_traffic,
            deny_out=deny_out,
            mask_request_host=mask_request_host,
        )

        sandbox_network_config.additional_properties = d
        return sandbox_network_config

    @property
    def additional_keys(self) -> list[str]:
        return list(self.additional_properties.keys())

    def __getitem__(self, key: str) -> Any:
        return self.additional_properties[key]

    def __setitem__(self, key: str, value: Any) -> None:
        self.additional_properties[key] = value

    def __delitem__(self, key: str) -> None:
        del self.additional_properties[key]

    def __contains__(self, key: str) -> bool:
        return key in self.additional_properties