SrSSKJr SSKJr SSKJr SSKrSSKrSSKJr SSK J r SSK J r SS K J r SS K Jr SS KJr SS KJr SS KJr SS KJr SS KJr SSKJr Sr\ R6"\ R8R:\ R8R<5"SS\ R>55r SSS.\ l!g)z=Command to SCP to/from a Cloud TPU Queued Resource's Node(s).)absolute_import)division)unicode_literalsN)FilesCompleter)base) exceptions)flags) ssh_utils)ssh)util)log) propertiescURSSSS9 URS[SSSS 9 URS S S S 9 URSSSS9 URSSSS9 URSSSS9 g)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.--node0a- TPU node(s) to connect to. The supported value is a single 0-based index of the node(s) in the case of a TPU Pod. It additionally supports a comma-separated list (e.g. '1,4,6'), range (e.g. '1-3'), or special keyword ``all" to run the command concurrently on each of the specified node(s). Note that when targeting multiple nodes, you should run 'ssh-add' with your private key prior to executing the gcloud command. Default: 'ssh-add ~/.ssh/google_compute_engine'. )defaultrN) add_argumentr)parsers 0lib/surface/compute/tpus/queued_resources/scp.py AddSCPArgsr %s    )%    :&  ,-N <.C  c:\rSrSrSrSrSrSr\S5r Sr Sr g) ScpV;Copy files to and from a Cloud TPU Queued Resource via SCP.T@c[RRU5 [R"UUR UR URS9 [U5 [R"USSS9 g)zPSet up arguments for this command. Args: parser: An argparse.ArgumentParser. )enable_batching_defaulttpuscp) resource_typeoperation_typeN) r BaseSSHCLIHelperArgs tpu_ssh_utils AddTPUSSHArgs _ENABLE_IAP_ENABLE_BATCHINGDEFAULT_BATCH_SIZEr r AddZoneFlag)clsrs rr.Scp.Args^s^##F+   # 6 6  v fE%Hr!c[R"5n[RRUR5nUR Vs/sH"n[RRU5PM$ nn[R RXSSS9 UR=(d USRnURnUR(dUH nXdlM SnUR(dSn[R"SS9Ul URc6[RRRR!SS9Ul ["R$"UR'55n U R!XqR5n [(R*"UR,U R.R05n [3U 5S:a+USR(a[4R6"SS 5e/n Sn [3U 5n[8R:"UR<[3U 55nS/U-n[?U 5HunnU RA[BRD"[8RFURHUUR'5URJUUUUUU4 S 95 U S RM5 U S- n X:XdM[(RN"U U5 Sn /n M U S:a[(RN"U U5 UVs/sH nUcMUPM nn[3U5U:a%[PRR"S RUU55 [8R:"UR<URV5n[8RX"UUU5 [PRZR]S [R"5U- -5 gs snfs snf)NT) single_remoterF)warn_on_account_user)requiredrz;cannot target multiple nodes while copying files to client.)targetargsz@Could not prepare all {} nodes, attempting to ssh into the rest.z!Completed execution in %s seconds)/timer FileReferenceFromPathrr SCPCommandVerifyremotehostuserGetDefaultSshUsernamezonerVALUEScomputeGetqueued_resource_utilsTPUQueuedResource ReleaseTrack qr_ssh_utils ParseNodeFlagnoder)nodeSpeclenrInvalidArgumentExceptionr/ParseBatchSize batch_size enumerater threadingThreadPrepareNodeForSCPnodeIdr2startWaitForNodeBatchCompletionr warningformatr3SCPIntoPreppedNodesstatusPrint)selfr= start_timedstsrcsrcsrDqr_nameusername_requestedqueued_resource_clientqueued_resource node_specsprep_nodes_threadscurrent_batch_size num_nodesprep_node_batch_size prepped_nodesindexrQ prepped_nodescp_batch_sizes rRunScp.RunosQJ   $ $T%5%5 6C7;|| D|C   & &s +|D DNN$48 ZZ )47>>FkkG ::#  ;; --4Hfk yy##++0044d4Cdi2DD -00))DO++ ?&&//J :tAw~~  / /  G  JI(77 ZFY&M , t   "44++##%''$  """$A  3//   3-6A--.@-P *)L  )  =I% kk 6)$ #11 00N%%  JJ+tyy{Z/GH} ETs )OOON) __name__ __module__ __qualname____firstlineno____doc__r1r2r3 classmethodr.ru__static_attributes__rwr!rr#r#Vs0C+II cr!r#r%a To copy a file (for example, a text file in the local home directory) to a Cloud Queued Resource, run: $ {command} ~/my-file my-qr: To copy a file into all nodes and workers in a Cloud TPU Queued Resource (with the default batch size), run: $ {command} ~/my-file my-qr: --worker=all --node=all To copy a file into all nodes and workers in a Cloud TPU Queued Resource with a batch size of 4, run: $ {command} ~/my-file my-qr: --worker=all --node=all --batch-size=4 To copy a file into all workers in the first node of a Cloud TPU Queued Resource, run: $ {command} ~/my-file my-qr: --worker=all To copy a file from a Cloud TPU Queued Resource to the home directory of the local computer, run: $ {command} my-qr:~/my-file ~/ To copy all files in a folder to a Cloud TPU Queued Resource, run: $ {command} ~/my-folder/ my-qr: --recurse )briefEXAMPLES)"r| __future__rrrrXr?argcomplete.completersrgooglecloudsdk.callioperr"googlecloudsdk.command_lib.computer r 8googlecloudsdk.command_lib.compute.tpus.queued_resourcesr rOr rL.googlecloudsdk.command_lib.compute.tpus.tpu_vmr/#googlecloudsdk.command_lib.util.sshgooglecloudsdk.corer rr ReleaseTracksrNALPHAGACommandr# detailed_helprwr!rrsD&' 1(.48XbO3#*.bD%%++T->->-A-AB{$,,{C{~K r!