update policy api with new policy lists

cisco-open · Oct 24, 2024 · 93b912c · 93b912c
1 parent b0c4ffc
commit 93b912c
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 3 deletions.
diff --git a/catalystwan/api/policy_api.py b/catalystwan/api/policy_api.py
@@ -66,6 +66,7 @@
 from catalystwan.endpoints.configuration.policy.list.ipv6_prefix import ConfigurationPolicyIPv6PrefixList
 from catalystwan.endpoints.configuration.policy.list.local_app import ConfigurationPolicyLocalAppList, LocalAppListInfo
 from catalystwan.endpoints.configuration.policy.list.local_domain import ConfigurationPolicyLocalDomainList
+from catalystwan.endpoints.configuration.policy.list.media_profile import ConfigurationPolicyMediaProfileList
 from catalystwan.endpoints.configuration.policy.list.mirror import ConfigurationPolicyMirrorList, MirrorListInfo
 from catalystwan.endpoints.configuration.policy.list.policer import ConfigurationPolicyPolicerClassList, PolicerListInfo
 from catalystwan.endpoints.configuration.policy.list.port import ConfigurationPolicyPortList, PortListInfo
@@ -84,6 +85,10 @@
 from catalystwan.endpoints.configuration.policy.list.sla import ConfigurationPolicySLAClassList, SLAClassListInfo
 from catalystwan.endpoints.configuration.policy.list.threat_grid_api_key import ConfigurationPolicyThreatGridApiKeyList
 from catalystwan.endpoints.configuration.policy.list.tloc import ConfigurationPolicyTLOCList, TLOCListInfo
