catalystwan 0.40.2.dev1__py3-none-any.whl → 2.0.0a1__py3-none-any.whl

catalystwan/models/policy/centralized.py

@@ -1,265 +0,0 @@
-# Copyright 2023 Cisco Systems, Inc. and its affiliates
-
-from typing import Any, List, Literal, Optional, Union, overload
-from uuid import UUID
-
-from pydantic import BaseModel, ConfigDict, Field, model_validator
-from typing_extensions import Annotated
-
-from catalystwan.models.policy.policy import (
-    AssemblyItemBase,
-    PolicyCreationPayload,
-    PolicyDefinition,
-    PolicyEditPayload,
-    PolicyInfo,
-)
-
-TrafficDataDirection = Literal[
-    "service",
-    "tunnel",
-    "all",
-    "from-service",
-    "from-tunnel",
-]
-
-ControlDirection = Literal[
-    "in",
-    "out",
-]
-
-AppRouteItemType = Literal["approute", "appRoute"]
-
-
-def assert_feature_defintion(definition: Any) -> "CentralizedPolicyDefinition":
-    assert isinstance(definition, CentralizedPolicyDefinition)
-    return definition
-
-
-class DataApplicationEntry(BaseModel):
-    model_config = ConfigDict(populate_by_name=True)
-    direction: TrafficDataDirection = "service"
-    site_lists: Optional[List[UUID]] = Field(
-        default=None, serialization_alias="siteLists", validation_alias="siteLists"
-    )
-    vpn_lists: List[UUID] = Field(default_factory=list, serialization_alias="vpnLists", validation_alias="vpnLists")
-    region_ids: Optional[List[str]] = Field(default=None, serialization_alias="regionIds", validation_alias="regionIds")
-    region_lists: Optional[List[UUID]] = Field(
-        default=None, serialization_alias="regionLists", validation_alias="regionLists"
-    )
-
-    # def apply_site_list(self, site_list_id: UUID):
-    #     self.site_lists.append(site_list_id)
-
-    # def apply_vpn_list(self, vpn_list_id: UUID):
-    #     self.vpn_lists.append(vpn_list_id)
-
-
-class ControlApplicationEntry(BaseModel):
-    model_config = ConfigDict(populate_by_name=True)
-    direction: ControlDirection
-    site_lists: Optional[List[UUID]] = Field(
-        default=None, serialization_alias="siteLists", validation_alias="siteLists"
-    )
-    region_lists: Optional[List[UUID]] = Field(
-        default=None, serialization_alias="regionLists", validation_alias="regionLists"
-    )
-    region_ids: Optional[List[str]] = Field(default=None, serialization_alias="regionIds", validation_alias="regionIds")
-
-
-class ControlPolicyItem(AssemblyItemBase):
-    type: Literal["control"] = "control"
-    entries: List[ControlApplicationEntry] = []
-
-    def assign_to_inbound_sites(self, site_lists: List[UUID]) -> None:
-        entry = ControlApplicationEntry(direction="in", site_lists=site_lists)
-        self.entries.append(entry)
-
-    def assign_to_outbound_sites(self, site_lists: List[UUID]) -> None:
-        entry = ControlApplicationEntry(direction="out", site_lists=site_lists)
-        self.entries.append(entry)
-
-    @overload
-    def assign_to_inbound_regions(self, *, region_ids: List[int]) -> None:
-        ...
-
-    @overload
-    def assign_to_inbound_regions(self, *, region_lists: List[UUID]) -> None:
-        ...
-
-    def assign_to_inbound_regions(
-        self, *, region_ids: Optional[List[int]] = None, region_lists: Optional[List[UUID]] = None
-    ) -> None:
-        _region_ids = [str(rid) for rid in region_ids] if region_ids else None
-        entry = entry = ControlApplicationEntry(direction="in", region_ids=_region_ids, region_lists=region_lists)
-        self.entries.append(entry)
-
-    @overload
-    def assign_to_outbound_regions(self, *, region_ids: List[int]) -> None:
-        ...
-
-    @overload
-    def assign_to_outbound_regions(self, *, region_lists: List[UUID]) -> None:
-        ...
-
-    def assign_to_outbound_regions(
-        self, *, region_ids: Optional[List[int]] = None, region_lists: Optional[List[UUID]] = None
-    ) -> None:
-        _region_ids = [str(rid) for rid in region_ids] if region_ids else None
-        entry = entry = ControlApplicationEntry(direction="out", region_ids=_region_ids, region_lists=region_lists)
-        self.entries.append(entry)
-
-
-class TrafficDataPolicyItem(AssemblyItemBase):
-    type: Literal["data"] = "data"
-    entries: List[DataApplicationEntry] = []
-
-    @overload
-    def assign_to(
-        self,
-        vpn_lists: List[UUID],
-        direction: TrafficDataDirection = "service",
-        *,
-        site_lists: List[UUID],
-    ) -> None:
-        ...
-
-    @overload
-    def assign_to(
-        self,
-        vpn_lists: List[UUID],
-        direction: TrafficDataDirection = "service",
-        *,
-        region_lists: List[UUID],
-    ) -> None:
-        ...
-
-    @overload
-    def assign_to(
-        self,
-        vpn_lists: List[UUID],
-        direction: TrafficDataDirection = "service",
-        *,
-        region_ids: List[int],
-    ) -> None:
-        ...
-
-    def assign_to(
-        self,
-        vpn_lists: List[UUID],
-        direction: TrafficDataDirection = "service",
-        *,
-        site_lists: Optional[List[UUID]] = None,
-        region_lists: Optional[List[UUID]] = None,
-        region_ids: Optional[List[int]] = None,
-    ) -> None:
-        entry = DataApplicationEntry(
-            direction=direction,
-            site_lists=site_lists,
-            vpn_lists=vpn_lists,
-            region_lists=region_lists,
-            region_ids=[str(rid) for rid in region_ids] if region_ids else None,
-        )
-        self.entries.append(entry)
-
-
-class HubAndSpokePolicyItem(AssemblyItemBase):
-    type: Literal["hubAndSpoke"] = "hubAndSpoke"
-
-
-class MeshPolicyItem(AssemblyItemBase):
-    type: Literal["mesh"] = "mesh"
-
-
-class AppRoutePolicyItem(AssemblyItemBase):
-    type: AppRouteItemType = "appRoute"
-    entries: List[DataApplicationEntry] = []
-
-
-class CFlowDPolicyItem(AssemblyItemBase):
-    type: Literal["cflowd"] = "cflowd"
-
-
-class VpnMembershipGroupPolicyItem(AssemblyItemBase):
-    type: Literal["vpnMembershipGroup"] = "vpnMembershipGroup"
-
-
-AnyAssemblyItem = Annotated[
-    Union[
-        TrafficDataPolicyItem,
-        ControlPolicyItem,
-        MeshPolicyItem,
-        HubAndSpokePolicyItem,
-        AppRoutePolicyItem,
-        CFlowDPolicyItem,
-        VpnMembershipGroupPolicyItem,
-    ],
-    Field(discriminator="type"),
-]
-
-
-class CentralizedPolicyDefinition(PolicyDefinition):
-    model_config = ConfigDict(populate_by_name=True)
-    region_role_assembly: List = Field(
-        default_factory=list, serialization_alias="regionRoleAssembly", validation_alias="regionRoleAssembly"
-    )
-    assembly: List[AnyAssemblyItem] = []
-    model_config = ConfigDict(populate_by_name=True)
-
-    def find_assembly_item_by_definition_id(self, definition_id: UUID) -> Optional[AnyAssemblyItem]:
-        for item in self.assembly:
-            if item.definition_id == definition_id:
-                return item
-        return None
-
-
-class CentralizedPolicy(PolicyCreationPayload):
-    model_config = ConfigDict(populate_by_name=True)
-    policy_type: Literal["feature", "cli"] = Field(
-        default="feature", serialization_alias="policyType", validation_alias="policyType"
-    )
-    policy_definition: Union[str, CentralizedPolicyDefinition] = Field(
-        serialization_alias="policyDefinition",
-        validation_alias="policyDefinition",
-    )
-
-    def add_traffic_data_policy(self, traffic_data_policy_id: UUID) -> TrafficDataPolicyItem:
-        policy_definition = assert_feature_defintion(self.policy_definition)
-        item = TrafficDataPolicyItem(definition_id=traffic_data_policy_id)
-        policy_definition.assembly.append(item)
-        return item
-
-    def add_control_policy(self, control_policy_id: UUID) -> ControlPolicyItem:
-        policy_definition = assert_feature_defintion(self.policy_definition)
-        item = ControlPolicyItem(definition_id=control_policy_id)
-        policy_definition.assembly.append(item)
-        return item
-
-    def add_mesh_policy(self, mesh_policy_id: UUID) -> None:
-        policy_definition = assert_feature_defintion(self.policy_definition)
-        policy_definition.assembly.append(MeshPolicyItem(definition_id=mesh_policy_id))
-
-    def add_hub_and_spoke_policy(self, hub_and_spoke_policy_id: UUID) -> None:
-        policy_definition = assert_feature_defintion(self.policy_definition)
-        policy_definition.assembly.append(HubAndSpokePolicyItem(definition_id=hub_and_spoke_policy_id))
-
-    @model_validator(mode="before")
-    @classmethod
-    def try_parse_policy_definition_string(cls, values):
-        # this is needed because GET /template/policy/vsmart contains string in policyDefinition field
-        # while POST /template/policy/vsmart requires a regular object
-        # it makes sense to reuse that model for both requests and present parsed data to the user
-        if not isinstance(values, dict):
-            return values
-        json_policy_type = values.get("policyType")
-        json_policy_definition = values.get("policyDefinition")
-        if json_policy_type == "feature" and isinstance(json_policy_definition, str):
-            values["policyDefinition"] = CentralizedPolicyDefinition.model_validate_json(json_policy_definition)
-        return values
-
-
-class CentralizedPolicyEditPayload(PolicyEditPayload, CentralizedPolicy):
-    rid: Optional[int] = Field(default=None, serialization_alias="@rid", validation_alias="@rid")
-
-
-class CentralizedPolicyInfo(PolicyInfo, CentralizedPolicyEditPayload):
-    pass

