mypy-boto3-apigatewaymanagementapi

import os import boto3 from mypy_boto3_apigatewaymanagementapi.client import ApiGatewayManagementApiClient # Initialize a boto3 client with type hints # For local testing, ensure AWS_REGION, AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY are set. # ApiGatewayManagementApi typically requires an endpoint_url. client: ApiGatewayManagementApiClient = boto3.client( "apigatewaymanagementapi", region_name=os.environ.get("AWS_REGION", "us-east-1"), endpoint_url=os.environ.get("APIGATEWAY_MGMT_ENDPOINT_URL", "https://example.execute-api.us-east-1.amazonaws.com/prod") ) # Example: Post data to a connection try: connection_id = "some-connection-id" # Replace with an actual connection ID data_to_send = b"Hello from mypy-boto3!" response = client.post_to_connection( ConnectionId=connection_id, Data=data_to_send ) print(f"Successfully posted to connection {connection_id}. Response: {response}") except client.exceptions.GoneException: print(f"Connection {connection_id} is gone.") except Exception as e: print(f"An error occurred: {e}") # Mypy will now validate client method calls and argument types: # client.non_existent_method() # Mypy would flag this as an error