如何导入boto3雅典娜客户例外

Question

我正在使用Boto3与Athena合作，如下所示：

def query_athena(query, output_path):
    client = boto3.client('athena')
    client.start_query_execution(
        ResultConfiguration={'OutputLocation': output_path},
        QueryString=query
    )

如docs, start_query_execution may raise InternalServerException, InvalidRequestException or <代码> toomanyrequestSexception 。我想对待如下：

def query_athena(query, output_path):
    client = boto3.client('athena')
    try:
        client.start_query_execution(
            ResultConfiguration={'OutputLocation': output_path},
            QueryString=query
        )
    except <AthenaException> as e:
        deal with e

其中＆lt; athenaException＆gt; 是我提到的三个例外之一，或者更好的是它们的超类。

我的问题是如何导入这些例外？文档显示它们为 athena.client.exceptions.internalserverexception ，但我似乎在任何boto3模块中都找不到这个athena.client。

Answer 1

我遇到了同样的困惑，但弄清楚了。文档中列出的异常不是boto3的内部，而是在boto3抛出客户端错误时响应中包含。

我在解决方案上的第一张照片看起来像这样。它假设您已经处理了S3输出位置，BOTO3会话等：

import boto3
from botocore.exceptions import ClientError

try:
    client = session.client('athena')
    response = client.start_query_execution(
        QueryString=q,
        QueryExecutionContext={
            'Database': database
        },
        ResultConfiguration={
            'OutputLocation': s3_output,
        }
    )
    filename = response['QueryExecutionId']
    print('Execution ID: ' + response['QueryExecutionId'])

except ClientError as e:
    response = e.response
    code = response['Error']['Code']
    message = response['Error']['Message']
    if code == 'InvalidRequestException':
        print(f'Error in query, {code}:\n{message}')
        raise e
    elif code == 'InternalServerException':
        print(f'AWS {code}:\n{message}')
        raise e
    elif code == 'TooManyRequestsException':
        # Handle a wait, retry, etc here
        pass