IoT-II-2023_iot-core-ws-lambda_yaira

import math
import boto3
from decimal import Decimal
import json

# Get dynamodb resource
dynamodb = boto3.resource('dynamodb')

def lambda_handler(event, context):
    
    # JSON object from IoT core has timestamp field
    if 'timestamp' in event:
        # Get table
        table = dynamodb.Table('princesitayaira')
        
        # Put item into table
        response = table.put_item(
        Item={
        'id_key': "IoT", # Primary key
        'timestamp': event['timestamp'], # Sort key
        'sensor': event['sensor']
        }
        )
        
        # Print put_item response
        print(response)
        
        # Get recently written item
        response = table.get_item(
        Key={'id_key': "IoT", 'timestamp': event['timestamp']}
        )
        
        # Print get_item response
        print(response)
        
        # Print table scan results
        #print(table.scan()['Items'])
        
        table = dynamodb.Table('ws_connection_id')
   
        response = table.get_item(
        Key={'id_key': "connectionid"}
        )
        print(response)
        
        if 'Item' in response:
            api_client = boto3.client('apigatewaymanagementapi',endpoint_url='https://1qjubqugf6.execute-api.us-east-1.amazonaws.com/production')
            connectionId=response['Item']['id']
            api_client.post_to_connection(ConnectionId=connectionId, Data=json.dumps({'sensor': event['sensor']}))
            print(connectionId)
            
        
        # Return
        return "DB updated"