Improved promote P to PE workflow.

gso/utils/helpers.py

@@ -306,6 +306,7 @@ def generate_fqdn(hostname: str, site_name: str, country_code: str) -> str:
 def generate_inventory_for_active_routers(
     router_role: RouterRole,
     exclude_routers: list[str] | None = None,
+    router_vendor: Vendor | None = None,
 ) -> dict:
     """Generate an Ansible-compatible inventory for executing playbooks.
 
@@ -313,6 +314,7 @@ def generate_inventory_for_active_routers(
 
     :param RouterRole router_role: The role of the routers to include in the inventory.
     :param list exclude_routers: List of routers to exclude from the inventory.
+    :param Vendor router_vendor: The vendor of the routers to include in the inventory.
     :return: A dictionary representing the inventory of active routers.
     :rtype: dict[str, Any]
     """
@@ -328,7 +330,9 @@ def generate_inventory_for_active_routers(
                     "vendor": str(router.router.vendor),
                 }
                 for router in all_routers
-                if router.router.router_role == router_role and router.router.router_fqdn not in exclude_routers
+                if router.router.router_role == router_role
+                and router.router.router_fqdn not in exclude_routers
+                and (router_vendor is None or router.router.vendor == router_vendor)
             }
         }
     }

gso/workflows/__init__.py

@@ -20,6 +20,7 @@ WF_USABLE_MAP.update({
     "terminate_site": ALL_ALIVE_STATES,
     "terminate_router": ALL_ALIVE_STATES,
     "terminate_iptrunk": ALL_ALIVE_STATES,
+    "promote_p_to_pe": [SubscriptionLifecycle.ACTIVE],
 })
 
 #  IP trunk workflows