retry.py

MAX_RETRY_COUNT = 7
DELAY_SECONDS = 5


@exception_handler
def handler(event, context):
    if 'Records' in event:  # SQS
        records = event.get('Records', [])
        logger.info(f'Received events from SQS with {len(records)} records')

        for record in records:
            payload = json.loads(record['body'])
            event_detail = payload[DETAIL]
            event_detail_type = payload[DETAIL_TYPE]
            _handle_event(event_detail_type, event_detail)
    else:  # EventBridge
        logger.info(f'Received event from EventBridge: {event}')
        event_detail_type = event[DETAIL_TYPE]
        event_detail = event[DETAIL]
        _handle_event(event_detail_type, event_detail)