boto-session-manager

from boto_session_manager import BotoSesManager # Initialize the session manager (uses default AWS credentials chain) bsm = BotoSesManager() # Get an S3 client. Clients are cached for reuse. s3_client = bsm.get_client("s3") # Alternatively, use attribute access for common services for auto-complete # s3_client = bsm.s3_client # Example: List S3 buckets try: response = s3_client.list_buckets() print("S3 Buckets:") for bucket in response.get("Buckets", []): print(f"- {bucket['Name']}") except Exception as e: print(f"Error listing buckets: {e}") # Example: Assuming an IAM role # This assumes 'MY_ASSUME_ROLE_ARN' environment variable is set # or replaced with a valid ARN. assume_role_arn = os.environ.get('MY_ASSUME_ROLE_ARN', 'arn:aws:iam::123456789012:role/MyTestRole') if '123456789012' not in assume_role_arn: # Simple check to avoid running with dummy ARN try: print(f"\nAttempting to assume role: {assume_role_arn}") assumed_bsm = bsm.assume_role( RoleArn=assume_role_arn, RoleSessionName="my-assumed-session" ) ec2_client_assumed = assumed_bsm.get_client("ec2") print(f"Successfully obtained EC2 client with assumed role: {ec2_client_assumed}") except Exception as e: print(f"Error assuming role: {e}")