import boto3 import os ddb = boto3.client('dynamodb') table_name = os.environ.get('TABLE_NAME') def handler(event, context): connection_id = event['requestContext']['connectionId'] delete_connection_id(connection_id) return { 'statusCode': 200 } def delete_connection_id(connection_id): ddb.delete_item( TableName=table_name, Key={ 'connectionId': {'S': connection_id} } )