From d8b2d6aa1c41489ab4ae95a0ddc09cd624360dd5 Mon Sep 17 00:00:00 2001
From: Simone Spinelli <simone.spinelli@geant.org>
Date: Wed, 5 Apr 2023 12:02:37 +0200
Subject: [PATCH] Try to fix device vendor

---
 .../workflows/device/create_device.py                  | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/geant_service_orchestrator/workflows/device/create_device.py b/geant_service_orchestrator/workflows/device/create_device.py
index 0e76bafc..37ddb652 100644
--- a/geant_service_orchestrator/workflows/device/create_device.py
+++ b/geant_service_orchestrator/workflows/device/create_device.py
@@ -11,20 +11,22 @@ from orchestrator.workflow import done, init, step, workflow
 from orchestrator.workflows.steps import resync, set_status
 from orchestrator.workflows.steps import store_process_subscription
 from orchestrator.workflows.utils import wrap_create_initial_input_form
-
 from products.product_types.device import DeviceInactive, DeviceProvisioning
+from products.product_types.device import DeviceVendor
+import ipaddress
 
 
 def initial_input_form_generator(product_name: str) -> FormGenerator:
-    class CreateUserForm(FormPage):
+    class CreateDeviceForm(FormPage):
         class Config:
             title = product_name
 
         fqdn: str
-        ts_address: str
+        ts_address: ipaddress.IPv4Address
         ts_port: int
+        device_vendor: DeviceVendor
 
-    user_input = yield CreateUserForm
+    user_input = yield CreateDeviceForm
 
     return user_input.dict()
 
-- 
GitLab