diff --git a/gso/workflows/vrf/create_vrf.py b/gso/workflows/vrf/create_vrf.py index e0af02280063a2441916f74b46ba379315ba80b0..89b39487d143f9d613bc8a85a50c5da3c90bc0b8 100644 --- a/gso/workflows/vrf/create_vrf.py +++ b/gso/workflows/vrf/create_vrf.py @@ -26,10 +26,11 @@ def initial_input_form_generator(product_name: str) -> FormGenerator: vrf_name: str route_distinguisher: str route_target: str + vrf_as_number: int user_input = yield CreateVRFForm user_input = user_input.model_dump() - summary_fields = ["vrf_name", "route_distinguisher", "route_target"] + summary_fields = ["vrf_name", "route_distinguisher", "route_target", "vrf_as_number"] yield from create_summary_form(user_input, product_name, summary_fields) return user_input @@ -52,11 +53,13 @@ def initialize_subscription( vrf_name: str, route_distinguisher: str, route_target: str, + vrf_as_number: int, ) -> State: """Initialise the subscription object in the service database.""" subscription.vrf.vrf_name = vrf_name subscription.vrf.route_distinguisher = route_distinguisher subscription.vrf.route_target = route_target + subscription.vrf.vrf_as_number = vrf_as_number subscription.description = f"VRF {vrf_name}" return {"subscription": subscription}