diff --git a/gso/workflows/l3_core_service/ias/create_imported_ias.py b/gso/workflows/l3_core_service/ias/create_imported_ias.py index 485c96c3b5d8690b31be22394f767435d12832d4..fb64b6077f2b24300f4c7d5c957e822dfaf585be 100644 --- a/gso/workflows/l3_core_service/ias/create_imported_ias.py +++ b/gso/workflows/l3_core_service/ias/create_imported_ias.py @@ -19,7 +19,6 @@ from gso.workflows.l3_core_service.base_create_imported_l3_core_service import ( from gso.workflows.l3_core_service.base_create_imported_l3_core_service import ( initialize_subscription, ) -from gso.workflows.l3_core_service.ias.shared import update_ias_subscription_model def initial_input_form_generator() -> FormGenerator: @@ -36,11 +35,14 @@ def initial_input_form_generator() -> FormGenerator: @step("Create subscription") -def create_subscription(partner: str) -> dict: +def create_subscription(partner: str, ias_flavor: IASFlavor) -> dict: """Create a new subscription object in the database.""" partner_id = get_partner_by_name(partner).partner_id product_id = get_product_id_by_name(ProductName.IMPORTED_IAS) subscription = ImportedIASInactive.from_product_id(product_id, partner_id) + + subscription.ias.ias_flavor = ias_flavor + return {"subscription": subscription, "subscription_id": subscription.subscription_id} @@ -54,7 +56,6 @@ def create_imported_ias() -> StepList: return ( begin >> create_subscription - >> update_ias_subscription_model >> store_process_subscription(Target.CREATE) >> initialize_subscription >> set_status(SubscriptionLifecycle.ACTIVE)