def push_public_key()

in ec2instanceconnectcli/key_publisher.py [0:0]

• 17 lines of code
• 3 McCabe index (conditional complexity)

def push_public_key(session, instance_id, user, pub_key, target_zone):
    """
    Creates a Boto3 client to make call to the EC2 Instance Connect Service and invokes the SendSSHPublicKey API

    :param session: A Botocore session to use to generate the boto client
    :type session: botocore.session.Session
    :param client_config: Contains AWS credentials, region and custom endpoint URL
    :type client_config: dict
    :param instance_id: Instance ID of the EC2 instance
    :type instance_id: basestring
    :param user: EC2 user to publish to on-instance
    :type user: basestring
    :param pub_key: Public key to be pushed
    :type pub_key: basestring
    :param target_zone: availability zone the instance lives in
    :type target_zone: basestring
    """

    try:
        client = session.create_client('ec2-instance-connect')
    except Exception as e:
        print("Error while trying to push the public key:\n" + str(e))
        sys.exit(1)

    params = {
              'InstanceId': instance_id,
              'InstanceOSUser': user,
              'SSHPublicKey': pub_key,
              'AvailabilityZone': target_zone
             }

    try:
        client.send_ssh_public_key(**params)
    except Exception as e:
        print("Error while pushing the public key:\n" + str(e))
        sys.exit(1)