def lambda_handler()

in source/core-api/lambda_functions/get_list_expired_tokens.py [0:0]
38 lines of code
6 McCabe index (conditional complexity)

def lambda_handler(event, context):
    """
    This function is the entry handler for Lambda.
    """
    
    print(event)
    client_event_id = deep_clean(event['queryStringParameters']['event_id'])
    headers = {
        'Content-Type': 'application/json',
        'Access-Control-Allow-Origin': '*'
    }
    if client_event_id == EVENT_ID:
        try:
            current_time = int(time.time())
            response = ddb_table.query(
                IndexName="EventExpiresIndex",
                ProjectionExpression="request_id",
                KeyConditionExpression=Key('event_id').eq(EVENT_ID) & Key('expires').lt(current_time))
            items = [item['request_id'] for item in response['Items']]
            while "LastEvaluatedKey" in response:
                response = ddb_table.query(
                    IndexName="EventExpiresIndex",
                    ProjectionExpression="request_id",
                    KeyConditionExpression=Key('event_id').eq(EVENT_ID) & Key('expires').lt(current_time),
                    ExclusiveStartKey=response["LastEvaluatedKey"])
                items.append([item['request_id'] for item in response['Items']])
            response = {
                "statusCode": 200,
                "headers": headers,
                "body": json.dumps(items)
            }
        except Exception as e:
            print(e)
            raise e                      
    else:
        response = {
            "statusCode": 400,
            "headers": headers,
            "body": json.dumps({"error": "Invalid event ID"})
        }
    print(response)
    return response