catalystwan 0.41.2.dev9__py3-none-any.whl → 0.41.3__py3-none-any.whl

catalystwan/endpoints/configuration/policy/definition/zone_based_firewall.py

@@ -4,14 +4,11 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyDefinitionEndpoints
-from catalystwan.models.policy.definition.zone_based_firewall import (
-    ZoneBasedFWPolicy,
-    ZoneBasedFWPolicyEditPayload,
-    ZoneBasedFWPolicyGetResponse,
-)
+from catalystwan.models.policy.definitions.zone_based_firewall import ZoneBasedFWPolicy
 from catalystwan.models.policy.policy_definition import (
     PolicyDefinitionEditResponse,
+    PolicyDefinitionEndpoints,
+    PolicyDefinitionGetResponse,
     PolicyDefinitionId,
     PolicyDefinitionInfo,
     PolicyDefinitionPreview,
@@ -19,6 +16,14 @@ from catalystwan.models.policy.policy_definition import (
 from catalystwan.typed_list import DataSequence
 
 
+class ZoneBasedFWPolicyEditPayload(ZoneBasedFWPolicy, PolicyDefinitionId):
+    pass
+
+
+class ZoneBasedFWPolicyGetResponse(ZoneBasedFWPolicy, PolicyDefinitionGetResponse):
+    pass
+
+
 class ConfigurationPolicyZoneBasedFirewallDefinition(APIEndpoints, PolicyDefinitionEndpoints):
     @post("/template/policy/definition/zonebasedfw")
     def create_policy_definition(self, payload: ZoneBasedFWPolicy) -> PolicyDefinitionId:

catalystwan/endpoints/configuration/policy/list/app.py

@@ -4,12 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.app import AppList, AppListEditPayload, AppListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import AppList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class AppListEditPayload(AppList, PolicyListId):
+    pass
+
+
+class AppListInfo(AppList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyApplicationList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/app")
     def create_policy_list(self, payload: AppList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/app_probe.py

@@ -4,16 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.app_probe import (
-    AppProbeClassList,
-    AppProbeClassListEditPayload,
-    AppProbeClassListInfo,
+from catalystwan.models.policy.lists import AppProbeClassList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
 )
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
 from catalystwan.typed_list import DataSequence
 
 
+class AppProbeClassListEditPayload(AppProbeClassList, PolicyListId):
+    pass
+
+
+class AppProbeClassListInfo(AppProbeClassList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyAppProbeClassList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/appprobe")
     def create_policy_list(self, payload: AppProbeClassList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/as_path.py

@@ -4,12 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.as_path import ASPathList, ASPathListEditPayload, ASPathListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import ASPathList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class ASPathListEditPayload(ASPathList, PolicyListId):
+    pass
+
+
+class ASPathListInfo(ASPathList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyASPathList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/aspath")
     def create_policy_list(self, payload: ASPathList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/class_map.py

@@ -4,12 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.class_map import ClassMapList, ClassMapListEditPayload, ClassMapListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import ClassMapList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class ClassMapListEditPayload(ClassMapList, PolicyListId):
+    pass
+
+
+class ClassMapListInfo(ClassMapList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyForwardingClassList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/class")
     def create_policy_list(self, payload: ClassMapList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/color.py

@@ -4,12 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.color import ColorList, ColorListEditPayload, ColorListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import ColorList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class ColorListEditPayload(ColorList, PolicyListId):
+    pass
+
+
+class ColorListInfo(ColorList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyColorList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/color")
     def create_policy_list(self, payload: ColorList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/community.py

@@ -4,12 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.communities import CommunityList, CommunityListEditPayload, CommunityListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import CommunityList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class CommunityListEditPayload(CommunityList, PolicyListId):
+    pass
+
+
+class CommunityListInfo(CommunityList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyCommunityList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/community")
     def create_policy_list(self, payload: CommunityList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/data_ipv6_prefix.py

@@ -4,16 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.data_ipv6_prefix import (
-    DataIPv6PrefixList,
-    DataIPv6PrefixListEditPayload,
-    DataIPv6PrefixListInfo,
+from catalystwan.models.policy.lists import DataIPv6PrefixList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
 )
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
 from catalystwan.typed_list import DataSequence
 
 
+class DataIPv6PrefixListEditPayload(DataIPv6PrefixList, PolicyListId):
+    pass
+
+
+class DataIPv6PrefixListInfo(DataIPv6PrefixList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyDataIPv6PrefixList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/dataipv6prefix")
     def create_policy_list(self, payload: DataIPv6PrefixList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/data_prefix.py

@@ -4,12 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.data_prefix import DataPrefixList, DataPrefixListEditPayload, DataPrefixListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import DataPrefixList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class DataPrefixListEditPayload(DataPrefixList, PolicyListId):
+    pass
+
+
+class DataPrefixListInfo(DataPrefixList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyDataPrefixList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/dataprefix")
     def create_policy_list(self, payload: DataPrefixList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/expanded_community.py

@@ -4,16 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.communities import (
-    ExpandedCommunityList,
-    ExpandedCommunityListEditPayload,
-    ExpandedCommunityListInfo,
+from catalystwan.models.policy.lists import ExpandedCommunityList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
 )
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
 from catalystwan.typed_list import DataSequence
 
 
+class ExpandedCommunityListEditPayload(ExpandedCommunityList, PolicyListId):
+    pass
+
+
+class ExpandedCommunityListInfo(ExpandedCommunityList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyExpandedCommunityList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/expandedcommunity")
     def create_policy_list(self, payload: ExpandedCommunityList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/fqdn.py

@@ -4,12 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.fqdn import FQDNList, FQDNListEditPayload, FQDNListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import FQDNList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class FQDNListEditPayload(FQDNList, PolicyListId):
+    pass
+
+
+class FQDNListInfo(FQDNList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyFQDNList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/fqdn")
     def create_policy_list(self, payload: FQDNList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/geo_location.py

@@ -4,12 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.geo_location import GeoLocationList, GeoLocationListEditPayload, GeoLocationListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import GeoLocationList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class GeoLocationListEditPayload(GeoLocationList, PolicyListId):
+    pass
+
+
+class GeoLocationListInfo(GeoLocationList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyGeoLocationList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/geolocation")
     def create_policy_list(self, payload: GeoLocationList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/ips_signature.py

@@ -4,16 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.ips_signature import (
-    IPSSignatureList,
-    IPSSignatureListEditPayload,
-    IPSSignatureListInfo,
+from catalystwan.models.policy.lists import IPSSignatureList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
 )
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
 from catalystwan.typed_list import DataSequence
 
 
+class IPSSignatureListEditPayload(IPSSignatureList, PolicyListId):
+    pass
+
+
+class IPSSignatureListInfo(IPSSignatureList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyIPSSignatureList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/ipssignature")
     def create_policy_list(self, payload: IPSSignatureList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/ipv6_prefix.py

@@ -4,12 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.ipv6_prefix import IPv6PrefixList, IPv6PrefixListEditPayload, IPv6PrefixListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import IPv6PrefixList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class IPv6PrefixListEditPayload(IPv6PrefixList, PolicyListId):
+    pass
+
+
+class IPv6PrefixListInfo(IPv6PrefixList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyIPv6PrefixList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/ipv6prefix")
     def create_policy_list(self, payload: IPv6PrefixList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/local_app.py

@@ -4,12 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.local_app import LocalAppList, LocalAppListEditPayload, LocalAppListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import LocalAppList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class LocalAppListEditPayload(LocalAppList, PolicyListId):
+    pass
+
+
+class LocalAppListInfo(LocalAppList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyLocalAppList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/localapp")
     def create_policy_list(self, payload: LocalAppList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/local_domain.py

@@ -4,12 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.local_domain import LocalDomainList, LocalDomainListEditPayload, LocalDomainListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import LocalDomainList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class LocalDomainListEditPayload(LocalDomainList, PolicyListId):
+    pass
+
+
+class LocalDomainListInfo(LocalDomainList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyLocalDomainList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/localdomain")
     def create_policy_list(self, payload: LocalDomainList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/mirror.py

@@ -4,12 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.mirror import MirrorList, MirrorListEditPayload, MirrorListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import MirrorList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class MirrorListEditPayload(MirrorList, PolicyListId):
+    pass
+
+
+class MirrorListInfo(MirrorList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyMirrorList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/mirror")
     def create_policy_list(self, payload: MirrorList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/policer.py

@@ -4,12 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.policer import PolicerList, PolicerListEditPayload, PolicerListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import PolicerList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class PolicerListEditPayload(PolicerList, PolicyListId):
+    pass
+
+
+class PolicerListInfo(PolicerList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyPolicerClassList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/policer")
     def create_policy_list(self, payload: PolicerList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/port.py

@@ -4,12 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.port import PortList, PortListEditPayload, PortListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import PortList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class PortListEditPayload(PortList, PolicyListId):
+    pass
+
+
+class PortListInfo(PortList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyPortList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/port")
     def create_policy_list(self, payload: PortList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/preferred_color_group.py

@@ -4,16 +4,25 @@
 from uuid import UUID
 
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.preferred_color_group import (
-    PreferredColorGroupList,
-    PreferredColorGroupListEditPayload,
-    PreferredColorGroupListInfo,
+from catalystwan.models.policy.lists import PreferredColorGroupList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
 )
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
 from catalystwan.typed_list import DataSequence
 
 
+class PreferredColorGroupListEditPayload(PreferredColorGroupList, PolicyListId):
+    pass
+
+
+class PreferredColorGroupListInfo(PreferredColorGroupList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPreferredColorGroupList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/preferredcolorgroup")
     def create_policy_list(self, payload: PreferredColorGroupList) -> PolicyListId:

catalystwan/endpoints/configuration/policy/list/prefix.py

@@ -3,14 +3,26 @@
 # mypy: disable-error-code="empty-body"
 from uuid import UUID
 
-from catalystwan.api.templates.models.cisco_vpn_model import PrefixList
 from catalystwan.endpoints import APIEndpoints, delete, get, post, put
-from catalystwan.endpoints.configuration.policy.abstractions import PolicyListEndpoints
-from catalystwan.models.policy.list.prefix import PrefixListEditPayload, PrefixListInfo
-from catalystwan.models.policy.policy_list import InfoTag, PolicyListId, PolicyListPreview
+from catalystwan.models.policy.lists import PrefixList
+from catalystwan.models.policy.policy_list import (
+    InfoTag,
+    PolicyListEndpoints,
+    PolicyListId,
+    PolicyListInfo,
+    PolicyListPreview,
+)
 from catalystwan.typed_list import DataSequence
 
 
+class PrefixListEditPayload(PrefixList, PolicyListId):
+    pass
+
+
+class PrefixListInfo(PrefixList, PolicyListInfo):
+    pass
+
+
 class ConfigurationPolicyPrefixList(APIEndpoints, PolicyListEndpoints):
     @post("/template/policy/list/prefix")
     def create_policy_list(self, payload: PrefixList) -> PolicyListId: