Chilkat Online Tools

StartQueryExecution Python Example

Athena

import sys
import chilkat

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

rest = chilkat.CkRest()

authAws = chilkat.CkAuthAws()
authAws.put_AccessKey("AWS_ACCESS_KEY")
authAws.put_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.put_Region("us-west-2")
authAws.put_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.get_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 = chilkat.CkJsonObject()
json.UpdateString("ClientRequestToken","string")
json.UpdateString("QueryExecutionContext.Catalog","string")
json.UpdateString("QueryExecutionContext.Database","string")
json.UpdateString("QueryString","string")
json.UpdateString("ResultConfiguration.AclConfiguration.S3AclOption","string")
json.UpdateString("ResultConfiguration.EncryptionConfiguration.EncryptionOption","string")
json.UpdateString("ResultConfiguration.EncryptionConfiguration.KmsKey","string")
json.UpdateString("ResultConfiguration.ExpectedBucketOwner","string")
json.UpdateString("ResultConfiguration.OutputLocation","string")
json.UpdateString("WorkGroup","string")

# The JSON request body created by the above code:

# {
#   "ClientRequestToken": "string",
#   "QueryExecutionContext": {
#     "Catalog": "string",
#     "Database": "string"
#   },
#   "QueryString": "string",
#   "ResultConfiguration": {
#     "AclConfiguration": {
#       "S3AclOption": "string"
#     },
#     "EncryptionConfiguration": {
#       "EncryptionOption": "string",
#       "KmsKey": "string"
#     },
#     "ExpectedBucketOwner": "string",
#     "OutputLocation": "string"
#   },
#   "WorkGroup": "string"
# }

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

sbRequestBody = chilkat.CkStringBuilder()
json.EmitSb(sbRequestBody)
sbResponseBody = chilkat.CkStringBuilder()
success = rest.FullRequestSb("POST","/",sbRequestBody,sbResponseBody)
if (success != True):
    print(rest.lastErrorText())
    sys.exit()

respStatusCode = rest.get_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 = chilkat.CkJsonObject()
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

QueryExecutionId = jResp.stringOf("QueryExecutionId")

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

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