diff --git a/etl/hubspot/scripts/scraper/main.py b/etl/hubspot/scripts/scraper/main.py index 56ac8b9a5..89c49b867 100644 --- a/etl/hubspot/scripts/scraper/main.py +++ b/etl/hubspot/scripts/scraper/main.py @@ -140,13 +140,7 @@ def handler(body: dict[str, Any], context: Any) -> None: f"Triggering Abri flows {abri_message['flows']} for HubSpot " f"deal ID {hubspot_deal_id}" ) - response = sqs_client.send_message( - QueueUrl=get_settings().ABRI_SQS_URL, - MessageBody=json.dumps(abri_message), - ) - logger.info( - f"Sent message to Abri queue. MessageId: {response['MessageId']}" - ) + _trigger_abri_lambda(sqs_client, abri_message) print("done") @@ -169,6 +163,14 @@ def _trigger_magicplan_fetcher( logger.info(f"Sent message to MagicPlan queue. MessageId: {response['MessageId']}") +def _trigger_abri_lambda(sqs_client: Any, abri_message: Dict[str, Any]) -> None: + response = sqs_client.send_message( + QueueUrl=get_settings().ABRI_SQS_URL, + MessageBody=json.dumps(abri_message), + ) + logger.info(f"Sent message to Abri queue. MessageId: {response['MessageId']}") + + def _trigger_pashub_fetcher( sqs_client: Any, deal_id: str, hubspot_deal: Dict[str, str] ) -> None: