Chilkat Online Tools

GetQueryExecution 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("QueryExecutionId","string")

# The JSON request body created by the above code:

# {
#   "QueryExecutionId": "string"
# }

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

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

EffectiveEngineVersion = jResp.StringOf("QueryExecution.EngineVersion.EffectiveEngineVersion")
SelectedEngineVersion = jResp.StringOf("QueryExecution.EngineVersion.SelectedEngineVersion")
Query = jResp.StringOf("QueryExecution.Query")
Catalog = jResp.StringOf("QueryExecution.QueryExecutionContext.Catalog")
Database = jResp.StringOf("QueryExecution.QueryExecutionContext.Database")
QueryExecutionId = jResp.StringOf("QueryExecution.QueryExecutionId")
S3AclOption = jResp.StringOf("QueryExecution.ResultConfiguration.AclConfiguration.S3AclOption")
EncryptionOption = jResp.StringOf("QueryExecution.ResultConfiguration.EncryptionConfiguration.EncryptionOption")
KmsKey = jResp.StringOf("QueryExecution.ResultConfiguration.EncryptionConfiguration.KmsKey")
ExpectedBucketOwner = jResp.StringOf("QueryExecution.ResultConfiguration.ExpectedBucketOwner")
OutputLocation = jResp.StringOf("QueryExecution.ResultConfiguration.OutputLocation")
StatementType = jResp.StringOf("QueryExecution.StatementType")
DataManifestLocation = jResp.StringOf("QueryExecution.Statistics.DataManifestLocation")
DataScannedInBytes = jResp.IntOf("QueryExecution.Statistics.DataScannedInBytes")
EngineExecutionTimeInMillis = jResp.IntOf("QueryExecution.Statistics.EngineExecutionTimeInMillis")
QueryPlanningTimeInMillis = jResp.IntOf("QueryExecution.Statistics.QueryPlanningTimeInMillis")
QueryQueueTimeInMillis = jResp.IntOf("QueryExecution.Statistics.QueryQueueTimeInMillis")
ServiceProcessingTimeInMillis = jResp.IntOf("QueryExecution.Statistics.ServiceProcessingTimeInMillis")
TotalExecutionTimeInMillis = jResp.IntOf("QueryExecution.Statistics.TotalExecutionTimeInMillis")
ErrorCategory = jResp.IntOf("QueryExecution.Status.AthenaError.ErrorCategory")
ErrorMessage = jResp.StringOf("QueryExecution.Status.AthenaError.ErrorMessage")
ErrorType = jResp.IntOf("QueryExecution.Status.AthenaError.ErrorType")
Retryable = jResp.IntOf("QueryExecution.Status.AthenaError.Retryable")
CompletionDateTime = jResp.IntOf("QueryExecution.Status.CompletionDateTime")
State = jResp.StringOf("QueryExecution.Status.State")
StateChangeReason = jResp.StringOf("QueryExecution.Status.StateChangeReason")
SubmissionDateTime = jResp.IntOf("QueryExecution.Status.SubmissionDateTime")
WorkGroup = jResp.StringOf("QueryExecution.WorkGroup")

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

# {
#   "QueryExecution": {
#     "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"
#   }
# }