File: //lib/python3.6/site-packages/awscli/customizations/emr/__pycache__/ssh.cpython-36.pyc
3
�T_3 � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ dZ G dd� de�Z
G d d
� d
e�Z G d d � d e�Z G d
d� de�Z
dS )� N)� constants)�emrutils)�sshutils)�Commandz�
A value for the variable Key Pair File can be set in the AWS CLI config file using the "aws configure set emr.key_pair_file <value>" command.
c @ s8 e Zd ZdZde Zdddd�dddd�gZd d
� Zd S ) �SocksZsockszFCreate a socks tunnel on port 8157 from your machine to the master.
%sz
cluster-idTz*Cluster Id of cluster you want to ssh into)�name�required� help_textz
key-pair-filez!Private key file to use for loginc C s� y�t j| j||jd�}|j}t j|� tjdd�}tj d�sFtj d�rldddddd d
d |jt
j d | g
}n d
dd |jt
j d | ddd
g}t dj
|�� tj|�}|S tk
r� t d� dS X d S )N)�session�parsed_globals�
cluster_idF)�delete�sshzssh.exez-ozStrictHostKeyChecking=nozServerAliveInterval=10z-NDZ8157z-i�@�puttyz-sshz-Nz-D� zDisabling Socks Tunnel.r )r � validate_and_find_master_dns�_sessionr �
key_pair_file�validate_ssh_with_key_file�tempfile�NamedTemporaryFiler �whichr �SSH_USER�print�join�
subprocess�call�KeyboardInterrupt)�self� parsed_argsr �
master_dns�key_file�f�command�rc� r& �/usr/lib/python3.6/ssh.py�_run_main_command' s,
zSocks._run_main_commandN)�__name__�
__module__� __qualname__�NAME�KEY_PAIR_FILE_HELP_TEXT� DESCRIPTION� ARG_TABLEr( r& r& r& r' r s
r c @ s@ e Zd ZdZde Zdddd�dddd�d d
d �gZd d
� ZdS )�SSHr z'SSH into master node of the cluster.
%sz
cluster-idTz*Cluster Id of cluster you want to ssh into)r r r z
key-pair-filez!Private key file to use for loginr$ z!Command to execute on Master Node)r r c C s� t j| j||jd�}|j}t j|� tjdd�}tj d�sDtj d�rzdddddd |jt
j d
| d g }|j r�|j
|j � nNd d
d |jt
j d
| d g}|j r�|j|j � |jd� |j
d� |j
|j� |j� tdj|�� tj|�}tj|j� |S )N)r
r r F)r
r zssh.exez-ozStrictHostKeyChecking=nozServerAliveInterval=10z-ir z-tr z-sshz3
read -n1 -r -p "Command completed. Press any key."z-mr )r r r r r r r r r r r r r$ �append�writer �closer r r r �os�remove)r r r r! r" r# r$ r% r& r& r' r( O s4
zSSH._run_main_commandN)r) r* r+ r, r- r. r/ r( r& r& r& r' r0 C s r0 c @ sJ e Zd ZdZde Zdddd�dddd�d dd
d�d d d
�gZdd� ZdS )�PutZputz!Put file onto the master node.
%sz
cluster-idTz/Cluster Id of cluster you want to put file onto)r r r z
key-pair-filez!Private key file to use for login�srcz!Source file path on local machine�destz$Destination file path on remote host)r r c C s� t j| j||jd�}|j}t j|� tjd�s8tjd�rZdddd|j|jt j
d | g}n dd dd|j|jt j
d | g}|j r�|d d |j |d< n |d d |jj d �d |d< t
d
j|�� tj|�}|S )N)r
r r �scpzscp.exez-rz-o StrictHostKeyChecking=noz-ir �pscpz-scp� �:�/r ���r> r> r> r> )r r r r r �validate_scp_with_key_filer r r7 r r r8 �splitr r r r )r r r r! r" r$ r% r& r& r' r( } s$
zPut._run_main_commandN)r) r* r+ r, r- r. r/ r( r& r& r& r' r6 o s r6 c @ sJ e Zd ZdZde Zdddd�dddd�d dd
d�d d d
�gZdd� ZdS )�Get�getz Get file from master node.
%sz
cluster-idTz/Cluster Id of cluster you want to get file from)r r r z
key-pair-filez!Private key file to use for loginr7 z Source file path on remote hostr8 z%Destination file path on your machine)r r c C s� t j| j||jd�}|j}t j|� tjd�s8tjd�r`dddd|jtj d | d |j
g}n&d d
dd|jtj d | d |j
g}|j r�|j |j � n|j |j
j
d �d � td
j|�� tj|�}|S )N)r
r r r9 zscp.exez-rz-o StrictHostKeyChecking=noz-ir r< r: z-scpr= r; r r> )r r r r r r? r r r r r7 r8 r1 r@ r r r r )r r r r! r" r$ r% r& r& r' r( � s&
zGet._run_main_commandN)r) r* r+ r, r- r. r/ r( r& r& r& r' rA � s rA )r4 r r Zawscli.customizations.emrr r r Z!awscli.customizations.emr.commandr r- r r0 r6 rA r& r& r& r' �<module> s ',(