catalystwan/models/policy/definition/access_control_list.py

@@ -1,193 +0,0 @@
-# Copyright 2024 Cisco Systems, Inc. and its affiliates
-
-from ipaddress import IPv4Address, IPv4Network
-from typing import List, Literal, Set, Tuple, Union
-from uuid import UUID
-
-from pydantic import ConfigDict, Field
-from typing_extensions import Annotated
-
-from catalystwan.models.common import IcmpMsgType
-from catalystwan.models.policy.policy_definition import (
-    AcceptDropActionType,
-    ActionSet,
-    ClassMapAction,
-    ClassMapListEntry,
-    CountAction,
-    DefinitionWithSequencesCommonBase,
-    DestinationDataPrefixListEntry,
-    DestinationIPEntry,
-    DestinationPortEntry,
-    DSCPEntry,
-    ICMPMessageEntry,
-    LogAction,
-    Match,
-    MirrorAction,
-    NextHopActionEntry,
-    PacketLengthEntry,
-    PLPEntry,
-    PolicerAction,
-    PolicyAcceptDropAction,
-    PolicyDefinitionBase,
-    PolicyDefinitionGetResponse,
-    PolicyDefinitionId,
-    PolicyDefinitionSequenceBase,
-    ProtocolEntry,
-    Reference,
-    SourceDataPrefixListEntry,
-    SourceIPEntry,
-    SourcePortEntry,
-    TCPEntry,
-    accept_action,
-)
-
-AclPolicySequenceMatchEntry = Annotated[
-    Union[
-        ClassMapListEntry,
-        DestinationDataPrefixListEntry,
-        DestinationIPEntry,
-        DestinationPortEntry,
-        DSCPEntry,
-        ICMPMessageEntry,
-        PacketLengthEntry,
-        PLPEntry,
-        ProtocolEntry,
-        SourceDataPrefixListEntry,
-        SourceIPEntry,
-        SourcePortEntry,
-        TCPEntry,
-    ],
-    Field(discriminator="field"),
-]
-
-AclPolicySequenceActions = Annotated[
-    Union[
-        CountAction,
-        LogAction,
-        MirrorAction,
-        ClassMapAction,
-        PolicerAction,
-    ],
-    Field(discriminator="type"),
-]
-
-
-class AclPolicyHeader(PolicyDefinitionBase):
-    type: Literal["acl"] = "acl"
-
-
-class AclPolicySequenceMatch(Match):
-    entries: List[AclPolicySequenceMatchEntry] = []
-
-
-class AclPolicySequence(PolicyDefinitionSequenceBase):
-    sequence_type: Literal["acl"] = Field(
-        default="acl", serialization_alias="sequenceType", validation_alias="sequenceType"
-    )
-    base_action: AcceptDropActionType = Field(
-        default="accept", serialization_alias="baseAction", validation_alias="baseAction"
-    )
-    match: AclPolicySequenceMatch = AclPolicySequenceMatch()
-    actions: List[Union[AclPolicySequenceActions, ActionSet]] = []
-    model_config = ConfigDict(populate_by_name=True)
-
-    def match_dscp(self, dscp: List[int]) -> None:
-        self._insert_match(DSCPEntry(value=dscp))
-
-    def match_icmp(self, icmp_message_types: List[IcmpMsgType]) -> None:
-        self._insert_match(ICMPMessageEntry(value=icmp_message_types))
-
-    def match_packet_length(self, packet_lengths: Tuple[int, int]) -> None:
-        self._insert_match(PacketLengthEntry.from_range(packet_lengths))
-
-    def match_low_plp(self) -> None:
-        self._insert_match(PLPEntry(value="low"))
-
-    def match_high_plp(self) -> None:
-        self._insert_match(PLPEntry(value="high"))
-
-    def match_protocols(self, protocols: Set[int]) -> None:
-        self._insert_match(ProtocolEntry.from_protocol_set(protocols))
-
-    def match_source_data_prefix_list(self, data_prefix_list_id: UUID) -> None:
-        self._insert_match(SourceDataPrefixListEntry(ref=[data_prefix_list_id]))
-
-    def match_source_ip(self, networks: List[IPv4Network]) -> None:
-        self._insert_match(SourceIPEntry.from_ipv4_networks(networks))
-
-    def match_source_port(self, ports: Set[int] = set(), port_ranges: List[Tuple[int, int]] = []) -> None:
-        self._insert_match(SourcePortEntry.from_port_set_and_ranges(ports, port_ranges))
-
-    def match_destination_data_prefix_list(self, data_prefix_list_id: UUID) -> None:
-        self._insert_match(DestinationDataPrefixListEntry(ref=[data_prefix_list_id]))
-
-    def match_destination_ip(self, networks: List[IPv4Network]) -> None:
-        self._insert_match(DestinationIPEntry.from_ipv4_networks(networks))
-
-    def match_destination_port(self, ports: Set[int] = set(), port_ranges: List[Tuple[int, int]] = []) -> None:
-        self._insert_match(DestinationPortEntry.from_port_set_and_ranges(ports, port_ranges))
-
-    def match_tcp(self) -> None:
-        self._insert_match(TCPEntry())
-
-    def match_class_map_list_entry(self, class_map_list_id: UUID) -> None:
-        self._insert_match(ClassMapListEntry(ref=class_map_list_id))
-
-    def associate_count_action(self, counter_name: str) -> None:
-        self._insert_action(CountAction(parameter=counter_name))
-
-    def associate_log_action(self) -> None:
-        self._insert_action(LogAction())
-
-    @accept_action
-    def associate_dscp_action(self, dscp: int) -> None:
-        self._insert_action_in_set(DSCPEntry(value=[dscp]))
-
-    @accept_action
-    def associate_next_hop_action(self, next_hop: IPv4Address) -> None:
-        self._insert_action_in_set(NextHopActionEntry(value=next_hop))
-
-    @accept_action
-    def associate_mirror_action(self, mirror_list_id: UUID) -> None:
-        self._insert_action(MirrorAction(parameter=Reference(ref=mirror_list_id)))
-
-    @accept_action
-    def associate_class_map_list_action(self, class_map_list_id: UUID) -> None:
-        self._insert_action(ClassMapAction(parameter=Reference(ref=class_map_list_id)))
-
-    @accept_action
-    def associate_policer_list_action(self, policer_list_id: UUID) -> None:
-        self._insert_action(PolicerAction(parameter=Reference(ref=policer_list_id)))
-
-
-class AclPolicy(AclPolicyHeader, DefinitionWithSequencesCommonBase):
-    sequences: List[AclPolicySequence] = []
-    default_action: PolicyAcceptDropAction = Field(
-        default=PolicyAcceptDropAction(type="drop"),
-        serialization_alias="defaultAction",
-        validation_alias="defaultAction",
-    )
-    model_config = ConfigDict(populate_by_name=True)
-
-    def set_default_action(self, action: AcceptDropActionType):
-        self.default_action = PolicyAcceptDropAction(type=action)
-
-    def add_sequence(
-        self, id_: int, name: str = "Access Control List", base_action: AcceptDropActionType = "accept"
-    ) -> AclPolicySequence:
-        seq = AclPolicySequence(
-            sequence_id=id_,
-            sequence_name=name,
-            base_action=base_action,
-            sequence_ip_type="ipv4",
-        )
-        self.add(seq)
-        return seq
-
-
-class AclPolicyEditPayload(AclPolicy, PolicyDefinitionId):
-    pass
-
-
-class AclPolicyGetResponse(AclPolicy, PolicyDefinitionGetResponse):
-    pass

catalystwan/models/policy/definition/access_control_list_ipv6.py

@@ -1,187 +0,0 @@
-# Copyright 2024 Cisco Systems, Inc. and its affiliates
-
-from ipaddress import IPv6Address, IPv6Network
-from typing import List, Literal, Set, Tuple, Union
-from uuid import UUID
-
-from pydantic import ConfigDict, Field
-from typing_extensions import Annotated
-
-from catalystwan.models.common import AcceptDropActionType
-from catalystwan.models.policy.policy_definition import (
-    ActionSet,
-    ClassMapAction,
-    ClassMapListEntry,
-    CountAction,
-    DefinitionWithSequencesCommonBase,
-    DestinationDataIPv6PrefixListEntry,
-    DestinationIPv6Entry,
-    DestinationPortEntry,
-    LogAction,
-    Match,
-    MirrorAction,
-    NextHeaderEntry,
-    NextHopActionEntry,
-    PacketLengthEntry,
-    PLPEntry,
-    PolicerAction,
-    PolicyAcceptDropAction,
-    PolicyDefinitionBase,
-    PolicyDefinitionGetResponse,
-    PolicyDefinitionId,
-    PolicyDefinitionSequenceBase,
-    Reference,
-    SourceDataIPv6PrefixListEntry,
-    SourceIPv6Entry,
-    SourcePortEntry,
-    TCPEntry,
-    TrafficClassEntry,
-    accept_action,
-)
-
-AclIPv6PolicySequenceMatchEntry = Annotated[
-    Union[
-        ClassMapListEntry,
-        DestinationDataIPv6PrefixListEntry,
-        DestinationIPv6Entry,
-        DestinationPortEntry,
-        NextHeaderEntry,
-        PacketLengthEntry,
-        PLPEntry,
-        SourceDataIPv6PrefixListEntry,
-        SourceIPv6Entry,
-        SourcePortEntry,
-        TCPEntry,
-        TrafficClassEntry,
-    ],
-    Field(discriminator="field"),
-]
-
-AclIPv6PolicySequenceActions = Annotated[
-    Union[
-        CountAction,
-        LogAction,
-        MirrorAction,
-        ClassMapAction,
-        PolicerAction,
-    ],
-    Field(discriminator="type"),
-]
-
-
-class AclIPv6PolicyHeader(PolicyDefinitionBase):
-    type: Literal["aclv6"] = "aclv6"
-
-
-class AclIPv6PolicySequenceMatch(Match):
-    entries: List[AclIPv6PolicySequenceMatchEntry] = []
-
-
-class AclIPv6PolicySequence(PolicyDefinitionSequenceBase):
-    sequence_type: Literal["aclv6"] = Field(
-        default="aclv6", serialization_alias="sequenceType", validation_alias="sequenceType"
-    )
-    base_action: AcceptDropActionType = Field(
-        default="accept", serialization_alias="baseAction", validation_alias="baseAction"
-    )
-    match: AclIPv6PolicySequenceMatch = AclIPv6PolicySequenceMatch()
-    actions: List[Union[AclIPv6PolicySequenceActions, ActionSet]] = []
-    model_config = ConfigDict(populate_by_name=True)
-
-    def match_next_header(self, next_header: int) -> None:
-        self._insert_match(NextHeaderEntry(value=str(next_header)))
-
-    def match_packet_length(self, packet_lengths: Tuple[int, int]) -> None:
-        self._insert_match(PacketLengthEntry.from_range(packet_lengths))
-
-    def match_low_plp(self) -> None:
-        self._insert_match(PLPEntry(value="low"))
-
-    def match_high_plp(self) -> None:
-        self._insert_match(PLPEntry(value="high"))
-
-    def match_source_data_prefix_list(self, data_prefix_list_id: UUID) -> None:
-        self._insert_match(SourceDataIPv6PrefixListEntry(ref=[data_prefix_list_id]))
-
-    def match_source_ip(self, networks: List[IPv6Network]) -> None:
-        self._insert_match(SourceIPv6Entry.from_ipv6_networks(networks))
-
-    def match_source_port(self, ports: Set[int] = set(), port_ranges: List[Tuple[int, int]] = []) -> None:
-        self._insert_match(SourcePortEntry.from_port_set_and_ranges(ports, port_ranges))
-
-    def match_destination_data_prefix_list(self, data_prefix_list_id: UUID) -> None:
-        self._insert_match(DestinationDataIPv6PrefixListEntry(ref=[data_prefix_list_id]))
-
-    def match_destination_ip(self, networks: List[IPv6Network]) -> None:
-        self._insert_match(DestinationIPv6Entry.from_ipv6_networks(networks))
-
-    def match_destination_port(self, ports: Set[int] = set(), port_ranges: List[Tuple[int, int]] = []) -> None:
-        self._insert_match(DestinationPortEntry.from_port_set_and_ranges(ports, port_ranges))
-
-    def match_tcp(self) -> None:
-        self._insert_match(TCPEntry())
-
-    def match_class_map_list_entry(self, class_map_list_id: UUID) -> None:
-        self._insert_match(ClassMapListEntry(ref=class_map_list_id))
-
-    def match_traffic_class(self, traffic_class: List[int]) -> None:
-        self._insert_match(TrafficClassEntry(value=" ".join([str(tc) for tc in traffic_class])))
-
-    def associate_count_action(self, counter_name: str) -> None:
-        self._insert_action(CountAction(parameter=counter_name))
-
-    def associate_log_action(self) -> None:
-        self._insert_action(LogAction())
-
-    @accept_action
-    def associate_next_hop_action(self, next_hop: IPv6Address) -> None:
-        self._insert_action_in_set(NextHopActionEntry(value=next_hop))
-
-    @accept_action
-    def associate_traffic_class_action(self, traffic_class: int) -> None:
-        self._insert_action_in_set(TrafficClassEntry(value=str(traffic_class)))
-
-    @accept_action
-    def associate_mirror_action(self, mirror_list_id: UUID) -> None:
-        self._insert_action(MirrorAction(parameter=Reference(ref=mirror_list_id)))
-
-    @accept_action
-    def associate_class_map_list_action(self, class_map_list_id: UUID) -> None:
-        self._insert_action(ClassMapAction(parameter=Reference(ref=class_map_list_id)))
-
-    @accept_action
-    def associate_policer_list_action(self, policer_list_id: UUID) -> None:
-        self._insert_action(PolicerAction(parameter=Reference(ref=policer_list_id)))
-
-
-class AclIPv6Policy(AclIPv6PolicyHeader, DefinitionWithSequencesCommonBase):
-    sequences: List[AclIPv6PolicySequence] = []
-    default_action: PolicyAcceptDropAction = Field(
-        default=PolicyAcceptDropAction(type="drop"),
-        serialization_alias="defaultAction",
-        validation_alias="defaultAction",
-    )
-    model_config = ConfigDict(populate_by_name=True)
-
-    def set_default_action(self, action: AcceptDropActionType):
-        self.default_action = PolicyAcceptDropAction(type=action)
-
-    def add_sequence(
-        self, id_: int, name: str = "Access Control List", base_action: AcceptDropActionType = "accept"
-    ) -> AclIPv6PolicySequence:
-        seq = AclIPv6PolicySequence(
-            sequence_id=id_,
-            sequence_name=name,
-            base_action=base_action,
-            sequence_ip_type="ipv6",
-        )
-        self.add(seq)
-        return seq
-
-
-class AclIPv6PolicyEditPayload(AclIPv6Policy, PolicyDefinitionId):
-    pass
-
-
-class AclIPv6PolicyGetResponse(AclIPv6Policy, PolicyDefinitionGetResponse):
-    pass

