File: //lib/python3/dist-packages/awscli/customizations/emr/__pycache__/ssh.cpython-310.pyc
o
.&�a3 � @ 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 ) �Socks�sockszFCreate a socks tunnel on port 8157 from your machine to the master.
%s�
cluster-idT�*Cluster Id of cluster you want to ssh into��name�required� help_text�
key-pair-file�!Private key file to use for loginc C s� zTt j| j||jd�}|j}t �|� tjdd�}t� d�s#t� d�r6dddddd d
d |jt
j d | g
}nd
dd |jt
j d | ddd
g}t d�
|�� t�|�}|W S tyb t d� Y dS w )N��session�parsed_globals�
cluster_idF��delete�ssh�ssh.exe�-o�StrictHostKeyChecking=no�ServerAliveInterval=10z-ND�8157�-i�@�putty�-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� r6 �?/usr/lib/python3/dist-packages/awscli/customizations/emr/ssh.py�_run_main_command' s: �
���
�
�zSocks._run_main_commandN��__name__�
__module__� __qualname__�NAME�KEY_PAIR_FILE_HELP_TEXT� DESCRIPTION� ARG_TABLEr8 r6 r6 r6 r7 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.
%sr Tr r
r r r4 z!Command to execute on Master Node�r r
c C s� t j| j||jd�}|j}t �|� tjdd�}t� d�s"t� d�r=dddddd |jt
j d
| d g }|j r<|�
|j � n'd d
d |jt
j d
| d g}|j rd|�|j � |�d� |�
d� |�
|j� |�� td�|�� t�|�}t�|j� |S )Nr Fr r r r r r r r z-tr r z3
read -n1 -r -p "Command completed. Press any key."z-mr )r r! r" r r# r$ r% r& r r' r r( r4 �append�writer �closer) r* r+ r, �os�remover. r6 r6 r7 r8 O sD �
��� �
�
zSSH._run_main_commandNr9 r6 r6 r6 r7 rA C s ���� rA c @ �J e Zd ZdZde Zdddd�dddd�d dd
d�d d d
�gZdd� ZdS )�Put�putz!Put file onto the master node.
%sr Tz/Cluster Id of cluster you want to put file ontor
r r �srcz!Source file path on local machine�destz$Destination file path on remote hostrB c C s� t j| j||jd�}|j}t �|� t�d�s t�d�r-dddd|j|jt j
d | g}ndd dd|j|jt j
d | g}|j rL|d
d |j |d
< n|d
d |j� d �d
|d
< t
d
�|�� t�|�}|S )Nr �scp�scp.exe�-r�-o StrictHostKeyChecking=nor r �pscp�-scp����:�/r )r r! r" r r# �validate_scp_with_key_filer r' rK r r( rL �splitr) r* r+ r, �r/ r0 r r1 r2 r4 r5 r6 r6 r7 r8 } s* �
� �
zPut._run_main_commandNr9 r6 r6 r6 r7 rI o s" �����
rI c @ rH )�Get�getz Get file from master node.
%sr Tz/Cluster Id of cluster you want to get file fromr
r r rK z Source file path on remote hostrL z%Destination file path on your machinerB c C s� t j| j||jd�}|j}t �|� t�d�s t�d�r0dddd|jtj d | d |j
g}nd d
dd|jtj d | d |j
g}|j rM|� |j � n |� |j
�
d �d � td
�|�� t�|�}|S )Nr rM rN rO rP r r rT rQ rR rU rS r )r r! r" r r# rV r r' r r( rK rL rC rW r) r* r+ r, rX r6 r6 r7 r8 � s8 �
���� ��
zGet._run_main_commandNr9 r6 r6 r6 r7 rY � s ����
rY )rF r+ r% �awscli.customizations.emrr r r �!awscli.customizations.emr.commandr r> r rA rI rY r6 r6 r6 r7 �<module> s
',(