ssh-ssm.py

#!/usr/bin/python
import time
import json
import boto3
#from dateutil import tz


def parse_command_id(send_command_output):
    return send_command_output['Command']['CommandId']

def fwd(instance):
    # https://aws.amazon.com/blogs/aws/new-port-forwarding-using-aws-system-manager-sessions-manager/
    #INSTANCE_ID=$(aws ec2 describe-instances   --filter "Name=tag:Name,Values=CodeStack/NewsBlogInstance"          --query "Reservations[].Instances[?State.Name == 'running'].InstanceId[]"              --output text)
# create the port forwarding tunnel
    prms = {
        "portNumber":["22"],
        "localPortNumber":["2222"]
    }
    prms_jsn = json.dumps(prms)
    print(f"""aws ssm start-session --target {instance} --document-name AWS-StartPortForwardingSession  --parameters '{prms_jsn}'""")

def main():
    ec2_client = boto3.client('ec2')
    ssm_client = boto3.client('ssm')

    # Get the list of instance IDs and their states
    instances_response = ec2_client.describe_instances()
    
    instances = [
        (instance['InstanceId'], instance['State']['Name'])
        for reservation in instances_response['Reservations']
        for instance in reservation['Instances']
    ]
    for reservation in instances_response['Reservations']:
        for instance in reservation['Instances']:
            print(instance)
            instance_id = instance['InstanceId']
            state = instance['State']['Name']            
            if state == 'running':
                #print(f"Starting command for instance: {instance_id}")    
                #print(f"aws ssm start-session --target  {instance_id}")
                fwd(instance_id)
        
if __name__ == "__main__":
    main()