AxjrdZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZejd Gd d ZdS) zH Runpod | CLI | Utils | SSH Command Connect and run commands over SSH. N) STOP_EVENT)get_pod_ssh_ip_port)get_ignore_list)find_ssh_key_fileT) autoresetcVeZdZdZdZdZdZdZdZdZ dZ d Z d Z dd Z d ZdS) SSHConnectionz"Connect and run commands over SSH.cb||_ t|\|_|_t |j|j|_t j|_|j t j |j |j|jd|jnV#t j $rD}ttjjd|dz|t#jdYd}~nd}~wwxYwt'jt&j|jdS)Nroot)portusername key_filename[]r)pod_idrpod_ippod_portrkey_fileparamiko SSHClientsshset_missing_host_key_policy AutoAddPolicyconnect SSHExceptionprintcoloramaForeREDsysexitsignalSIGINT_signal_handler)selfrerrs IC:\Projects\Historical\venv\Lib\site-packages\runpod/cli/utils/ssh_cmd.py__init__zSSHConnection.__init__s  ).handle_streamOsI ? ??%-f---/>>> ? ?r/z && zsource /root/.bashrczsource /etc/rp_environmentzHwhile IFS= read -r -d '' line; do export "$line"; done < /proc/1/environT)targetargsdaemonN) joinr exec_command threadingThreadrrGREENrr start) r&commandsrEcommand full_command_stdoutstderr stdout_thread stderr_threads r( run_commandszSSHConnection.run_commandsLs ? ? ?   ! !G!;;*0` L!% 5 5l C C Avv%,$hm14;?M &,$hm/=M    ! ! !    ! ! !       5 ! !r/c|j5}|||ddddS#1swxYwYdS)z+Copy local file to remote machine over SSH.N)r open_sftpput)r& local_path remote_pathsftps r(put_filezSSHConnection.put_fileps X   ! ! .T HHZ - - - . . . . . . . . . . . . . . . . . .>AAc|j5}|||ddddS#1swxYwYdS)z.Fetch a remote file to local machine over SSH.N)rrYget)r&r\r[r]s r(get_filezSSHConnection.get_fileus X   ! ! .T HH[* - - - . . . . . . . . . . . . . . . . . .r_czdg|zd|jgz}tj|ddS)z(Launch an interactive terminal over SSH.rroot@TcheckN)r8r subprocessrun)r&cmds r(launch_terminalzSSHConnection.launch_terminalzsHg--///3H4;3H3H2IIs$''''''r/Fc Lgd}tD]}|d|g|r|d|ddd||d|jd|gt j|d S) a`Sync a local directory to a remote directory over SSH. A .runpodignore file can be used to ignore files and directories. This file should be placed in the root of the local directory to sync. Args: local_path (str): The local directory to sync. remote_path (str): The remote directory to sync. )rsyncz-avzz --no-ownerz --no-groupz --excludez--quietz-ezssh  rd:Tre)rextendappendrIr8rrgrh)r&r[r\quiet rsync_cmdpatterns r(rlzSSHConnection.rsyncsBAA &(( 5 5G   k73 4 4 4 4  (   Y ' ' ':sxx 5 5 7 788::3 33k33     ~it4444r/c8|jdS)zClose the SSH connection.N)rr1r-s r(r1zSSHConnection.closes r/N)F)__name__ __module__ __qualname____doc__r)r.r5r8r%rWr^rbrjrlr1r,r/r(r r s,,;;;*    "!"!"!H... ... ((( 5555:r/r )rxr#rgr!rKrr runpod.clirrp_inforrp_runpodignorer rp_userspacerinitr r,r/r(r~s   !!!!!!((((((,,,,,,++++++  GGGGGGGGGGr/