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

catalystwan/models/policy/definition/amp.py

@@ -1,65 +0,0 @@
-# Copyright 2024 Cisco Systems, Inc. and its affiliates
-
-from typing import List, Literal, Optional
-
-from pydantic import BaseModel, ConfigDict, Field
-
-from catalystwan.models.common import (
-    AmpFileAlertLevel,
-    AmpFileAnalysisFileTypes,
-    AmpFileAnalysisServer,
-    AmpFileReputationServer,
-    PolicyModeType,
-    VpnId,
-)
-from catalystwan.models.policy.policy_definition import (
-    PolicyDefinitionBase,
-    PolicyDefinitionGetResponse,
-    PolicyDefinitionId,
-)
-
-OptionalAmpFileAnalysisServer = Literal["", AmpFileAnalysisServer]
-OptionalAmpFileAlertLevel = Literal["", AmpFileAlertLevel]
-
-
-class AdvancedMalwareProtectionDefinition(BaseModel):
-    model_config = ConfigDict(populate_by_name=True)
-    match_all_vpn: bool = Field(validation_alias="matchAllVpn", serialization_alias="matchAllVpn")
-    file_reputation_cloud_server: AmpFileReputationServer = Field(
-        validation_alias="fileReputationCloudServer", serialization_alias="fileReputationCloudServer"
-    )
-    file_reputation_est_server: AmpFileReputationServer = Field(
-        validation_alias="fileReputationEstServer", serialization_alias="fileReputationEstServer"
-    )
-    file_reputation_alert: AmpFileAlertLevel = Field(
-        validation_alias="fileReputationAlert", serialization_alias="fileReputationAlert"
-    )
-    file_analysis_enabled: Optional[bool] = Field(
-        default=False, validation_alias="fileAnalysisEnabled", serialization_alias="fileAnalysisEnabled"
-    )
-    file_analysis_file_types: List[AmpFileAnalysisFileTypes] = Field(
-        default_factory=list, validation_alias="fileAnalysisFileTypes", serialization_alias="fileAnalysisFileTypes"
-    )
-    file_analysis_alert: OptionalAmpFileAlertLevel = Field(
-        default="", validation_alias="fileAnalysisAlert", serialization_alias="fileAnalysisAlert"
-    )
-    file_analysis_cloud_server: OptionalAmpFileAnalysisServer = Field(
-        default="", validation_alias="fileAnalysisCloudServer", serialization_alias="fileAnalysisCloudServer"
-    )
-    target_vpns: List[VpnId] = Field(
-        default_factory=list, validation_alias="targetVpns", serialization_alias="targetVpns"
-    )
-
-
-class AdvancedMalwareProtectionPolicy(PolicyDefinitionBase):
-    type: Literal["advancedMalwareProtection"] = "advancedMalwareProtection"
-    mode: PolicyModeType = "security"
-    definition: AdvancedMalwareProtectionDefinition
-
-
-class AdvancedMalwareProtectionPolicyEditPayload(AdvancedMalwareProtectionPolicy, PolicyDefinitionId):
-    pass
-
-
-class AdvancedMalwareProtectionPolicyGetResponse(AdvancedMalwareProtectionPolicy, PolicyDefinitionGetResponse):
-    pass

catalystwan/models/policy/definition/app_route.py

