Python 使用 Athena 的 boto3 参数化查询

Question

I cant seem to find the document on how to pass execution parameters to Athena using boto3. I did find how to do it using aws cli, like so:

aws athena start-query-execution 
--query-string "SELECT * FROM table WHERE x = ? AND y = ?"
--query-execution-context "Database"="default" 
--result-configuration "OutputLocation"="s3://..."
--execution-parameters "1" "2"

Is there a way to do the same using boto3 with something like:

import boto3

client = boto3.client(
            "athena",
            aws_access_key_id=XXX,
            aws_secret_access_key=YYY,
            region_name=ZZZ,
)
response = client.start_query_execution(
            QueryString="SELECT * FROM table WHERE x = ? AND y = ?",
            QueryExecutionContext={"Database": "default"},
            ResultConfiguration={"OutputLocation": "s3://..."},
            WorkGroup=self._kwargs.get('workgroup'),
)

Is it possible to do it with boto3 without using prepared statements?

Answer 1

You can also use awswrangler to do it very simply:

import awswrangler as wr
df = wr.athena.read_sql_query(
    sql="SELECT * FROM table WHERE x=:x; AND y=:y;",
    params={"x": "'x_value'", "y": "'y_value'"}
)

To read more how the read_sql_query function works and list of params: https://aws-sdk-pandas.readthedocs.io/en/stable/stubs/awswrangler.athena.read_sql_query.html