Move template types to coresponding converters

catalystwan/utils/config_migration/converters/feature_template/aaa.py

@@ -5,6 +5,8 @@
 
 
 class AAATemplateConverter:
+    supported_template_types = ("cisco_aaa", "cedge_aaa", "aaa")
+
     @staticmethod
     def create_parcel(name: str, description: str, template_values: dict) -> AAAParcel:
         """

catalystwan/utils/config_migration/converters/feature_template/banner.py

@@ -2,6 +2,8 @@
 
 
 class BannerTemplateConverter:
+    supported_template_types = ("cisco_banner",)
+
     @staticmethod
     def create_parcel(name: str, description: str, template_values: dict) -> BannerParcel:
         """

catalystwan/utils/config_migration/converters/feature_template/basic.py

@@ -4,6 +4,8 @@
 
 
 class SystemToBasicTemplateConverter:
+    supported_template_types = ("cisco_system", "system-vsmart", "system-vedge")
+
     @staticmethod
     def create_parcel(name: str, description: str, template_values: dict) -> BasicParcel:
         """

catalystwan/utils/config_migration/converters/feature_template/bfd.py

@@ -2,6 +2,8 @@
 
 
 class BFDTemplateConverter:
+    supported_template_types = ("cisco_bfd",)
+
     @staticmethod
     def create_parcel(name: str, description: str, template_values: dict) -> BFDParcel:
         """

catalystwan/utils/config_migration/converters/feature_template/factory_method.py

@@ -20,21 +20,20 @@
 
 logger = logging.getLogger(__name__)
 
-# TODO: Move tuples inside of template converter classes then assamble them here
-supported_parcel_converters: Dict[Any, FeatureTemplateConverter] = {
-    ("cisco_aaa", "cedge_aaa", "aaa"): AAATemplateConverter,  # type: ignore[dict-item]
-    ("cisco_bfd",): BFDTemplateConverter,  # type: ignore[dict-item]
-    ("cisco_logging", "logging"): LoggingTemplateConverter,  # type: ignore[dict-item]
-    (
-        "cisco_security",
-        "security",
-        "security-vsmart",
-        "security-vedge",
-    ): SecurityTemplateConverter,  # type: ignore[dict-item]
-    ("cisco_omp", "omp-vedge", "omp-vsmart"): OMPTemplateConverter,
-    ("cedge_global",): GlobalTemplateConverter,
-    ("cisco_banner",): BannerTemplateConverter,
-    ("cisco_system", "system-vsmart", "system-vedge"): SystemToBasicTemplateConverter,
+available_converters = [
+    AAATemplateConverter,
+    BannerTemplateConverter,
+    SecurityTemplateConverter,
+    SystemToBasicTemplateConverter,
+    BFDTemplateConverter,
+    GlobalTemplateConverter,
+    LoggingTemplateConverter,
+    OMPTemplateConverter,
+]
+
+
+supported_parcel_converters: Dict[Any, Any] = {
+    converter.supported_template_types: converter for converter in available_converters  # type: ignore
 }
 
 

catalystwan/utils/config_migration/converters/feature_template/global_.py

@@ -2,6 +2,8 @@
 
 
 class GlobalTemplateConverter:
+    supported_template_types = ("cedge_global",)
+
     @staticmethod
     def create_parcel(name: str, description: str, template_values: dict) -> GlobalParcel:
         """

catalystwan/utils/config_migration/converters/feature_template/logging_.py

@@ -6,6 +6,8 @@
 
 
 class LoggingTemplateConverter:
+    supported_template_types = ("cisco_logging", "logging")
+
     @staticmethod
     def create_parcel(name: str, description: str, template_values: dict) -> LoggingParcel:
         """

catalystwan/utils/config_migration/converters/feature_template/omp.py

@@ -3,6 +3,8 @@
 
 
 class OMPTemplateConverter:
+    supported_template_types = ("cisco_omp", "omp-vedge", "omp-vsmart")
+
     @staticmethod
     def create_parcel(name: str, description: str, template_values: dict) -> OMPParcel:
         """

catalystwan/utils/config_migration/converters/feature_template/security.py

@@ -6,6 +6,13 @@
 
 
 class SecurityTemplateConverter:
+    supported_template_types = (
+        "cisco_security",
+        "security",
+        "security-vsmart",
+        "security-vedge",
+    )
+
     @staticmethod
     def create_parcel(name: str, description: str, template_values: dict) -> SecurityParcel:
         """