@@ -1,246 +0,0 @@
-# Copyright 2024 Cisco Systems, Inc. and its affiliates
-from ipaddress import IPv4Network, IPv6Network
-from typing import List, Literal, Optional, Set, Tuple, Union, overload
-from uuid import UUID
-
-from pydantic import ConfigDict, Field
-from typing_extensions import Annotated
-
-from catalystwan.models.common import (
-    DestinationRegion,
-    DNSEntryType,
-    SequenceIpType,
-    ServiceAreaValue,
-    TLOCColor,
-    TrafficCategory,
-    TrafficTargetType,
-)
-from catalystwan.models.policy.policy_definition import (
-    AppListEntry,
-    BackupSlaPrefferedColorAction,
-    CloudSaaSAction,
-    CountAction,
-    DefinitionWithSequencesCommonBase,
-    DestinationDataIPv6PrefixListEntry,
-    DestinationDataPrefixListEntry,
-    DestinationIPEntry,
-    DestinationIPv6Entry,
-    DestinationPortEntry,
-    DestinationRegionEntry,
-    DNSAppListEntry,
-    DNSEntry,
-    DSCPEntry,
-    LogAction,
-    Match,
-    PLPEntry,
-    PLPEntryType,
-    PolicyActionBase,
-    PolicyDefinitionBase,
-    PolicyDefinitionGetResponse,
-    PolicyDefinitionId,
-    PolicyDefinitionSequenceBase,
-    ProtocolEntry,
-    SaaSAppListEntry,
-    ServiceAreaEntry,
-    SlaClassAction,
-    SlaNotMetAction,
-    SourceDataIPv6PrefixListEntry,
-    SourceDataPrefixListEntry,
-    SourceIPEntry,
-    SourceIPv6Entry,
-    SourcePortEntry,
-    TrafficCategoryEntry,
-    TrafficToEntry,
-)
-
-AppRoutePolicySequenceEntry = Annotated[
-    Union[
-        AppListEntry,
-        DestinationDataIPv6PrefixListEntry,
-        DestinationDataPrefixListEntry,
-        DestinationIPEntry,
-        DestinationIPv6Entry,
-        DestinationPortEntry,
-        DestinationRegionEntry,
-        DNSAppListEntry,
-        DNSEntry,
-        DSCPEntry,
-        PLPEntry,
-        ProtocolEntry,
-        SaaSAppListEntry,
-        ServiceAreaEntry,
-        SourceDataIPv6PrefixListEntry,
-        SourceDataPrefixListEntry,
-        SourceIPEntry,
-        SourceIPv6Entry,
-        SourcePortEntry,
-        TrafficCategoryEntry,
-        TrafficToEntry,
-    ],
-    Field(discriminator="field"),
-]
-
-AppRoutePolicySequenceActionEntry = Annotated[
-    Union[
-        BackupSlaPrefferedColorAction,
-        CloudSaaSAction,
-        CountAction,
-        LogAction,
-        SlaClassAction,
-    ],
-    Field(discriminator="type"),
-]
-
-
-class AppRoutePolicySequenceMatch(Match):
-    entries: List[AppRoutePolicySequenceEntry] = []
-
-
-class AppRoutePolicySequence(PolicyDefinitionSequenceBase):
-    sequence_type: Literal["appRoute"] = Field(
-        default="appRoute", serialization_alias="sequenceType", validation_alias="sequenceType"
-    )
-    match: AppRoutePolicySequenceMatch = AppRoutePolicySequenceMatch()
-    actions: List[AppRoutePolicySequenceActionEntry] = []
-    model_config = ConfigDict(populate_by_name=True)
-
-    def match_app_list(self, app_list_id: UUID) -> None:
-        self._insert_match(AppListEntry(ref=[app_list_id]))
-
-    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_ipv6(self, networks: List[IPv6Network]) -> None:
-        self._insert_match(DestinationIPv6Entry.from_ipv6_networks(networks))
-
-    def match_destination_region(self, region: DestinationRegion) -> None:
-        self._insert_match(DestinationRegionEntry(value=region))
-
-    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_dns(self, dns: DNSEntryType) -> None:
-        self._insert_match(DNSEntry(value=dns))
-
-    def match_dns_app_list(self, dns_app_list_id: UUID) -> None:
-        self._insert_match(DNSAppListEntry(ref=dns_app_list_id))
-
-    def match_dscp(self, dscp: List[int]) -> None:
-        self._insert_match(DSCPEntry(value=dscp))
-
-    def match_plp(self, plp: PLPEntryType) -> None:
-        self._insert_match(PLPEntry(value=plp))
-
-    def match_protocols(self, protocols: Set[int]) -> None:
-        self._insert_match(ProtocolEntry.from_protocol_set(protocols))
-
-    def match_saas_app_list(self, saas_app_list_id: UUID) -> None:
-        self._insert_match(SaaSAppListEntry(ref=saas_app_list_id))
-
-    def match_service_areas(self, service_areas: Set[ServiceAreaValue]) -> None:
-        self._insert_match(ServiceAreaEntry(value=list(service_areas)))
-
-    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_ipv6(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_traffic_category(self, traffic_category: TrafficCategory) -> None:
-        self._insert_match(TrafficCategoryEntry(value=traffic_category))
-
-    def match_traffic_to(self, traffic_to: TrafficTargetType) -> None:
-        self._insert_match(TrafficToEntry(value=traffic_to))
-
-    def associate_backup_sla_preferred_color_action(self, tloc_colors: List[TLOCColor]) -> None:
-        self._insert_action(BackupSlaPrefferedColorAction(parameter=tloc_colors))
-
-    def associate_cloud_saas_action(self) -> None:
-        self._insert_action(CloudSaaSAction())
-
-    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())
-
-    @overload
-    def associate_sla_class_action(
-        self, sla_class: UUID, not_met_action: Optional[SlaNotMetAction] = None, *, preferred_color: List[TLOCColor]
-    ) -> None:
-        ...
-
-    @overload
-    def associate_sla_class_action(
-        self, sla_class: UUID, not_met_action: Optional[SlaNotMetAction] = None, *, preferred_color_group: UUID
-    ) -> None:
-        ...
-
-    def associate_sla_class_action(
-        self,
-        sla_class: UUID,
-        not_met_action: Optional[SlaNotMetAction] = None,
-        *,
-        preferred_color: Optional[List[TLOCColor]] = None,
-        preferred_color_group: Optional[UUID] = None,
-    ) -> None:
-        if preferred_color is not None:
-            action = SlaClassAction.from_params(
-                sla_class=sla_class, not_met_action=not_met_action, preferred_color=preferred_color
-            )
-        elif preferred_color_group is not None:
-            action = SlaClassAction.from_params(
-                sla_class=sla_class, not_met_action=not_met_action, preferred_color_group=preferred_color_group
-            )
-        else:
-            action = SlaClassAction()
-        self._insert_action(action)
-
-
-class AppRouteDefaultAction(PolicyActionBase):
-    type: Literal["slaClass"] = "slaClass"
-    ref: UUID
-
-
-class AppRoutePolicyHeader(PolicyDefinitionBase):
-    type: Literal["appRoute"] = "appRoute"
-
-
-class AppRoutePolicy(AppRoutePolicyHeader, DefinitionWithSequencesCommonBase):
-    sequences: List[AppRoutePolicySequence] = []
-    default_action: Optional[AppRouteDefaultAction] = Field(
-        default=None, serialization_alias="defaultAction", validation_alias="defaultAction"
-    )
-    model_config = ConfigDict(populate_by_name=True)
-
-    def set_default_action(self, sla_class: Optional[UUID]) -> None:
-        self.default_action = AppRouteDefaultAction(ref=sla_class) if sla_class else None
-
-    def add_sequence(
-        self,
-        name: str = "App Route",
-        sequence_ip_type: SequenceIpType = "ipv4",
-    ) -> AppRoutePolicySequence:
-        seq = AppRoutePolicySequence(
-            sequence_name=name,
-            sequence_ip_type=sequence_ip_type,
-        )
-        self.add(seq)
-        return seq
-
-
-class AppRoutePolicyEditPayload(AppRoutePolicy, PolicyDefinitionId):
-    pass
-
-
-class AppRoutePolicyGetResponse(AppRoutePolicy, PolicyDefinitionGetResponse):
-    pass

