diff --git a/gso/workflows/device/create_device.py b/gso/workflows/device/create_device.py index 6e85e1659ad2361155aed956ad8c004d6fe9035d..08b204b565b2ddac96ef5050d45335f4d366e03d 100644 --- a/gso/workflows/device/create_device.py +++ b/gso/workflows/device/create_device.py @@ -6,7 +6,7 @@ from orchestrator.db.models import ProductTable, SubscriptionTable # noinspection PyProtectedMember from orchestrator.forms import FormPage -from orchestrator.forms.validators import Choice, choice_list +from orchestrator.forms.validators import Choice from orchestrator.targets import Target from orchestrator.types import FormGenerator, State from orchestrator.types import SubscriptionLifecycle, UUIDstr @@ -119,13 +119,13 @@ def get_info_from_ipam(subscription: DeviceProvisioning) -> State: @step("Initialize subscription") def initialize_subscription( - subscription: device.DeviceInactive, - hostname: str, - ts_address: ipaddress.IPv4Address, - ts_port: str, - device_vendor: device_pb.DeviceVendor, - device_site: str, - device_role: device_pb.DeviceRole, + subscription: device.DeviceInactive, + hostname: str, + ts_address: ipaddress.IPv4Address, + ts_port: str, + device_vendor: device_pb.DeviceVendor, + device_site: str, + device_role: device_pb.DeviceRole, ) -> State: subscription.device.device_ts_address = str(ts_address) subscription.device.device_ts_port = str(ts_port) @@ -152,7 +152,7 @@ def initialize_subscription( @step("Provision device [DRY RUN]") def provision_device_dry( - subscription: DeviceProvisioning, process_id: UUIDstr + subscription: DeviceProvisioning, process_id: UUIDstr ) -> State: provisioning_proxy.provision_device(subscription, process_id) @@ -170,7 +170,7 @@ def provision_device_dry( @step("Provision device [FOR REAL]") def provision_device_real( - subscription: DeviceProvisioning, process_id: UUIDstr + subscription: DeviceProvisioning, process_id: UUIDstr ) -> State: provisioning_proxy.provision_device(subscription, process_id, False) @@ -195,18 +195,18 @@ def provision_device_real( ) def create_device(): return ( - init - >> create_subscription - >> store_process_subscription(Target.CREATE) - >> initialize_subscription - >> get_info_from_ipam - >> provision_device_dry - >> await_pp_results - >> confirm_pp_results - >> provision_device_real - >> await_pp_results - >> confirm_pp_results - >> set_status(SubscriptionLifecycle.ACTIVE) - >> resync - >> done + init + >> create_subscription + >> store_process_subscription(Target.CREATE) + >> initialize_subscription + >> get_info_from_ipam + >> provision_device_dry + >> await_pp_results + >> confirm_pp_results + >> provision_device_real + >> await_pp_results + >> confirm_pp_results + >> set_status(SubscriptionLifecycle.ACTIVE) + >> resync + >> done )