jSrSSKJr SSKJr SSKJr SSKrSSKrSSKJr SSKJ r SSK J r SS K J r SS K Jr SS K Jr SS KJr SS KJr SS KJr SSKJr SSKJr SSKJr SSKr"SS\R25r"SS\R25r"SS\R25r"SS\R25r"SS\R25rSrSr S7Sjr!Sr"S7Sjr#S r$S7S!jr%S7S"jr&S7S#jr'S$r(S%r)"S&S'\*5r+S(r,S)r-S*r.S+r/S7S,jr0S8S-jr1S.r2S/r3S0r4"S1S2\*5r5"S3S4\*5r6"S5S6\65r7g)9z(CLI Utilities for Cloud TPU VM commands.)absolute_import)division)unicode_literalsN) base_classes)metadata_utils)apis)waiter)base) exceptions) labels_util)map_util)log) properties) resourcesc\rSrSrSrSrg)NoFieldsSpecifiedError(z7Error if no fields are specified for an update request.N__name__ __module__ __qualname____firstlineno____doc____static_attributes__r:lib/googlecloudsdk/command_lib/compute/tpus/tpu_vm/util.pyrr(s?rrc\rSrSrSrSrg)AttachDiskError,z[R"URUR5n[ TS9nUR Rc)UR R5UR l[R"U5HUupVUR RRRUR RRXVS95 MW U$)zRequest hook for merging the metadata and metadata from file. Args: unused_ref: ref to the service. args: The args for this method. request: The request to be made. Returns: Request with metadata field populated. versionkeyvalue) rConstructMetadataDictmetadatametadata_from_fileGetMessagesModulenodeNode MetadataValuesix iteritemsadditionalPropertiesappendAdditionalProperty) unused_refargsrequest metadata_dict tpu_messagesrBrC api_versions rProcessMergeMetadata..ProcessJs#88 t..0M$[9L||$*//==?gllmmM2  ll0077    ) ) < < = $%3 NrrrUrVs` r MergeMetadatarYGs, .rc&UnURS5(a/nURS5(a [[X R-55nURS5(a*[[U5[UR5- 5nU$)zGenerate the change to the tags on a resource based on the arguments. Args: args: The args for this method. tags: The current list of tags. Returns: The change to the tags after all of the arguments are applied. clear_tagsadd_tags remove_tags)IsKnownAndSpecifiedsortedsetr\r])rQtags tags_updates rGetTagsUpdateFromArgsrccs|+ l++K j))[==89:K m,,[)C0@0@,AABK rc^U4SjnU$)zARequest hook for constructing the updateMask for update requests.c>[5n[TS9nURS5(aURS5 URS5(aURS5 URS5(d,URS5(dURS5(a[R R U5nUR5(awURURRURR5nUR(a,URURl URS5 URS 5(d,URS 5(dURS 5(al[XRR5n[U5[URR5:wa!XrRlURS 5 URS 5(a[ R""SUR$5nURR'5URl[*R,"U5HUupURR(R.R1URR&R3XS95 MW URS5 GOURS5(d-URS5(dURS5(Gap0nURR(b?URR(R.Hn U R4XR6'M [8R:"USSUR<SS5n [8R:"U SUR>SSS5n [8R:"U SSSUR@S5n URR'5URl[*R,"U 5HUupURR(R.R1URR&R3XS95 MW URS5 URS5(GaURS5(dS/Ul![EU5n U (a/Ul!SupURFRI5H=n U S:XaURFSnMU S:XaURFSnM1[KSU -5e US:Xa!URLRNRPnOvU(aUS:Xa[URLRNRRn[UURB5S:a![KS[WURB5-5eO[KSU-5e[URB5nURMUUS 9nTS!:Xa0[YU5Ul-[]URR^U5 URR^R1U5 URS"5 GOURS#5(GaURS5(dS/Ul![EU5n U (a/Ul!URR^(d [aS$5e/nURR^HnUR1URb5 M [eU5HunnURfU:waMU (aURR^U O[URR^URZ5[URB5- nU(dURR^U OF[YU5URR^Ul- O [aS%URf-S&-5eURS"5 U(d [iS'5eS(RkU5Ul6U$))aRequest hook for constructing the updateMask for update requests. Args: unused_ref: ref to the service. args: The args for this method. request: The request to be made. Returns: Request with updateMask field populated. Raises: NoFieldsSpecifiedError: if no fields were specified. AttachDiskError: if the request for attaching a disk is invalid. DetachDiskError: if the request for detaching a disk is invalid. r? description internal_ipsz"network_config.enable_external_ips update_labels remove_labels clear_labelslabelsr\r]r[rarFNrArEupdate_metadataremove_metadataclear_metadata attach_diskworkerall)rrmodesourcezAargument --attach-disk: valid keys are [mode, source]; received: z read-onlyz read-writez]argument --attach-disk: can only attach disks in read-write to at most one worker; received: zcargument --attach-disk: key mode: can only attach disks in read-write or read-only mode; received: )rs sourceDiskv2alpha1 data_disks detach_diskzDargument --detach-disk: No data disks to detach from current TPU VM.z0argument --detach-disk: The specified data disk z) is not currently attached to the TPU VM.z]No fields would change as a result of this update; must specify at least one field to update.,)7r`rGr^addr DiffFromUpdateArgsMayHaveUpdatesApplyrI LabelsValuerHrk needs_updatercrarrDrFrJrErKrLrMrNrOrCrBr ApplyMapFlagsrnrlrmrpValidateWorkerIdsFieldrokeysr AttachedDiskModeValueValuesEnum READ_ONLY READ_WRITElenstrr_ workerIdsPreprocessDiskToAttach dataDisksr#rv enumerateryrjoin updateMask)rPrQrR update_maskrT labels_diff labels_updaterbrSrBrCitemmetadata_updateis_all_workers_specifiedrsrt mode_enumrpdisk_to_attachsource_disk_listdiski source_disk worker_diffrUs rrV#GenerateUpdateMask..Processzs"%K$[9L   ..oom$ //oo:;   11   11   00$$33D9k  # # % %#))    ) ) LL  !   % % - 4 4',,  //( #   ,,   //   ..)$ 0A0ABk [ S!2!23 3'   455$:: '')m*//==?gll m4*# 2299    + + > > ? & '5ooj!  " "#4 5 5  " "#4 5 5  " "#3 4 4m    *LL))>>D$(JJ- !?!..}d/3T5H5H$/35o!../3/C/CT4/35o!..d/3T5I5I4Qo*//==?gll o6*# 2299    + + > > ? & '7ooj!  ..  % %h / /g !7!= ! ld!!&&(# &=!!&)$ H_##H-& !  )   --AAKK t|+ --AALL t{{ a 3585EF    79= >  4;;f#001n  "#)&> w||55~F ll##N3ool# ! !- 0 0  % %h / /g !7!= ! \\ # #   ,,(($0)%&67.![   { *  #ll$$Q'  LL " "1 % / /136t{{3CD ll$$Q' .4[.A q!+ 8 > !L M  ool#  " ' ((+.G NrrrXs` rGenerateUpdateMaskrwsiV .rcBURbSURlU$)zUnset acceleratorType flag when it conflicts with topology arguments. Args: unused_ref: ref to the service. args: The args for this method. request: The request to be made. Returns: Request with metadata field populated. N)topologyrHacceleratorType)rPrQrRs rRemoveConflictingDefaultsr(s ]]#'GLL .rc0[R"SU5$)Ntpu)rrGr?s rrGrG8s  w //rc^U4SjnU$)z/Declarative request hook for TPU Start command.cF>AA[TS9R5nX2lU$Nr?)rGStartNodeRequeststartNodeRequest)refrQrR start_requestrUs rrV!StartRequestHook..Process?s'  %k:KKMM, NrrrXs` rStartRequestHookr< .rc^U4SjnU$)z.Declarative request hook for TPU Stop command.cF>AA[TS9R5nX2lU$r)rGStopNodeRequeststopNodeRequest)rrQrR stop_requestrUs rrV StopRequestHook..ProcessLs'  $[9IIKL* NrrrXs` rStopRequestHookrIrrc[R"UR5R5nUR S5(+=(a US:g$)NV1API_VERSION_UNSPECIFIED)rK text_type apiVersionupper startswith)rHrUs r IsTPUVMNoderVsA doo.446+$$T* * 7668rc^A[[RR[U55$)zRemoves Cloud TPU V1 API nodes from the 'list' output. Used with 'compute tpus tpu-vm list'. Args: response: response to ListNodes. args: the arguments for the list command. Returns: A response with only TPU VM (non-V1 API) nodes. )listrKmovesfilterrresponserQs rFilterTPUVMNodesr\s# cii{H5 66rc\rSrSrSrSrSrg)GuestAttributesListEntryilz%Holder for GetGuestAttributes output.c4XlX lX0lX@lgr.) worker_id namespacerBrC)selfrrrBrCs r__init__!GuestAttributesListEntry.__init__osNNHJr)rBrrCrNrrrrrrrrrrrrls -rrc A/n[UR5H\up4URRH=nUR [ X5R URUR55 M? M^ U$)a7Transforms the GuestAttributes into a flatter list. This is needed to make clearer output in the case of TPU pods, since they have many workers. Args: response: response to GetGuestAttributes. args: the arguments for the GetGuestAttributes command. Returns: A list of GuestAttributesListEntry objects. ) rguestAttributes queryValueitemsrNrrrBrC)rrQlstrgaentrys rTransformGuestAttributesrvsc  #112ea$$ jj "1oouyy%++ NP%3 *rc[U5Hup#URUR:waM!URUR:wa [S5e[ UR 5[ UR 5- (d [S5e[ [ UR UR -55UlX M g)aPreprocesses and validates the disk to attach. Validates the disk to attach is not already attached to the TPU VM with different mode or same mode and worker. Deletes the disk from the current_data_disks_list if it is already attached to the TPU VM with same mode but different worker. If the disk is currently attached to the TPU VM with same mode, joins the current worker list and the new worker list. Args: current_data_disks_list: the list of data disks currently attached to the TPU VM. disk_to_attach: the disk to attach to the TPU VM. Raises: AttachDiskError: if the disk is already attached to the TPU VM with different mode. AttachDiskError: if the disk is already attached to the TPU VM with same mode and worker. zWargument --attach-disk: the disk is already attached to the TPU VM with different mode.zZargument --attach-disk: the disk is already attached to the same set of workers of TPU VM.N)rrvrsrr`rr_)current_data_disks_listrrrs rrrs*23ga .333 ^(((  $  (( )C,? ?  /  & DNN^55 56 8N "!4rc[UR5S:XaURSS:XagURHOnUS:Xa%[UR5S:a [SS5eUR5(aME[SS5e g) ayChecks that the worker are numberic strings only. The only exception is "all" which is a special value that means all workers. If "all" is specified return True. Args: args: the arguments for the update command. Returns: True if only one string "all" is specified in args.worker False otherwise. Raises: WorkerIdsError: if the worker are not numberic strings only. rurrqTzargument --workerz,"all" cannot be specified with other worker.zQworker must be numeric strings only or "all". e.g. --worker=0,1,2 or --worker=allF)rrpr+ isnumeric)rQws rrrs  t{{1~6  ;;aEzc$++&*   8  ;;==   7   rcA[U5(aU$[RRS5 [R "S5 g)zVerifies that the node is a TPU VM node. If it is not a TPU VM node, exit with an error instead. Args: response: response to GetNode. args: the arguments for the list command. Returns: The response to GetNode if the node is TPU VM. zYERROR: Please use "gcloud compute tpus describe" for Cloud TPU nodes that are not TPU VM.ruN)rrerrPrintsysexitrs rCheckTPUVMNoders7  O''--./((1+rc^U4SjnU$)z2Request hook for parsing boot disk configurations.c>U(aURS5(dU$SnSnURR5H$nXSU4;dM [SR XC55e [ T S9nURR US5R5S:HnURR US5nU(a:T S :wa[R"S S 5eUc[S R Xx55e0n U(aURUS 9n XS'T S :Xa U(aUU S'U (a!UR"S0U D6URl U$)z$Parses configurations for boot disk. boot_diskzkms-keyzconfidential-computez.--boot-disk only supports arguments: {} and {}r?FalsetrueNrwz --boot-diskzBconfidential-compute is only available in the alpha release track.zQargument --boot-disk: with confidential-compute={} requires kms-key; received: {}) kmsKeyNamecustomerEncryptionKeyenableConfidentialComputer)r^rrr&formatrGgetlowerr InvalidArgumentExceptionCustomerEncryptionKeyBootDiskConfigrHbootDiskConfig) rPrQrRkms_key_arg_nameconfidential_compute_arg_namearg_namerTenable_confidential_computekms_keyboot_disk_config_kwargscustomer_encryption_keyrUs rrV,ParseBootDiskConfigurations..Processst t// << n $:!NN'') ,IJ J( < C C-   *%[9L 8'BHHJ   nn  !148G"  "11  P   ( --3V+.6  ! , B B!C!9P56j %@ %9:$0$?$?%% #%%gll! NrrrXs` rParseBootDiskConfigurationsrs1f .rc^U4SjnU$)z2Request hook for setting the source machine image.c>URS5(ah[TS9nURR(dUR 5URlUR URRlU$)z9Sets the source machine image in the request if provided.imager?) IsSpecifiedrGrHrrr sourceImage)rPrQrRrTrUs rrVSetImage..Process%s[   &{;l \\ ( (&2&A&A&C #04 gll!!- NrrrXs` rSetImager"s .rc[R"SS5n[R"SS5nURUS9nURR U5nUR $)zBReturns the Cloud project number associated with the `project_id`.cloudresourcemanagerv1) projectId)rrGGetClientInstance&CloudresourcemanagerProjectsGetRequestprojectsGet projectNumber) project_idcrm_message_moduleresource_managerreqr2s rProjectIdToProjectNumberr1se--.DdK++,BDIAA B #  % % ) )# .'   rcSnSnSnURS5(a[R"X1R5(ac[R R RR5n[U5nURXqRUR5nU(aXRRRlU$)zRequest hook for creating the target reservation name. Args: unused_ref: ref to the service. args: The args for this method. request: The request to be made. Returns: Request with reservationName field populated. z^[a-zA-Z0-9-]+$z(projects/{}/locations/{}/reservations/{}N reservation)r^rematchrrr0r1r2r3rrzonerHschedulingConfigreservationName)rPrQrRshort_reservation_name_patternfull_reservation_name_patternreservation_namer project_numbers rCreateReservationNamer;s$5 "L m,,$&6&622""''//99;J-j9N4;; 4#3#34DLL!!1 .rc^U4SjnU$)z'Sets the provisioning model enum value.c|>[T5nUR(a;URRRUR R lU$UR(d;URRRUR R lU$URRSS5R5nURRU5nXRR R lU$![a*n[R"SURS35UeSnAff=f)N-_z--provisioning-modelzV is not a valid provisioning model, must be one of [standard, spot, reservation-bound])rGspotSchedulingConfig ProvisioningModelValueValuesEnumSPOTrHrprovisioningModelprovisioning_modelSTANDARDreplacer TypeErrorr r)rrQrRrTnormalized_candidatecandidate_enumerUs rrV%SetProvisioningModel..ProcessXs$[1L yy  ' ' H H M Mll##5n  " "  ' ' H H Q Qll##5n !44<O@G ;; / / B B7 KKrc[RRRR SS9n[ R RUUUS.SS9nURRUR5UUS9nURRRU5n [ R RU RSS9n UR!X5$) z'Updates the TPU node in the given zone.Tr:r<r?r@)rCrHrz!tpu.projects.locations.operations)rB)rr0r1r2rrrDrEr4%TpuProjectsLocationsNodesPatchRequestrGr5rHPatchParseRelativeNamerCWaitForOperation) rrCrrHrpoller_messager2rIrR operation operation_refs r UpdateNodeTPUNode.UpdateNodes$$,,00$0?G$-$6$6$<$< ! 2 %=% !mmAA * 7 7 9 B G  44::7CI&&88#F9M   ??rcUb URcOURRRURRRR X#S9/S9$SnURHnUR U:XdMUn O UbX4lU$URRURRRR X#S95 U$)a Updates a key in the TPU metadata object. If the key does not exist, it is added. Args: metadata: tpu.messages.Node.MetadataValue, the TPU's metadata. key: str, the key to be updated. value: str, the new value for the key. Returns: The updated metadata. NrA)rM)rMr4rIrJrOrBrCrN)rrErBrCrxs rUpdateMetadataKeyTPUNode.UpdateMetadataKeys888@ ]]   - -mm  ..AAB( ).** D  * * # + j O##** --   * * = = > $% Orc[R"URRURR5n[R "X1U5$r.)r CloudOperationPollerr5rHprojects_locations_operationsWaitFor)rrYmessageoperation_pollers rrVTPUNode.WaitForOperationsA22 ,, 113 >>*7 CCr)r3r5r4N)rr) rrrrrrr7rrOrZr^rVrrrrr.r.vs-.E =L$@0 DDrr.c\rSrSrSrSrSrg)SSHPreppedNodeiaObject that has all the data needed to successfully SSH into a node. Attributes: worker_ips: The IPs of the workers of the node. ssh_helper: The ssh_helper used to SSH into the node. id: The id of the node. tpu_name: The unqualified TPU VM name. instance_names: The name of the instances of the workers of the node. project: The project associated with the node. command_list: The list of the commands passed into ssh. remainder: The remainder list of ssh_args used to pass into the SSH command. host_key_suffixes: The host key suffixes associated with the node. user: The user executing the SSH command. release_track: The release track for the SSH protos (Alpha, Beta, etc.). enable_batching: A bool indicating if the user enabled batching for the node. cXlX lX0lX@l/UlSUlSUl/UlSUl/Ul SUl /Ul gr.) tpu_nameuserr6enable_batching worker_ipsr7idinstance_namesr2 command_list remainderhost_key_suffixes)rrjrkr6rls rrSSHPreppedNode.__init__sSMI&*DODODGDDLDDNDr) rprlrrrnror2r6rqr7rjrkrmNrrrrrhrhs $  rrhc,^\rSrSrSrU4SjrSrU=r$)SCPPreppedNodei zObject that has all the data needed to successfully SCP into a node. Attributes: srcs: The sources for SCP. dst: The destination for SCP. cF>[[U] XX45 XPlX`lgr.)superrursrcsdst)rrjrkr6rlrxry __class__s rrSCPPreppedNode.__init__s# .$( IHr)ryrx)rrrrrrr __classcell__)rzs@rruru srru)r0)rw)8r __future__rrrrrgooglecloudsdk.api_lib.computerrgooglecloudsdk.api_lib.utilrr googlecloudsdk.callioper r $googlecloudsdk.command_lib.util.argsr r googlecloudsdk.coresdk_core_exceptionsrrrrKErrorrrr#r&r+r4r<rYrcrrrGrrrrobjectrrrrrrrrrr,r.rhrurrrrsE/&' 79,.(.<9A#*) @066@E)//EE)//E9!4!:!:99(..9<  8(nb  0  8 7 v ,%#PB(6r 6@mDfmD`  V  F^r