Wf $SrSSKrSSKJr SSKJr SSKJr SSKJr SSK J r SSK Jr S"Sjr S"S jr S#S jr S$S jrS#S jrS"S jrS"SjrS%SjrS&SjrS'S\R(S\S\S\S\4 SjjrS(S\R(S\S\S\S\4 SjjrS)S\R(S\S\S\S\4 SjjrS*SjrS+SjrS,SjrS-SjrS.SjrS/SjrS0Sjr S#S jr!S1S!jr"g)2z&Flag utils for clusters command group.N)utils) arg_parsers) exceptions)parser_arguments) flag_typesc US;a[R"SU35eURSS[R"[ R "US9SS9UUS9 g ) z-Adds a config flag for the given API version.v1alphaUnsupported API version: z--configz@Configuration of the cluster specs in the form of a JSON object.) api_versionT)specenable_shorthandhelptyperequiredhiddenN)r ToolException add_argumentr ArgObjectrGetClusterFlagTypeparserr rrs Alib/googlecloudsdk/command_lib/cluster_director/clusters/flags.py AddConfigrsc #  " "%>{m#L MM M  ''K@   c US;a[R"SU35eURS[R"S5[ R "[SS9UUS9 g) z3Adds an update mask flag for the given API version.r r z --update-maskzq Update mask to specify the fields to update. For e.g. --update-mask "description,labels" T) value_typerrNrrrtextwrapdedentrrstrrs r AddUpdateMaskr#,sb #  " "%>{m#L MM ??   C$ G  rcUS;a[R"SU35eURS[R"S5[ US9 g)z2Adds a description flag for the given API version.r r z --descriptionzY Description of the cluster. For e.g. --description {description} rrrNrrrr r!r"rr rs rAddDescriptionr(=sR #  " "%>{m#L MM ??    rcUS;a[R"SU35eSU3nU(aSU3nURSU3[R"SUS35[ R US9 U(aXURSU3[R"S US 35[R"[S 9[R"5US 9 g g )z-Adds a labels flag for the given API version.r r remove-add---z@ Cluster labels as key value pairs. For e.g. --z key1=value1,key2=value2 r%zL Parameters to remove cluster label by key. For e.g. --z {key1},{key2},...  element_typerractionrN) rrrr r!rLABELrArgListr" FlattenActionrnamer rinclude_update_flagsremove_flag_names r AddLabelsr8Ms #  " "%>{m#L MMtf% $=D 4&k ?? 6          __"''(    c 2((* rc US;a[R"SU35eURS[R"S5[ R "[[S.S/SS9US 9 g ) z5Adds a create network flag for the given API version.r r z--create-networkz{ Parameters to create a network. For e.g. --create-network name={network},description={description} )r5 descriptionr5Tr required_keysrr%Nrr's rAddCreateNetworkr=rst #  " "%>{m#L MM ??        rcUS;a[R"SU35eURS[R"S5[ UUS9 g)z.Adds a network flag for the given API version.r r z --networkzY Reference of existing network name. For e.g. --network {network} rNr&rs rAddNetworkSourcer?sU #  " "%>{m#L MM ??    rcUS;a[R"SU35eURS[R"S5[ UUS9 g)z-Adds a subnet flag for the given API version.r r z--subnetz{ Reference of existing subnetwork name. For e.g. --subnet regions/{region}/subnetworks/{subnetwork} rNr&rs rAddSubnetSourcerAsU #  " "%>{m#L MM ??    rc US;a[R"SU35e[R"[R"U55nU(aSnUR SU3[ R"S5[R"[URR[[URR[S./SQSSS 9[R"5US 9 g ) z8Adds a create filestores flag for the given API version.r r zadd-new-filestore-instancesr,a Parameters to create a filestore instance. For e.g. --create-filestores name=locations/{location}/instances/{filestore},tier=REGIONAL,capacityGb={filestoreSize},fileshare={fileshare} Supported tier values: - ZONAL - REGIONAL Supported protocol values: - NFSV3 - NFSV41 - If not specified, defaults to NFSV3 Defaults: - protocol: NFSV3 )r5tier capacityGb fileshareprotocolr:)r5rCrDrETr r<rrepeatedr/N)rr api_utilsGetMessagesModuleGetReleaseTrackrr r!rrr"NewFilestoreConfigTierValueValuesEnumintProtocolValueValuesEnumr3rr5r rr6messagess rAddCreateFilestoresrRs #  " "%>{m#L MM  ( ()B)B;)O P( (D 4&k ??  "  11EE"55MM  D  & & ( C"rcUS;a[R"SU35eSnU(aSnURSU3[R"SUS35[ R "[S9[ R"5US 9 U(aXURSU3[R"S US 35[ R "[S9[ R"5US 9 g g ) z1Adds a filestores flag for the given API version.r r zremove-filestore-instanceszadd-filestore-instancesr,zG Reference of existing filestore instance. For e.g. --z2 locations/{location}/instances/{filestore} r-r/zc Parameters to remove filestore instance config by filestore name. For e.g. --ze locations/{location}/instances/{filestore1},locations/{location}/instances/{filestore2},... N rrrr r!rr2r"r3r4s r AddFilestoresrUs #  " "%>{m#L MM1 $D 4&k ?? 6    C 0  & & (      __"''(    c 2((* rrr5r rr6c US;a[R"SU35e[R"[R"U55nU(aSnUR SU3[ R"SUS35[R"[URR[[S.S/S S S 9[R"5US 9 g ) zJAdds a create Google Cloud Storage buckets flag for the given API version.r r zadd-new-storage-bucketsr,zQ Parameters to create a Google Cloud Storage bucket. For e.g. --au name={bucket-path} Supported storageClass values: - STANDARD - NEARLINE - COLDLINE - ARCHIVE Defaults: - storageClass: STANDARD Note: - Either storageClass or enableAutoclass can be set. - if enableAutoclass is set, enableHNS should not be set. - HNS: Hierarchical namespace )r5 storageClassenableAutoclass enableHNSr5TrGr/N)rrrIrJrKrr r!rrr"NewBucketConfigStorageClassValueValuesEnumboolr3rPs rAddCreateGcsBucketsr]s #  " "%>{m#L MM  ( ()B)B;)O P( $D 4&k ?? 6  &  **FF!%     & & ( G$rcUS;a[R"SU35eSU3nU(aSU3nURSU3[R"SUS35[ R "[S9[ R"5US 9 U(aXURSU3[R"S US 35[ R "[S9[ R"5US 9 g g ) zCAdds a Google Cloud Storage buckets flag for the given API version.r r zremove-storage-z add-storage-r,zP Reference of existing Google Cloud Storage bucket. For e.g. --z {bucket-path} r-r/zb Parameters to remove Google Cloud Storage bucket by bucket name. For e.g. --z! {bucket1},{bucket2},... NrTr4s r AddGcsBucketsr_7s  #  " "%>{m#L MM&tf- $ D 4&k ?? 6    C 0  & & (      __"''(    c 2((* rc .US;a[R"SU35eU(aSnURSU3[R"SUS35[ R "[[[[S./SQS S S 9[ R"5US 9 g ) z5Adds a create lustres flag for the given API version.r r zadd-new-lustre-instancesr,zE Parameters to create a Lustre instance. For e.g. --zd name=locations/{location}/instances/{lustre},capacityGb={lustreSize},filesystem={filesystem} )r5 filesystemrDr:)r5rDraTrGr/N) rrrr r!rrr"rNr3)rr5r rr6s rAddCreateLustresrb]s #  " "%>{m#L MM %D 4&k ?? 6     =  & & ( 'rcUS;a[R"SU35eSnU(aSnURSU3[R"SUS35[ R "[S9[ R"5US 9 U(aXURSU3[R"S US 35[ R "[S9[ R"5US 9 g g ) z.Adds a lustres flag for the given API version.r r zremove-lustre-instanceszadd-lustre-instancesr,zD Reference of existing Lustre instance. For e.g. --z/ locations/{location}/instances/{lustre} r-r/z] Parameters to remove lustre instance config by lustre name. For e.g. --z_ locations/{location}/instances/{lustre1},locations/{location}/instances/{lustre2},... NrTr4s r AddLustresrds #  " "%>{m#L MM. !D 4&k ?? 6    C 0  & & (      __"''(    c 2((* rc US;a[R"SU35eSU3nU(aSU3nURSU3[R"SUS35[ R "[[[S./SQS S S 9[ R"5US 9 U(aXURSU3[R"S US 35[ R"[S9[ R"5US 9 gg)z;Adds an on demand instances flag for the given API version.r r r*r+r,zO Parameters to define cluster on demand instances. For e.g. --< id={computeId},zone={zone},machineType={machineType} idzone machineTypeTrGr/z` Parameters to remove on demand instances config by compute id. For e.g. --' {computeId1},{computeId2},... r-N rrrr r!rrr"r3r2r4s rAddOnDemandInstancesrm# #  " "%>{m#L MMtf% $=D 4&k ?? 6     6  & & ( %(    __"''(    c 2((* rc US;a[R"SU35eSU3nU(aSU3nURSU3[R"SUS35[ R "[[[S./SQS S S 9[ R"5US 9 U(aXURSU3[R"S US 35[ R"[S9[ R"5US 9 gg)z6Adds an spot instances flag for the given API version.r r r*r+r,zJ Parameters to define cluster spot instances. For e.g. --rfrgTrGr/zZ Parameters to remove spot instance config by compute id. For e.g. --rkr-Nrlr4s rAddSpotInstancesrprnrc US;a[R"SU35eSU3nU(aSU3nURSU3[R"SUS35[ R "[[[S./SQS S S 9[ R"5US 9 U(aXURSU3[R"S US 35[ R"[S9[ R"5US 9 gg)z:Adds an reserved instances flag for the given API version.r r r*r+r,zN Parameters to define cluster reserved instances. For e.g. --zd id={computeId},reservation=zones/{zone}/reservations/{reservation},machineType={machineType} )rh reservationrjTrGr/z^ Parameters to remove reserved instance config by compute id. For e.g. --rkr-Nrlr4s rAddReservedInstancesrss# #  " "%>{m#L MMtf% $=D 4&k ?? 6      =  & & ( %(    __"''(    c 2((* rc US;a[R"SU35eSU3nU(aSU3nURSU3[R"SUS35[ R "[[[[S./SQS S S 9[ R"5US 9 U(aXURSU3[R"S US 35[ R"[S9[ R"5US 9 gg)z:Adds an DWS Flex instances flag for the given API version.r r r*r+r,zN Parameters to define cluster DWS Flex instances. For e.g. --zO id={computeId},zone={zone},machineType={machineType},maxDuration=10000s )rhrirj maxDurationTrGr/z^ Parameters to remove DWS Flex instance config by compute id. For e.g. --rkr-Nrlr4s rAddDwsFlexInstancesrv3s' #  " "%>{m#L MMtf% $=D 4&k ?? 6      E  & & ( '*    __"''(    c 2((* rc  US;a[R"SU35eSU3nSU3nU(aSU3nURSU3[R"SUS35[ R "[[[[[ R "5[RS .S S /S S S 9[ R"5UUS9 U(aURSU3[R"SUS35[ R "[[[S.S /S S S 9[ R"5UUS9 URSU3[R"SUS35[ R"[S9[ R"5UUS9 gg)z6Adds a slurm node sets flag for the given API version.r r update-r*r+r,zP Parameters to define slurm cluster nodeset config. For e.g. --a id={nodesetId},computeId={computeId},staticNodeCount={staticNodeCount},maxDynamicNodeCount={maxDynamicNodeCount},startupScript="echo hello",labels="{key1=value1,key2=value2}" Defaults: - staticNodeCount: 1 Note: - startupScript: - Either str or file_path - For file_path, only bash file format (.sh or .bash) is supported. - For file_path, only absolute path is supported. )rh computeIdstaticNodeCountmaxDynamicNodeCount startupScriptlabelsrhryTrGrrr0rrz_ Parameters to define and update slurm cluster nodeset config. For e.g. --zd id={nodesetId},staticNodeCount={staticNodeCount},maxDynamicNodeCount={maxDynamicNodeCount} )rhrzr{zZ Parameters to remove slurm nodeset config by nodeset id. For e.g. --z' {nodesetId1},{nodesetId2},... r-N) rrrr r!rrr"rNrr1r3r2rr5r rrr6update_flag_namer7s rAddSlurmNodeSetsrcs #  " "%>{m#L MMtf%tf% $=D 4&k ?? 6     !$%(*446"((  {+  & & ( ? B    __"''(    " "#&'*  &!  ((*'*    __"''(    c 2((* -rc US;a[R"SU35eSU3nSU3nU(aSU3nURSU3[R"SUS35[ R "[[ R "[S S 9[S .S S /S S S9[ R"5UUS9 U(aURSU3[R"SUS35[ R "[[ R "[S S 9[S .S /S S S9[ R"5UUS9 URSU3[R"SUS35[ R"[S9[ R"5UUS9 gg)z7Adds a slurm partitions flag for the given API version.r r rxr*r+r,zL Parameters to define slurm cluster partitions. For e.g. --zO id={partitionId},nodesetIds=[{nodesetId1},{nodesetId2}],exclusive=false T)rrH)rh nodesetIds exclusiverhrrGr~za Parameters to define and update slurm cluster partition config. For e.g. --zQ id={partitionId},nodesetIds=[{nodesetId1},{nodesetId2}],exclusive=false z^ Parameters to remove slurm partition config by partition id. For e.g. --z+ {partitionId1},{partitionId2},... r-N) rrrr r!rrr"r\r3r2rs rAddSlurmPartitionsrs #  " "%>{m#L MMtf%tf% $=D 4&k ?? 6   '11 4  |,  & & ( +.    __"''(    " ")33"T"  &!  ((*+.    __"''(    c 2((* 1rcUS;a[R"SU35eURS[R"S5[ US9 g)z>Adds a slurm default partition flag for the given API version.r r z--slurm-default-partitionz Parameters to define slurm cluster default partition. For e.g. --slurm-default-partition {partitionId} r%Nr&r's rAddSlurmDefaultPartitionrsR #  " "%>{m#L MM! ??    rc US;a[R"SU35eUnU(a*SU3nSUS3n[[R"5S.n/n OESn[ [ [[ [ [R"5[RS.nS S /n URS U3[R"U5[R"UU S S 9UUS9 g)z7Adds a slurm login node flag for the given API version.r r rxz Parameters to update slurm cluster login node. Only count and startupScript can be updated. For e.g. --z( count=2,startupScript="echo hello" )countr|a7 Parameters to define slurm cluster login node. For e.g. --slurm-login-node machineType={machineType},zone={zone},count={count},enableOSLogin=true,enablePublicIPs=true,startupScript="echo hello",labels="{key1=value1,key2=value2}" Defaults: - count: 1 - enableOSLogin: true - enablePublicIPs: true Note: - startupScript: - Either str or file_path - For file_path, only bash file format (.sh or .bash) is supported. - For file_path, only absolute path is supported. )rjrir enableOSLoginenablePublicIPsr|r}rjrir,Tr;rN) rrrNrrr"r\rr1rr r!) rr5r rrr6 flag_name help_textr req_keyss rAddSlurmLoginNoders #  " "%>{m#L MM)$ I; I$..0 DH I"$..0"" Dv&H 9+ ??9 %       r)NFF)NF)r}NFF)zcreate-filestoresNFF) filestoresNFF)zcreate-bucketsNFF)bucketsNFF)zcreate-lustresNFF)lustresNFF)zon-demand-instancesNFF)zspot-instancesNFF)zreserved-instancesNFF)zdws-flex-instancesNFF)zslurm-node-setsNFFF)zslurm-partitionsNFFF)zslurm-login-nodeNFFF)#__doc__r *googlecloudsdk.api_lib.hypercomputeclusterrrIgooglecloudsdk.callioperrr4googlecloudsdk.command_lib.cluster_director.clustersrrr#r(r8r=r?rArRrUArgumentInterceptorr"r\r]r_rbrdrmrprsrvrrrrrrrs -I/.4KF  " $   "J."&   /h   #P!!& 1  0 01 11  1  1l!& #  0 0# ##  #  #P!!&   0 0        J   #P   ,b   ,b   ,b   -d    Ql    IX $    >r