From 1ca8bec992af7116d96c545efe8049047950307c Mon Sep 17 00:00:00 2001
From: Neda Moeini <neda.moeini@geant.org>
Date: Wed, 27 Nov 2024 15:55:19 +0100
Subject: [PATCH] Update create VRF WF to add AS number.

---
 gso/workflows/vrf/create_vrf.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/gso/workflows/vrf/create_vrf.py b/gso/workflows/vrf/create_vrf.py
index e0af0228..89b39487 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}
-- 
GitLab