types-aiobotocore-athena

import asyncio import os from aiobotocore.session import get_session from types_aiobotocore_athena.client import AthenaClient from types_aiobotocore_athena.type_defs import StartQueryExecutionOutputTypeDef async def list_athena_workgroups() -> None: session = get_session() # For demonstration, using dummy credentials. Replace with actual config or env vars. aws_access_key_id = os.environ.get('AWS_ACCESS_KEY_ID', 'testing') aws_secret_access_key = os.environ.get('AWS_SECRET_ACCESS_KEY', 'testing') aws_session_token = os.environ.get('AWS_SESSION_TOKEN', 'testing') async with session.create_client( "athena", region_name="us-east-1", aws_access_key_id=aws_access_key_id, aws_secret_access_key=aws_secret_access_key, aws_session_token=aws_session_token ) as client: client: AthenaClient # Explicit type annotation is optional but recommended try: response = await client.list_work_groups() print("Athena Workgroups:") for wg in response.get('WorkGroups', []): print(f" - {wg.get('Name')} (State: {wg.get('State')})") except client.exceptions.ClientError as e: print(f"Error listing workgroups: {e}") async def run_example(): await list_athena_workgroups() if __name__ == "__main__": asyncio.run(run_example())