Add customer_code field to DecommissionWorkerRequest and update decommission function
taylorwalton committed
Aug 6, 2025 at 14:29 UTC
31a0f976ca7b482163c1c5b41ee1affef7327925
2 files changed
+9
-1
backend/app/customer_provisioning/schema/wazuh_worker.py
+5
@@ -97,6 +97,11 @@ class DecommissionWorkerRequest(BaseModel):
97
example="SOCFortress",
98
description="The name of the customer",
99
)
100
+ customer_code: str = Field(
101
+ ...,
102
+ example="socfortress",
103
+ description="The code of the customer",
104
+ )
105
portainer_deployment: Optional[bool] = Field(
106
None,
107
example=True,
backend/app/customer_provisioning/services/decommission.py
+4
-1
@@ -137,7 +137,10 @@ async def decomission_wazuh_customer(
137
138
# Decommission Wazuh Worker
139
await decommission_wazuh_worker(
140
- request=DecommissionWorkerRequest(customer_name=customer_meta.customer_name),
140
+ request=DecommissionWorkerRequest(
141
+ customer_name=customer_meta.customer_name,
142
+ customer_code=customer_meta.customer_code,
143
+ ),
144
session=session,
145
)
146