IoT-II-2023-api-ws-lambda-yaira

import json
import boto3

def lambda_handler(event, context):
    dynamodb = boto3.resource('dynamodb')
    table = dynamodb.Table('ws_connection_id')
   
    if event['requestContext']['routeKey']=='$connect':
        connectionId = event['requestContext']['connectionId']
        print(event)
        
        response = table.get_item(
            Key={'id_key': "connectionid"}
            )
            
        if 'Item' in response:
            response = table.update_item(
            Key={
            'id_key': "connectionid"
            },
            UpdateExpression="set id=:id",
            ExpressionAttributeValues={
            ':id': connectionId
            },
            ReturnValues="UPDATED_NEW"
            )
        else:
            table.put_item(
            Item={'id_key': "connectionid",'id':connectionId}
            )
    if event['requestContext']['routeKey']=='$disconnect':

        
        response = table.delete_item(
            Key={'id_key': "connectionid"}
            )
        
    return {
        'statusCode': 200,
        'body': json.dumps('Hello from Lambda!')
    }