Remove deprecated function, use class method. Minor change: use real …

…uuid in config group from device template
CiscoDevNet · Mar 28, 2024 · 69580a3 · 69580a3
1 parent 5f7e3b6
commit 69580a3
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 85 deletions.
diff --git a/catalystwan/tests/config_migration/test_flatten_general_templates.py b/catalystwan/tests/config_migration/test_flatten_general_templates.py
diff --git a/catalystwan/utils/config_migration/device_templates/__init__.py b/catalystwan/utils/config_migration/device_templates/__init__.py
diff --git a/catalystwan/utils/config_migration/device_templates/flatten_general_templates.py b/catalystwan/utils/config_migration/device_templates/flatten_general_templates.py
diff --git a/catalystwan/workflows/config_migration.py b/catalystwan/workflows/config_migration.py
@@ -18,7 +18,6 @@
 from catalystwan.utils.config_migration.converters.feature_template import create_parcel_from_template
 from catalystwan.utils.config_migration.converters.policy.policy_lists import convert as convert_policy_list
 from catalystwan.utils.config_migration.creators.config_pusher import UX2ConfigPusher, UX2ConfigRollback
-from catalystwan.utils.config_migration.device_templates import flatten_general_templates
 from catalystwan.utils.config_migration.reverters.config_reverter import UX2ConfigReverter
 
 logger = logging.getLogger(__name__)
@@ -96,7 +95,7 @@ def transform(ux1: UX1Config) -> UX2Config:
     ux2 = UX2Config()
     # Create Feature Profiles and Config Group
     for dt in ux1.templates.device_templates:
-        templates = flatten_general_templates(dt.general_templates)
+        templates = dt.get_flattened_general_templates()
 
         # Create Feature Profiles
         fp_system_uuid = uuid4()
@@ -132,7 +131,7 @@ def transform(ux1: UX1Config) -> UX2Config:
         transformed_cg = TransformedConfigGroup(
             header=TransformHeader(
                 type="config_group",
-                origin=uuid4(),
+                origin=UUID(dt.template_id),
                 subelements=set([fp_system_uuid, fp_other_uuid]),
             ),
             config_group=ConfigGroupCreationPayload(