+from catalystwan.endpoints.configuration.policy.list.translation_profile import (
+    ConfigurationPolicyTranslationProfileList,
+)
+from catalystwan.endpoints.configuration.policy.list.translation_rules import ConfigurationPolicyTranslationRulesList
 from catalystwan.endpoints.configuration.policy.list.trunkgroup import ConfigurationPolicyTrunkGroupList
 from catalystwan.endpoints.configuration.policy.list.umbrella_data import ConfigurationPolicyUmbrellaDataList
 from catalystwan.endpoints.configuration.policy.list.url_allow_list import (
@@ -191,8 +196,11 @@
 from catalystwan.models.policy.list.ips_signature import IPSSignatureListInfo
 from catalystwan.models.policy.list.ipv6_prefix import IPv6PrefixListInfo
 from catalystwan.models.policy.list.local_domain import LocalDomainListInfo
+from catalystwan.models.policy.list.media_profile import MediaProfileList, MediaProfileListInfo
 from catalystwan.models.policy.list.scalable_group_tag import ScalableGroupTagList, ScalableGroupTagListInfo
 from catalystwan.models.policy.list.threat_grid_api_key import ThreatGridApiKeyList, ThreatGridApiKeyListInfo
+from catalystwan.models.policy.list.translation_profile import TranslationProfileList, TranslationProfileListInfo
+from catalystwan.models.policy.list.translation_rules import TranslationRulesList, TranslationRulesListInfo
 from catalystwan.models.policy.list.trunkgroup import TrunkGroupList, TrunkGroupListInfo
 from catalystwan.models.policy.list.umbrella_data import UmbrellaDataList, UmbrellaDataListInfo
 from catalystwan.models.policy.localized import (
@@ -233,29 +241,32 @@
     ExtendedCommunityList: ConfigurationPolicyExtendedCommunityList,
     FQDNList: ConfigurationPolicyFQDNList,
     GeoLocationList: ConfigurationPolicyGeoLocationList,
+    IdentityList: ConfigurationPolicyIdentityList,
     IPSSignatureList: ConfigurationPolicyIPSSignatureList,
     IPv6PrefixList: ConfigurationPolicyIPv6PrefixList,
     LocalAppList: ConfigurationPolicyLocalAppList,
     LocalDomainList: ConfigurationPolicyLocalDomainList,
+    MediaProfileList: ConfigurationPolicyMediaProfileList,
     MirrorList: ConfigurationPolicyMirrorList,
     PolicerList: ConfigurationPolicyPolicerClassList,
-    ScalableGroupTagList: ConfigurationPolicyScalableGroupTagList,
     PortList: ConfigurationPolicyPortList,
     PreferredColorGroupList: ConfigurationPreferredColorGroupList,
     PrefixList: ConfigurationPolicyPrefixList,
     ProtocolNameList: ConfigurationPolicyProtocolNameList,
     RegionList: ConfigurationPolicyRegionList,
+    ScalableGroupTagList: ConfigurationPolicyScalableGroupTagList,
     SiteList: ConfigurationPolicySiteList,
     SLAClassList: ConfigurationPolicySLAClassList,
     ThreatGridApiKeyList: ConfigurationPolicyThreatGridApiKeyList,
     TLOCList: ConfigurationPolicyTLOCList,
+    TranslationProfileList: ConfigurationPolicyTranslationProfileList,
+    TranslationRulesList: ConfigurationPolicyTranslationRulesList,
     TrunkGroupList: ConfigurationPolicyTrunkGroupList,
     UmbrellaDataList: ConfigurationPolicyUmbrellaDataList,
-    URLBlockList: ConfigurationPolicyURLBlockList,
     URLAllowList: ConfigurationPolicyURLAllowList,
+    URLBlockList: ConfigurationPolicyURLBlockList,
     VPNList: ConfigurationPolicyVPNList,
     ZoneList: ConfigurationPolicyZoneList,
-    IdentityList: ConfigurationPolicyIdentityList,
 }
 
 POLICY_DEFINITION_ENDPOINTS_MAP: Mapping[type, type] = {
@@ -485,6 +496,10 @@ def get(self, type: Type[LocalAppList]) -> DataSequence[LocalAppListInfo]:
     def get(self, type: Type[LocalDomainList]) -> DataSequence[LocalDomainListInfo]:
         ...
 
+    @overload
+    def get(self, type: Type[MediaProfileList]) -> DataSequence[MediaProfileListInfo]:
+        ...
+
     @overload
     def get(self, type: Type[MirrorList]) -> DataSequence[MirrorListInfo]:
         ...
@@ -529,6 +544,14 @@ def get(self, type: Type[ThreatGridApiKeyList]) -> DataSequence[ThreatGridApiKey
     def get(self, type: Type[TLOCList]) -> DataSequence[TLOCListInfo]:
         ...
 
+    @overload
+    def get(self, type: Type[TranslationProfileList]) -> DataSequence[TranslationProfileListInfo]:
+        ...
+
+    @overload
+    def get(self, type: Type[TranslationRulesList]) -> DataSequence[TranslationRulesListInfo]:
+        ...
+
     @overload
     def get(self, type: Type[TrunkGroupList]) -> DataSequence[TrunkGroupListInfo]:
         ...
@@ -627,6 +650,10 @@ def get(self, type: Type[LocalAppList], id: UUID) -> LocalAppListInfo:
     def get(self, type: Type[LocalDomainList], id: UUID) -> LocalDomainListInfo:
         ...
 
+    @overload
+    def get(self, type: Type[MediaProfileList], id: UUID) -> MediaProfileListInfo:
+        ...
+
     @overload
     def get(self, type: Type[MirrorList], id: UUID) -> MirrorListInfo:
         ...
@@ -671,6 +698,14 @@ def get(self, type: Type[ThreatGridApiKeyList], id: UUID) -> ThreatGridApiKeyLis
     def get(self, type: Type[TLOCList], id: UUID) -> TLOCListInfo:
         ...
 
+    @overload
+    def get(self, type: Type[TranslationProfileList], id: UUID) -> TranslationProfileListInfo:
+        ...
+
+    @overload
+    def get(self, type: Type[TranslationRulesList], id: UUID) -> TranslationRulesListInfo:
+        ...
+
     @overload
     def get(self, type: Type[TrunkGroupList], id: UUID) -> TrunkGroupListInfo:
         ...

diff --git a/catalystwan/endpoints/endpoints_container.py b/catalystwan/endpoints/endpoints_container.py
@@ -63,6 +63,7 @@
 from catalystwan.endpoints.configuration.policy.list.ipv6_prefix import ConfigurationPolicyIPv6PrefixList
 from catalystwan.endpoints.configuration.policy.list.local_app import ConfigurationPolicyLocalAppList
 from catalystwan.endpoints.configuration.policy.list.local_domain import ConfigurationPolicyLocalDomainList
+from catalystwan.endpoints.configuration.policy.list.media_profile import ConfigurationPolicyMediaProfileList
 from catalystwan.endpoints.configuration.policy.list.mirror import ConfigurationPolicyMirrorList
 from catalystwan.endpoints.configuration.policy.list.policer import ConfigurationPolicyPolicerClassList
 from catalystwan.endpoints.configuration.policy.list.port import ConfigurationPolicyPortList
@@ -73,6 +74,10 @@
 from catalystwan.endpoints.configuration.policy.list.site import ConfigurationPolicySiteList
 from catalystwan.endpoints.configuration.policy.list.sla import ConfigurationPolicySLAClassList
 from catalystwan.endpoints.configuration.policy.list.tloc import ConfigurationPolicyTLOCList
+from catalystwan.endpoints.configuration.policy.list.translation_profile import (
+    ConfigurationPolicyTranslationProfileList,
+)
+from catalystwan.endpoints.configuration.policy.list.translation_rules import ConfigurationPolicyTranslationRulesList
 from catalystwan.endpoints.configuration.policy.list.trunkgroup import ConfigurationPolicyTrunkGroupList
 from catalystwan.endpoints.configuration.policy.list.url_allow_list import ConfigurationPolicyURLAllowList
 from catalystwan.endpoints.configuration.policy.list.url_block_list import ConfigurationPolicyURLBlockList
@@ -129,6 +134,7 @@ def __init__(self, session: ManagerSession):
         self.ipv6_prefix = ConfigurationPolicyIPv6PrefixList(session)
         self.local_app = ConfigurationPolicyLocalAppList(session)
         self.local_domain = ConfigurationPolicyLocalDomainList(session)
+        self.media_profile = ConfigurationPolicyMediaProfileList(session)
         self.mirror = ConfigurationPolicyMirrorList(session)
         self.policer = ConfigurationPolicyPolicerClassList(session)
         self.port = ConfigurationPolicyPortList(session)
@@ -139,6 +145,8 @@ def __init__(self, session: ManagerSession):
         self.site = ConfigurationPolicySiteList(session)
         self.sla = ConfigurationPolicySLAClassList(session)
         self.tloc = ConfigurationPolicyTLOCList(session)
+        self.translation_profile = ConfigurationPolicyTranslationProfileList(session)
+        self.translation_rules = ConfigurationPolicyTranslationRulesList(session)
         self.trunkgroup = ConfigurationPolicyTrunkGroupList(session)
         self.url_allow_list = ConfigurationPolicyURLAllowList(session)
         self.url_block_list = ConfigurationPolicyURLBlockList(session)

diff --git a/catalystwan/models/policy/__init__.py b/catalystwan/models/policy/__init__.py
@@ -29,6 +29,7 @@
 from catalystwan.models.policy.list.ipv6_prefix import IPv6PrefixList, IPv6PrefixListInfo
 from catalystwan.models.policy.list.local_app import LocalAppList, LocalAppListInfo
 from catalystwan.models.policy.list.local_domain import LocalDomainList, LocalDomainListInfo
+from catalystwan.models.policy.list.media_profile import MediaProfileList
 from catalystwan.models.policy.list.mirror import MirrorList, MirrorListInfo
 from catalystwan.models.policy.list.policer import PolicerList, PolicerListInfo
 from catalystwan.models.policy.list.port import PortList, PortListInfo
@@ -41,6 +42,8 @@
 from catalystwan.models.policy.list.sla import SLAClassList, SLAClassListInfo
 from catalystwan.models.policy.list.threat_grid_api_key import ThreatGridApiKeyList, ThreatGridApiKeyListInfo
 from catalystwan.models.policy.list.tloc import TLOCList, TLOCListInfo
+from catalystwan.models.policy.list.translation_profile import TranslationProfileList
+from catalystwan.models.policy.list.translation_rules import TranslationRulesList
 from catalystwan.models.policy.list.trunkgroup import TrunkGroupList, TrunkGroupListInfo
 from catalystwan.models.policy.list.umbrella_data import UmbrellaDataList, UmbrellaDataListInfo
 from catalystwan.models.policy.list.url import URLAllowList, URLAllowListInfo, URLBlockList, URLBlockListInfo
@@ -136,6 +139,7 @@
         IPv6PrefixList,
         LocalAppList,
         LocalDomainList,
+        MediaProfileList,
         MirrorList,
         PolicerList,
         PortList,
@@ -148,6 +152,8 @@
         SLAClassList,
         ThreatGridApiKeyList,
         TLOCList,
+        TranslationProfileList,
+        TranslationRulesList,
         TrunkGroupList,
         UmbrellaDataList,
         URLAllowList,