dZddlmZddlmZddlmZddlZddlmZddlm Z ddlm Z dd l m Z ddl Z d Z d d Zd Zy)z0Helpers for running commands external to gcloud.)absolute_import)division)unicode_literalsN) ssh_utils) containers)ssh)logc  tj|tjtj}t|jj dD]?}t jj|t jjA|j|jdk7r1t|jjjy#t$ru}t j j#d|dt%j&|dj)|}t j j#d|zYd}~yd}~wwxYw) a Runs a subprocess and prints out the output. Args: proc_name: The name of the subprocess to call. Used for error logging. command_list: A list with all the arguments for a subprocess call. Must be able to be passed to a subprocess.Popen call. )stdoutstderrrzError running z:  zINVOCATION: %sN) subprocessPopenPIPEiterr readliner outwriteflushwait returncodeOSErrorr readstriperrPrintsix text_typejoin) proc_name command_listproclecommand_list_strs Blib/googlecloudsdk/command_lib/compute/diagnose/external_helper.py RunSubprocessr(s 7   Z__Z__ FD $++&& + ggmmA ggmmo, IIK ! DKK$$&,,. // 7GGMMIs}}Q7GHIxx -GGMM"%55667sC&C)) E'2A+E""E'ctj|}tj|} tj||} d} d} |j sE|j j} |jtj||j} tj|| || } g}tjd|}tj| | | | ||}|r4tdj!|j#|j$y|j'|j$||}t(j*j-|S)aRuns a SSH command to a Google Compute Engine VM. Args: command_list: List with the ssh command to run. instance: The GCE VM object. user: The user to be used for the SSH command. args: The args used to call the gcloud instance. ssh_helper: ssh_utils.BaseSSHCLIHelper instance initialized for the command. explicit_output_file: Use this file for piping stdout of the SSH command, instead of using stdout. This is useful for capturing the command and analyzing it. explicit_error_file: Use this file for piping stdout of the SSH command, instead of using stdout. This is useful for capturing the command and analyzing it. dry_run: Whether or not this is a dry-run (only print, not run). Returns: The exit code of the SSH command Raises: ssh.CommandError: there was an error running a SSH command N) identity_fileoptions extra_flagsremote_command remainderrr)explicit_output_fileexplicit_error_file)rGetExternalIPAddressGetInternalIPAddressrRemoteplainkeyskey_file GetConfig HostKeyAliasstrict_host_key_checkingParseAndSubstituteSSHFlagsrGetRemoteCommand SSHCommand DryRunLogr BuildenvRunr rr)r"instanceuserargs ssh_helperr/r0dry_runexternal_addressinternal_addressremoter*r+r,r.r-cmd return_codes r'RunSSHCommandToInstancerK7s+:33H=33H= ::& -&- ' OO,,M""9#9#9(#C#'#@#@BG..tV=M/?A+)..t\B.  !#  #  chhsyy012 nn/-/+''--/ cHtjjd|zy)Nz[COMMAND TO RUN]: %s )r rr)msgs r'r=r=ws''--(3./rL)NNF)__doc__ __future__rrrr"googlecloudsdk.command_lib.computer#googlecloudsdk.command_lib.util.sshrrgooglecloudsdk.corer rr(rKr=rLr'rUs@7&'8:3# 7>2604$)=@0rL