Merge pull request #504 from CiscoDevNet/dev/ux-migration-models-fix

model fixes

catalystwan/models/configuration/config_migration.py

@@ -8,6 +8,7 @@
 from catalystwan.models.configuration.feature_profile.common import FeatureProfileCreationPayload
 from catalystwan.models.configuration.feature_profile.sdwan.policy_object import AnyPolicyObjectParcel
 from catalystwan.models.configuration.feature_profile.sdwan.system import AnySystemParcel
+from catalystwan.models.configuration.topology_group import TopologyGroup
 from catalystwan.models.policy import (
     AnyPolicyDefinition,
     AnyPolicyList,
@@ -69,9 +70,11 @@ class UX1Config(BaseModel):
 
 
 class UX2Config(BaseModel):
-    # All UX2 Configuration items - Mega Model
     # All UX2 Configuration items - Mega Model
     model_config = ConfigDict(populate_by_name=True)
+    topology_groups: List[TopologyGroup] = Field(
+        default=[], serialization_alias="topologyGroups", validation_alias="topologyGroups"
+    )
     config_groups: List[ConfigGroup] = Field(
         default=[], serialization_alias="configurationGroups", validation_alias="configurationGroups"
     )

catalystwan/models/configuration/topology_group.py

@@ -0,0 +1,13 @@
+from typing import List, Literal, Optional
+from uuid import UUID
+
+from pydantic import BaseModel, Field
+
+
+class TopologyGroup(BaseModel):
+    name: str
+    solution: Literal["sdwan"] = "sdwan"
+    profiles: List[UUID] = []
+    from_topology_group: Optional[UUID] = Field(
+        default=None, serialization_alias="fromTopologyGroup", validation_alias="fromTopologyGroup"
+    )

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "catalystwan"
-version = "0.31.0dev3"
+version = "0.31.0dev4"
 description = "Cisco Catalyst WAN SDK for Python"
 authors = ["kagorski <[email protected]>"]
 readme = "README.md"