catalystwan/models/policy/definition/cflowd.py

@@ -1,95 +0,0 @@
-# Copyright 2024 Cisco Systems, Inc. and its affiliates
-
-from typing import List, Literal, Optional
-
-from pydantic import BaseModel, ConfigDict, Field
-
-from catalystwan.models.common import VpnId
-from catalystwan.models.policy.policy_definition import (
-    Optimized,
-    PolicyDefinitionBase,
-    PolicyDefinitionGetResponse,
-    PolicyDefinitionId,
-)
-
-IpProtocol = Literal["ipv4", "ipv6", "both"]
-TransportProtocol = Literal["transport_udp", "transport_tcp"]
-ExportState = Literal["enable"]
-
-
-class Collector(BaseModel):
-    model_config = ConfigDict(populate_by_name=True)
-    vpn: VpnId
-    address: str
-    port: int = Field(ge=1024, le=65536)
-    transport: TransportProtocol
-    source_interface: Optional[str] = Field(
-        default=None, validation_alias="sourceInterface", serialization_alias="sourceInterface"
-    )
-    export_spread: Optional[ExportState] = Field(
-        default=None, validation_alias="exportSpread", serialization_alias="exportSpread"
-    )
-    bfd_metrics_export: Optional[ExportState] = Field(
-        default=None, validation_alias="bfd-metrics-export", serialization_alias="bfd-metrics-export"
-    )
-    export_interval: Optional[int] = Field(
-        default=None, validation_alias="export-interval", serialization_alias="export-interval", ge=0, le=86400
-    )
-
-    def enable_bfd_metrics_exporting(self, export_interval: int):
-        self.bfd_metrics_export = "enable"
-        self.export_interval = export_interval
-
-
-class CustomizedIpv4RecordFields(BaseModel):
-    model_config = ConfigDict(populate_by_name=True)
-    collect_tos: bool = Field(default=False, validation_alias="collectTos", serialization_alias="collectTos")
-    collect_dscp_output: bool = Field(
-        default=False, validation_alias="collectDscpOutput", serialization_alias="collectDscpOutput"
-    )
-
-
-class CflowdDefinition(BaseModel):
-    model_config = ConfigDict(populate_by_name=True)
-    collectors: List[Collector] = []
-    protocol: IpProtocol = "ipv4"
-
-    flow_active_timeout: Optional[int] = Field(
-        default=None, validation_alias="flowActiveTimeout", serialization_alias="flowActiveTimeout", ge=30, le=3600
-    )
-
-    flow_inactive_timeout: Optional[int] = Field(
-        default=None, validation_alias="flowInactiveTimeout", serialization_alias="flowInactiveTimeout", ge=1, le=3600
-    )
-
-    template_refresh: Optional[int] = Field(
-        default=None, validation_alias="templateRefresh", serialization_alias="templateRefresh", ge=60, le=86400
-    )
-
-    flow_sampling_interval: Optional[int] = Field(
-        default=None,
-        validation_alias="flowSamplingInterval",
-        serialization_alias="flowSamplingInterval",
-        ge=1,
-        le=65536,
-    )
-
-    customized_ipv4_record_fields: CustomizedIpv4RecordFields = Field(
-        default=CustomizedIpv4RecordFields(),
-        validation_alias="customizedIpv4RecordFields",
-        serialization_alias="customizedIpv4RecordFields",
-    )
-
-
-class CflowdPolicy(PolicyDefinitionBase):
-    type: Literal["cflowd"] = "cflowd"
-    optimized: Optional[Optimized] = None
-    definition: CflowdDefinition
-
-
-class CflowdPolicyEditPayload(CflowdPolicy, PolicyDefinitionId):
-    pass
-
-
-class CflowdPolicyGetResponse(CflowdPolicy, PolicyDefinitionGetResponse):
-    pass