Read Segment Header API Key

add_kinesis_stream.py

import boto3
import json

def lambda_handler(event, context):
  ...
  str_event = json.dumps(event)
  partition_key = event['userId']
  stream_name = 'Segment'
  try:
      client = boto3.client('kinesis')
      client.put_record(
          StreamName=stream_name,
          Data=str.encode(str_event),
          PartitionKey=partition_key,
      )
      # Return 202 to indicate that the event is received but not processed
      response = {
          'statusCode': '202',
          'headers': {
              'Content-Type': 'application/json',
          }
      }
  except ProvisionedThroughputExceededException as e:
    response = {
      'statusCode': '429',
      'headers': {
          'Content-Type': 'application/json',
      }
    }
  except Exception as e:
    print('Exception occurred: {}'.format(e))
    response = {
      'statusCode': '500',
      'headers': {
          'Content-Type': 'application/json',
      }
    }
  return response