~dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Zdd lm Z dd lmZd Zej&ej(j*Gd d ej,Zej&ej(j0GddeZddde_y)z+Command to SCP to/from a Cloud TPU VM Node.)absolute_import)division)unicode_literals)FilesCompleter)base)flags) ssh_utils)ssh) propertiesc|jddd|jdtddd |jd d d |jddd|jdddy)z>Additional flags and positional args to be passed to *scp(1)*.z --scp-flagappendzR Additional flags to be passed to *scp(1)*. This flag may be repeated. )actionhelpsourceszSpecifies the files to copy.z[[USER@]INSTANCE:]SRC+) completerrmetavarnargs destinationz-Specifies a destination for the source files.z[[USER@]INSTANCE:]DEST)rrz --recurse store_truezUpload directories recursively.z --compresszEnable compression.N) add_argumentr)parsers &lib/surface/compute/tpus/tpu_vm/scp.py AddSCPArgsrs     )%    :&(  ,-NP <.CEc.eZdZdZdZdZedZdZy)Scp.Copy files to and from a Cloud TPU VM via SCP.Fctjj|tj||j |j t|tj|ddy)z\Set up arguments for this command. Args: parser: An argparse.ArgumentParser. tpuscp) resource_typeoperation_typeN) r BaseSSHCLIHelperArgs tpu_ssh_utils AddTPUSSHArgs _ENABLE_IAP_ENABLE_BATCHINGrr AddZoneFlag)clsrs rr%zScp.ArgsCsN##F+9M9MNv fE%Hrc tjj|j}|jDcgc]!}tjj|#}}tj j ||d|jxs|dj}|j}|js|D] }||_ d}|jsd}tjd|_ |j9tjjjjd|_ dg}t!j"|||j%|j&||d||| d} |j&r;|dr6t!j(|j*t-|dj.} t!j0||| ycc}w)NT) single_remoterF)warn_on_account_user)required)r FileReferenceFromPathrr SCPCommandVerifyremotehostuserGetDefaultSshUsernamezoner VALUEScomputeGetr&PrepareNodeForSCP ReleaseTrackr)ParseBatchSize batch_sizelen worker_ipsSCPIntoPreppedNodes) selfargsdstsrcsrcsr5tpu_nameusername_requested prepped_nodescp_batch_sizes rRunzScp.RunOs   $ $T%5%5 6C7;|| D|C   & &s +|D DNN$48 ZZ )47>>F{{H ::#  ;; --4Hfk yy##++0044d4Cdi6L##       N a$33 //3|A99:n%%lD.IM Es&GN) __name__ __module__ __qualname____doc__r(r) classmethodr%rMrrrr;s,6+ I I(JrrceZdZdZdZdZy)ScpAlphaz6Copy files to and from a Cloud TPU VM via SCP (Alpha).TN)rNrOrPrQr(r)rSrrrUrUzs>+rrUra% To copy a file (for example, a text file in the local home directory) to a Cloud TPU VM, run: $ {command} ~/my-file my-tpu: To copy a file into all workers in a Cloud TPU VM, run: $ {command} ~/my-file my-tpu: --worker=all To copy a file from a Cloud TPU VM to the home directory of the local computer, run: $ {command} my-tpu:~/my-file ~/ To copy all files in a folder to a Cloud TPU VM, run: $ {command} ~/my-folder/ my-tpu: --recurse )briefEXAMPLESN)rQ __future__rrrargcomplete.completersrgooglecloudsdk.callioper"googlecloudsdk.command_lib.computerr .googlecloudsdk.command_lib.compute.tpus.tpu_vmr r&#googlecloudsdk.command_lib.util.sshgooglecloudsdk.corer r ReleaseTracksr>GACommandrALPHArU detailed_helprSrrrds2&'2(48O3*E8D%%(();J$,,;J*;J|D%%++,s- 9 r