Skip to content

Instantly share code, notes, and snippets.

@Havoc24k

Havoc24k/create_ec2_ssh_config.py

Last active September 1, 2021 12:27
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save Havoc24k/ec4b43cad0cf919d5f3c7796bb94a3b1 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save Havoc24k/ec4b43cad0cf919d5f3c7796bb94a3b1 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. Thanasis Politis revised this gist Feb 4, 2018. 1 changed file with 1 addition and 1 deletion.
    2 changes: 1 addition & 1 deletion create_ec2_ssh_config.py
    View file
    Original file line number Diff line number Diff line change
    @@ -53,7 +53,7 @@ def main():
    host <client>.<environment>.<name>
    Hostname <ec2-public-ip>
    IdentityFile path_to_ssh_key
    IdentityFile <path_to_ssh_key>
    User <instance_username>
    port <ssh_port>
  2. Thanasis Politis revised this gist Feb 4, 2018. 1 changed file with 1 addition and 1 deletion.
    2 changes: 1 addition & 1 deletion create_ec2_ssh_config.py
    View file
    Original file line number Diff line number Diff line change
    @@ -51,7 +51,7 @@ def main():
    Using those tags we create an ssh config entry for each instance.
    and append it to the config file.
    host <client>.<environment>.name
    host <client>.<environment>.<name>
    Hostname <ec2-public-ip>
    IdentityFile path_to_ssh_key
    User <instance_username>
  3. Thanasis Politis revised this gist Feb 4, 2018. 1 changed file with 0 additions and 1 deletion.
    1 change: 0 additions & 1 deletion create_ec2_ssh_config.py
    View file
    Original file line number Diff line number Diff line change
    @@ -61,7 +61,6 @@ def main():
    for page in response_iterator:
    for reservation in page['Reservations']:
    for instance in reservation['Instances']:
    print(instance)
    try:
    host_line = ""
    host = ""
  4. Thanasis Politis created this gist Feb 4, 2018.
    100 changes: 100 additions & 0 deletions create_ec2_ssh_config.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,100 @@
    #!/usr/bin/python

    """AWS EC2 SSH config Generator."""

    import boto3
    import os

    # The location and name of our generated config file
    path_to_config = '/.ssh/aws_demo.config'

    # The SSH key we use to connet to those instances
    path_to_ssh_key = "~/.ssh/aws_demo.pem"

    # The SSH username to use
    instance_username = 'ec2-user'

    # The SSH port to connect to
    ssh_port = 22


    def main():
    """Main."""
    try:
    """
    Using the security credentialsa and the location we set
    when we run `$ awscli configure` we connect to AWS
    and get the list of instances on the specific location
    """
    aws_client = boto3.client('ec2')
    paginator = aws_client.get_paginator('describe_instances')
    response_iterator = paginator.paginate(
    DryRun=False,
    PaginationConfig={
    'MaxItems': 100,
    'PageSize': 10
    }
    )

    """
    Open the config file we specified to be written
    """
    ssh_config_file = open(os.path.expanduser(
    '~') + path_to_config, 'w')

    ssh_config_file.write("##########################\n")
    ssh_config_file.write("##### AWS SSH CONFIG #####\n")
    ssh_config_file.write("##########################\n\n")

    """
    We iterate the results and read the tags for each instance.
    Using those tags we create an ssh config entry for each instance.
    and append it to the config file.
    host <client>.<environment>.name
    Hostname <ec2-public-ip>
    IdentityFile path_to_ssh_key
    User <instance_username>
    port <ssh_port>
    """
    for page in response_iterator:
    for reservation in page['Reservations']:
    for instance in reservation['Instances']:
    print(instance)
    try:
    host_line = ""
    host = ""
    env = ""
    if 'PublicIpAddress' in instance:
    public_ip = instance['PublicIpAddress']
    for tag in instance['Tags']:
    if tag['Key'] == "Client":
    client = tag['Value']
    if tag['Key'] == "Name":
    name = tag['Value']
    if tag['Key'] == "Environment":
    env = tag['Value']

    host = "{}.{}.{}".format(
    client, env, name).replace(" ", "-")
    host_line += "##########################\n"
    host_line += "host {}\n".format(host.lower())
    host_line += " Hostname {}\n".format(public_ip)
    host_line += " IdentityFile {}\n".format(
    path_to_ssh_key)
    host_line += " User {}\n".format(
    instance_username)
    host_line += " port {}\n".format(ssh_port)
    host_line += "##########################\n"
    host_line += "\n"
    ssh_config_file.write(host_line)
    except Exception as e:
    raise e

    print("File updated: " + os.path.expanduser('~') + path_to_config)
    except Exception as e:
    print(e)

    if __name__ == '__main__':
    main()