Chilkat Tools

BatchGetQueryExecution Python Example

Athena

import sys
import chilkat2

# This example requires the Chilkat API to have been previously unlocked.
# See Global Unlock Sample for sample code.

rest = chilkat2.Rest()

authAws = chilkat2.AuthAws()
authAws.AccessKey = "AWS_ACCESS_KEY"
authAws.SecretKey = "AWS_SECRET_KEY"

# Don't forget to change the region to your particular region. (Also make the same change in the call to Connect below.)
authAws.Region = "us-west-2"
authAws.ServiceName = "athena"
# SetAuthAws causes Chilkat to automatically add the following headers: Authorization, X-Amz-Date
rest.SetAuthAws(authAws)

# URL: https://athena.us-west-2.amazonaws.com/
# Use the same region as specified above.
success = rest.Connect("athena.us-west-2.amazonaws.com",443,True,True)
if (success != True):
    print("ConnectFailReason: " + str(rest.ConnectFailReason))
    print(rest.LastErrorText)
    sys.exit()

# The following code creates the JSON request body.
# The JSON created by this code is shown below.

# Use this online tool to generate code from sample JSON:
# Generate Code to Create JSON

json = chilkat2.JsonObject()
json.UpdateString("QueryExecutionIds[0]","string")

# The JSON request body created by the above code:

# {
#   "QueryExecutionIds": [
#     "string"
#   ]
# }

rest.AddHeader("Content-Type","application/x-amz-json-1.1")
rest.AddHeader("X-Amz-Target","AmazonAthena.BatchGetQueryExecution")

sbRequestBody = chilkat2.StringBuilder()
json.EmitSb(sbRequestBody)
sbResponseBody = chilkat2.StringBuilder()
success = rest.FullRequestSb("POST","/",sbRequestBody,sbResponseBody)
if (success != True):
    print(rest.LastErrorText)
    sys.exit()

respStatusCode = rest.ResponseStatusCode
print("response status code = " + str(respStatusCode))
if (respStatusCode != 200):
    print("Response Header:")
    print(rest.ResponseHeader)
    print("Response Body:")
    print(sbResponseBody.GetAsString())
    sys.exit()

jResp = chilkat2.JsonObject()
jResp.LoadSb(sbResponseBody)

# The following code parses the JSON response.
# A sample JSON response is shown below the sample code.

# Use this online tool to generate parsing code from sample JSON:
# Generate Parsing Code from JSON

