precommit-fixes
taylorwalton committed
Feb 25, 2025 at 18:03 UTC
003c8bcc2c36c93041b427e1bcc69d1229cb99da
1 file changed
+18
-10
backend/app/incidents/services/incident_alert.py
+18
-10
@@ -513,14 +513,12 @@ async def create_alert_full(alert_payload: CreatedAlertPayload, customer_code: s
513
alert_context_id = (
514
await create_alert_context_payload(source=alert_payload.source, alert_payload=alert_payload.alert_context_payload, session=session)
515
).id
516
- asset = (
517
- await create_asset_context_payload(
518
- customer_code=customer_code,
519
- asset_payload=alert_payload,
520
- alert_context_id=alert_context_id,
521
- alert_id=alert_id,
522
- session=session,
523
- )
516
+ asset = await create_asset_context_payload(
517
+ customer_code=customer_code,
518
+ asset_payload=alert_payload,
519
+ alert_context_id=alert_context_id,
520
+ alert_id=alert_id,
521
+ session=session,
522
)
523
if alert_payload.ioc_payload is not None:
524
ioc_id = (
@@ -541,9 +539,19 @@ async def create_alert_full(alert_payload: CreatedAlertPayload, customer_code: s
539
logger.info(f"Creating alert for customer code {customer_code} with alert context ID {alert_context_id} and asset ID {asset.id}")
540
alert_payload.alert_id = alert_id
541
if asset is not None:
544
- await handle_customer_notifications(customer_code=customer_code, asset_name=asset.asset_name, alert_payload=alert_payload, session=session)
542
+ await handle_customer_notifications(
543
+ customer_code=customer_code,
544
+ asset_name=asset.asset_name,
545
+ alert_payload=alert_payload,
546
+ session=session,
547
+ )
548
else:
546
- await handle_customer_notifications(customer_code=customer_code, asset_name="No asset found", alert_payload=alert_payload, session=session)
549
+ await handle_customer_notifications(
550
+ customer_code=customer_code,
551
+ asset_name="No asset found",
552
+ alert_payload=alert_payload,
553
+ session=session,
554
+ )
555
556
await add_alert_to_document(CreateAlertRequest(index_name=alert_payload.index_name, alert_id=alert_payload.index_id), alert_id)
557