dZddlmZddlmZddlmZddlZddlmZddlmZddl m Z dd l m Z dd l mZdd lmZd d dZdddZdZej(ej*j,Gddej.Zee_y)zCommand for attaching a persistent disk to a TPU VM. Allows TPU VM users to attach persistent disks to TPUs in a form that is decoupled from the Create and Delete lifecycle of the actual TPU VM. )absolute_import)division)unicode_literalsN)base) exceptions) resource_args)util)concept_parsers) propertieszWRead-write. It is an error to attach a disk in read-write mode to more than one TPU VM.z Read-only.) read-write read-onlyaP *{command}* is used to attach a disk to a TPU VM. For example, $ gcloud compute tpus tpu-vm attach-disk example-tpu --disk=example-disk --mode=read-only --zone=us-central1-a attaches the disk named 'example-disk' to a TPU VM named 'example-tpu' in read-only mode in zone ``us-central1-a''. a! To attach a disk named 'my-disk' for read-only access to a TPU named 'my-tpu', run: $ {command} my-tpu --disk=my-disk --mode=read-only To attach a disk named 'my-read-write-disk' for read-write access to a TPU named 'my-tpu', run: $ {command} my-tpu --disk=my-read-write-disk To attach a regional disk named for read-write access to a TPU named 'my-tpu', run: $ {command} my-tpu --disk=projects/tpu-test-env-one-vm/region/us-central1/disks/example-disk ) DESCRIPTIONEXAMPLESctjjdtjddj |dj |y)zAdds a TPU Name resource argument. NOTE: May be used only if it's the only resource arg in the command. Args: parser: the argparse parser for the command. verb: str, the verb to describe the resource, such as 'to attach'. tpuTPUzThe TPU function name {}.TrequiredN)r ConceptParser ForResourcerGetTPUResourceSpecformat AddToParser)parserverbs .lib/surface/compute/tpus/tpu_vm/attach_disk.pyAddTPUResourceArgrIsK++ &&u-!((. ,![(c&eZdZdZedZdZy) AttachDiskzAttach disk to TPU VM.cv|jddd|jdtddt|d y ) z\Set up arguments for this command. Args: parser: An argparse.ArgumentParser. z--diskz-The name of the disk to attach to the TPU VM.T)helprz--moder zSpecifies the mode of the disk.)choicesdefaultr"zto attach diskN) add_argument MODE_OPTIONSr)rs rArgszAttachDisk.Args^sN  <   . 0 f./rc |j9tjjjj d|_|j j j}tj|j}|j |j|j}tj|stjdd|jdk(r0|j j"j$j&|_nU|jdk(r0|j j"j$j(|_ntjddt+j,d |j.}|r |j.}nYtjj0j2j d}d |zd z|jzd z|j.z}|j4sC|j j#|j| }|j4j7|ng} |j4D]} | j7| j8|| vrC|j j#|j| }|j4j7|ntjdd|j;|j|j|ddS)NTrrzthis command is only available for Cloud TPU VM nodes. To access this node, please see https://cloud.google.com/tpu/docs/creating-deleting-tpus.r r modez6can only attach disks in read-write or read-only mode.z'projects/.+/(zones|regions)/.+/disks/.+z projects/z/zones/z/disks/)r) sourceDiskz'disk is already attached to the TPU VM. data_diskszAttaching disk to TPU VM)namezonenode update_maskpoller_message)r-r VALUEScomputeGetCONCEPTSrParse tpu_utilsTPUNode ReleaseTrackName IsTPUVMNoderBadArgumentExceptionr)messages AttachedDiskModeValueValuesEnum READ_WRITE READ_ONLYrematchdiskcoreproject dataDisksappendr* UpdateNode) selfargs tpu_name_refrr.source_path_match source_pathrEdisk_to_attachsource_disk_listrCs rRunzAttachDisk.Runss yy##++0044d4Cdi==$$**,L   D--/ 0C 77<$$& 2D   &  + +  F GG  yyL ,,++??JJdi k !,,++??IIdi  + + J !K!%,IIk!!&&..22D2Ag  ) +dii 7) Cdii O >>||00yy 1n nnN+..$0! , ,22"3  n--- <> > >>    YY  1  rN)__name__ __module__ __qualname____doc__ staticmethodr'rPrrr r Zs00(>rr )rT __future__rrrrAgooglecloudsdk.callioperr.googlecloudsdk.command_lib.compute.tpus.tpu_vmrr r6(googlecloudsdk.command_lib.util.conceptsr googlecloudsdk.corer r& DETAILED_HELPr ReleaseTracksr8ALPHACommandr detailed_helprVrrras'' (.HLD* !      >)"D%%++,VV-Vp) r