i = 0
count_i = jResp.SizeOfArray("QueryExecutions")
while i < count_i :
    jResp.I = i
    EffectiveEngineVersion = jResp.StringOf("QueryExecutions[i].EngineVersion.EffectiveEngineVersion")
    SelectedEngineVersion = jResp.StringOf("QueryExecutions[i].EngineVersion.SelectedEngineVersion")
    Query = jResp.StringOf("QueryExecutions[i].Query")
    Catalog = jResp.StringOf("QueryExecutions[i].QueryExecutionContext.Catalog")
    Database = jResp.StringOf("QueryExecutions[i].QueryExecutionContext.Database")
    QueryExecutionId = jResp.StringOf("QueryExecutions[i].QueryExecutionId")
    S3AclOption = jResp.StringOf("QueryExecutions[i].ResultConfiguration.AclConfiguration.S3AclOption")
    EncryptionOption = jResp.StringOf("QueryExecutions[i].ResultConfiguration.EncryptionConfiguration.EncryptionOption")
    KmsKey = jResp.StringOf("QueryExecutions[i].ResultConfiguration.EncryptionConfiguration.KmsKey")
    ExpectedBucketOwner = jResp.StringOf("QueryExecutions[i].ResultConfiguration.ExpectedBucketOwner")
    OutputLocation = jResp.StringOf("QueryExecutions[i].ResultConfiguration.OutputLocation")
    StatementType = jResp.StringOf("QueryExecutions[i].StatementType")
    DataManifestLocation = jResp.StringOf("QueryExecutions[i].Statistics.DataManifestLocation")
    DataScannedInBytes = jResp.IntOf("QueryExecutions[i].Statistics.DataScannedInBytes")
    EngineExecutionTimeInMillis = jResp.IntOf("QueryExecutions[i].Statistics.EngineExecutionTimeInMillis")
    QueryPlanningTimeInMillis = jResp.IntOf("QueryExecutions[i].Statistics.QueryPlanningTimeInMillis")
    QueryQueueTimeInMillis = jResp.IntOf("QueryExecutions[i].Statistics.QueryQueueTimeInMillis")
    ServiceProcessingTimeInMillis = jResp.IntOf("QueryExecutions[i].Statistics.ServiceProcessingTimeInMillis")
    TotalExecutionTimeInMillis = jResp.IntOf("QueryExecutions[i].Statistics.TotalExecutionTimeInMillis")
    ErrorCategory = jResp.IntOf("QueryExecutions[i].Status.AthenaError.ErrorCategory")
    ErrorMessage = jResp.StringOf("QueryExecutions[i].Status.AthenaError.ErrorMessage")
    ErrorType = jResp.IntOf("QueryExecutions[i].Status.AthenaError.ErrorType")
    Retryable = jResp.IntOf("QueryExecutions[i].Status.AthenaError.Retryable")
    CompletionDateTime = jResp.IntOf("QueryExecutions[i].Status.CompletionDateTime")
    State = jResp.StringOf("QueryExecutions[i].Status.State")
    StateChangeReason = jResp.StringOf("QueryExecutions[i].Status.StateChangeReason")
    SubmissionDateTime = jResp.IntOf("QueryExecutions[i].Status.SubmissionDateTime")
    WorkGroup = jResp.StringOf("QueryExecutions[i].WorkGroup")
    i = i + 1

i = 0
count_i = jResp.SizeOfArray("UnprocessedQueryExecutionIds")
while i < count_i :
    jResp.I = i
    ErrorCode = jResp.StringOf("UnprocessedQueryExecutionIds[i].ErrorCode")
    ErrorMessage = jResp.StringOf("UnprocessedQueryExecutionIds[i].ErrorMessage")
    QueryExecutionId = jResp.StringOf("UnprocessedQueryExecutionIds[i].QueryExecutionId")
    i = i + 1

# A sample JSON response body parsed by the above code:

# {
#   "QueryExecutions": [
#     {
#       "EngineVersion": {
#         "EffectiveEngineVersion": "string",
#         "SelectedEngineVersion": "string"
#       },
#       "Query": "string",
#       "QueryExecutionContext": {
#         "Catalog": "string",
#         "Database": "string"
#       },
#       "QueryExecutionId": "string",
#       "ResultConfiguration": {
#         "AclConfiguration": {
#           "S3AclOption": "string"
#         },
#         "EncryptionConfiguration": {
#           "EncryptionOption": "string",
#           "KmsKey": "string"
#         },
#         "ExpectedBucketOwner": "string",
#         "OutputLocation": "string"
#       },
#       "StatementType": "string",
#       "Statistics": {
#         "DataManifestLocation": "string",
#         "DataScannedInBytes": number,
#         "EngineExecutionTimeInMillis": number,
#         "QueryPlanningTimeInMillis": number,
#         "QueryQueueTimeInMillis": number,
#         "ServiceProcessingTimeInMillis": number,
#         "TotalExecutionTimeInMillis": number
#       },
#       "Status": {
#         "AthenaError": {
#           "ErrorCategory": number,
#           "ErrorMessage": "string",
#           "ErrorType": number,
#           "Retryable": boolean
#         },
#         "CompletionDateTime": number,
#         "State": "string",
#         "StateChangeReason": "string",
#         "SubmissionDateTime": number
#       },
#       "WorkGroup": "string"
#     }
#   ],
#   "UnprocessedQueryExecutionIds": [
#     {
#       "ErrorCode": "string",
#       "ErrorMessage": "string",
#       "QueryExecutionId": "string"
#     }
#   ]
# }
Chilkat Online Tools

BatchGetQueryExecution Python Example

Athena
