AWS Athena Query Via Lambda

lambda_function.py

import json
import time
import boto3

client=boto3.client("athena")

def lambda_handler(event, context):
    QUERY="SELECT * from data order by age desc limit 10;"
    DATABASE="data"
    OutputPath="s3://sri-reco-demo/Athena/"
    response=client.start_query_execution(
        QueryString=QUERY,
        QueryExecutionContext={
                'Database': DATABASE
        },
        ResultConfiguration={
                'OutputLocation': OutputPath
            
        });
    query_id=response['QueryExecutionId']
    time.sleep(5)
    query_stats=client.get_query_execution(QueryExecutionId=query_id)
    Status=query_stats['QueryExecution']['Status']['State']
        
    return {
        'statusCode': 200,
        'body': json.dumps(Status)
    }