catalystwan/models/policy/definition/aip.py

@@ -1,60 +0,0 @@
-# Copyright 2024 Cisco Systems, Inc. and its affiliates
-
-from typing import Literal, Optional
-
-from pydantic import BaseModel, ConfigDict, Field, model_validator
-from typing_extensions import Self
-
-from catalystwan.models.policy.policy_definition import (
-    PolicyDefinitionBase,
-    PolicyDefinitionGetResponse,
-    PolicyDefinitionId,
-    Reference,
-)
-
-TlsDecryptionAction = Literal["decrypt", "neverDecrypt", "skipDecrypt"]
-
-
-class AdvancedInspectionProfileDefinition(BaseModel):
-    model_config = ConfigDict(populate_by_name=True)
-
-    tls_decryption_action: TlsDecryptionAction = Field(
-        validation_alias="tlsDecryptionAction", serialization_alias="tlsDecryptionAction"
-    )
-
-    advanced_malware_protection: Optional[Reference] = Field(
-        default=None, validation_alias="advancedMalwareProtection", serialization_alias="advancedMalwareProtection"
-    )
-
-    intrusion_prevention: Optional[Reference] = Field(
-        default=None, validation_alias="intrusionPrevention", serialization_alias="intrusionPrevention"
-    )
-    url_filtering: Optional[Reference] = Field(
-        default=None, validation_alias="urlFiltering", serialization_alias="urlFiltering"
-    )
-
-    ssl_utd_decrypt_profile: Optional[Reference] = Field(
-        default=None, validation_alias="sslUtdDecryptProfile", serialization_alias="sslUtdDecryptProfile"
-    )
-
-    @model_validator(mode="after")
-    def check_at_least_one_optional_filled(self) -> Self:
-        if not any([self.advanced_malware_protection, self.intrusion_prevention, self.url_filtering]):
-            raise ValueError(
-                "At least one of Intrusion Prevention, URL Filtering, Advanced Malware Protection shall be filled."
-            )
-        return self
-
-
-class AdvancedInspectionProfilePolicy(PolicyDefinitionBase):
-    type: Literal["advancedInspectionProfile"] = "advancedInspectionProfile"
-    mode: Literal["unified"] = "unified"
-    definition: AdvancedInspectionProfileDefinition
-
-
-class AdvancedInspectionProfilePolicyEditPayload(AdvancedInspectionProfilePolicy, PolicyDefinitionId):
-    pass
-
-
-class AdvancedInspectionProfilePolicyGetResponse(AdvancedInspectionProfilePolicy, PolicyDefinitionGetResponse):
-    pass