From d1baec3ab9efacefad769a16b5e9ccc12625a290 Mon Sep 17 00:00:00 2001
From: Hakan Calim <hakan.calim@fau.de>
Date: Tue, 28 Nov 2023 15:14:16 +0100
Subject: [PATCH] NAT-329: fied complexity issue from ruff

---
 gso/workflows/iptrunk/create_iptrunk.py | 17 ++++-------------
 1 file changed, 4 insertions(+), 13 deletions(-)

diff --git a/gso/workflows/iptrunk/create_iptrunk.py b/gso/workflows/iptrunk/create_iptrunk.py
index 8bb09a22..0c0ee530 100644
--- a/gso/workflows/iptrunk/create_iptrunk.py
+++ b/gso/workflows/iptrunk/create_iptrunk.py
@@ -106,12 +106,8 @@ def initial_input_form_generator(product_name: str) -> FormGenerator:
 
         @validator("side_a_ae_members", allow_reuse=True)
         def validate_interface_name_members(cls, side_a_ae_members: list[LAGMember]) -> list[LAGMember]:
-            if get_router_vendor(router_a) == RouterVendor.NOKIA:
-                ret_val = side_a_ae_members
-            else:
-                ret_val = validate_interface_name_list(side_a_ae_members)
-
-            return ret_val
+            vendor = get_router_vendor(router_a)
+            return validate_interface_name_list(side_a_ae_members, vendor)
 
     user_input_side_a = yield CreateIptrunkSideAForm
     # Remove the selected router for side A, to prevent any loops
@@ -162,13 +158,8 @@ def initial_input_form_generator(product_name: str) -> FormGenerator:
 
         @validator("side_b_ae_members", allow_reuse=True)
         def validate_interface_name_members(cls, side_b_ae_members: list[LAGMember]) -> list[LAGMember]:
-            ret_val = None
-            if get_router_vendor(router_b) == RouterVendor.NOKIA:
-                ret_val = side_b_ae_members
-            else:
-                ret_val = validate_interface_name_list(side_b_ae_members)
-
-            return ret_val
+            vendor = get_router_vendor(router_b)
+            return validate_interface_name_list(side_b_ae_members, vendor)
 
     user_input_side_b = yield CreateIptrunkSideBForm
 
-- 
GitLab