Z0dZddlZddlmZddlmZddlmZddlm Z ddlmZddZ ddZ dd Z dd Z dd Zdd Zdd ZddZddZddZddZddZddZ ddZ d dZ d!dZ d"dZ d#dZ d$dZddZddZy)%z&Flag utils for clusters command group.N)utils) arg_parsers) exceptions) flag_typesc |dvrtjd||jddtjt j |d||y ) 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|dvrtjd||jdtjdt j td||y) z3Adds an update mask flag for the given API version.rr z --update-maskzq Update mask to specify the fields to update. For e.g. --update-mask "description,labels" T) value_typer rNrrrtextwrapdedentrrstrrs r AddUpdateMaskr"+sb #  " "%>{m#L MM ??   C$ G  rc|dvrtjd||jdtjdt |y)z2Adds a description flag for the given API version.rr z --descriptionzY Description of the cluster. For e.g. --description {description} rrrNrrrrr r!rr rs rAddDescriptionr'<sR #  " "%>{m#L MM ??    rc|dvrtjd|d|}|rd|}|jd|tjd|dt j ||r[|jd|tjd |d tjt tj| y y )z-Adds a labels flag for the given API version.rr 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) rrrrr rLABELrArgListr! FlattenActionrnamer rinclude_update_flagsremove_flag_names r AddLabelsr7Ls #  " "%>{m#L MMtf% $=D 4&k ?? 6          __"''(    c 2((* rc |dvrtjd||jdtjdt j ttddgd| y ) z5Adds a create network flag for the given API version.rr z--create-networkz{ Parameters to create a network. For e.g. --create-network name={network},description={description} )r4 descriptionr4Tr required_keysr r$Nrr&s rAddCreateNetworkr<qst #  " "%>{m#L MM ??        rc|dvrtjd||jdtjdt ||y)z.Adds a network flag for the given API version.rr z --networkzY Reference of existing network name. For e.g. --network {network} rNr%rs rAddNetworkSourcer>sU #  " "%>{m#L MM ??    rc|dvrtjd||jdtjdt ||y)z-Adds a subnet flag for the given API version.rr z--subnetz{ Reference of existing subnetwork name. For e.g. --subnet regions/{region}/subnetworks/{subnetwork} rNr%rs rAddSubnetSourcer@sU #  " "%>{m#L MM ??    rc |dvrtjd|tjtj|}|j dt jdtjt|jjtt|jjtdgdddtj| y ) z8Adds a create filestores flag for the given API version.rr z--create-filestoresa  Parameters to create a filestore instance. For e.g. --create-filestores name=locations/{location}/instances/{filestore},tier=BASIC_HDD,capacityGb={filestoreSize},fileshare={fileshare} Supported tier values: - BASIC_HDD - BASIC_SSD - HIGH_SCALE_SSD - ZONAL - ENTERPRISE - REGIONAL Supported protocol values: - NFSV3 - NFSV41 - If not specified, defaults to NFSV3 Defaults: - protocol: NFSV3 )r4tier capacityGb fileshareprotocolr9)r4rBrCrDTr r;r repeatedr.N)rr api_utilsGetMessagesModuleGetReleaseTrackrrr rrr!NewFilestoreConfigTierValueValuesEnumintProtocolValueValuesEnumr2rr rmessagess rAddCreateFilestoresrQs #  " "%>{m#L MM  ( ()B)B;)O P( ??  *  11EE"55MM  D  & & ( K&rc|dvrtjd||jdtjdt j tt j|y)z1Adds a filestores flag for the given API version.rr z --filestoresz Reference of existing filestore instance. For e.g. --filestores locations/{location}/instances/{filestore} r,r.N rrrrr rr1r!r2r&s r AddFilestoresrTsi #  " "%>{m#L MM ??    C 0  & & (  rc ||dvrtjd|tjtj|}|j dt jdtjt|jjttddgddtj| y ) zJAdds a create Google Cloud Storage buckets flag for the given API version.rr z--create-bucketsa Parameters to create a Google Cloud Storage bucket. For e.g. --create-buckets 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 )r4 storageClassenableAutoclass enableHNSr4TrFr.N)rrrHrIrJrrr rrr!NewBucketConfigStorageClassValueValuesEnumboolr2rOs rAddCreateGcsBucketsr\s #  " "%>{m#L MM  ( ()B)B;)O P( ??  &  **FF!%     & & ( G$rc|dvrtjd||jdtjdt j tt j|y)zCAdds a Google Cloud Storage buckets flag for the given API version.rr z --bucketszl Reference of existing Google Cloud Storage bucket. For e.g. --buckets {bucket-path} r,r.NrSr&s r AddGcsBucketsr^i #  " "%>{m#L MM ??    C 0  & & (  rc |dvrtjd||jdtjdt j ttttdgdddt j| y ) z5Adds a create lustres flag for the given API version.rr z--create-lustresz Parameters to create a Lustre instance. For e.g. --create-lustres name=locations/{location}/instances/{lustre},capacityGb={lustreSize},filesystem={filesystem} )r4 filesystemrCr9)r4rCraTrFr.N) rrrrr rrr!rMr2r&s rAddCreateLustresrb&s #  " "%>{m#L MM ??     =  & & ( 'rc|dvrtjd||jdtjdt j tt j|y)z.Adds a lustres flag for the given API version.rr z --lustreszz Reference of existing Lustre instance. For e.g. --lustres locations/{location}/instances/{lustre} r,r.NrSr&s r AddLustresrdAr_rc |dvrtjd|d|}|rd|}|jd|tjd|dt j tttdgdd d t j| |r[|jd|tjd |d t jtt j| yy)z;Adds an on demand instances flag for the given API version.rr r)r*r+zO Parameters to define cluster on demand instances. For e.g. --< id={computeId},zone={zone},machineType={machineType} idzone machineTypeTrFr.z` Parameters to remove on demand instances config by compute id. For e.g. --' {computeId1},{computeId2},... r,N rrrrr rrr!r2r1r3s rAddOnDemandInstancesrmR# #  " "%>{m#L MMtf% $=D 4&k ?? 6     6  & & ( %(    __"''(    c 2((* rc |dvrtjd|d|}|rd|}|jd|tjd|dt j tttdgdd d t j| |r[|jd|tjd |d t jtt j| yy)z6Adds an spot instances flag for the given API version.rr r)r*r+zJ Parameters to define cluster spot instances. For e.g. --rfrgTrFr.zZ Parameters to remove spot instance config by compute id. For e.g. --rkr,Nrlr3s rAddSpotInstancesrprnrc |dvrtjd|d|}|rd|}|jd|tjd|dt j tttdgdd d t j| |r[|jd|tjd |d t jtt j| yy)z:Adds an reserved instances flag for the given API version.rr r)r*r+zN Parameters to define cluster reserved instances. For e.g. --zd id={computeId},reservation=zones/{zone}/reservations/{reservation},machineType={machineType} )rh reservationrjTrFr.z^ Parameters to remove reserved instance config by compute id. For e.g. --rkr,Nrlr3s rAddReservedInstancesrss# #  " "%>{m#L MMtf% $=D 4&k ?? 6      =  & & ( %(    __"''(    c 2((* rc |dvrtjd|d|}|rd|}|jd|tjd|dt j ttttdgdd d t j| |r[|jd|tjd |d t jtt j| yy)z:Adds an DWS Flex instances flag for the given API version.rr r)r*r+zN Parameters to define cluster DWS Flex instances. For e.g. --zO id={computeId},zone={zone},machineType={machineType},maxDuration=10000s )rhrirj maxDurationTrFr.z^ Parameters to remove DWS Flex instance config by compute id. For e.g. --rkr,Nrlr3s rAddDwsFlexInstancesrvs' #  " "%>{m#L MMtf% $=D 4&k ?? 6      E  & & ( '*    __"''(    c 2((* rc |dvrtjd|d|}d|}|rd|}|jd|tjd|dt j ttttt j tjd d d gd d t j|||r|jd|tjd|dt j tttdd gd d t j|||jd|tjd|dt jtt j||yy)z6Adds a slurm node sets flag for the given API version.rr 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 startupScriptlabelsrhryTrFrrr/rrz_ 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) rrrrr rrr!rMrr0r2r1rr4r rrr5update_flag_namer6s rAddSlurmNodeSetsrs #  " "%>{m#L MMtf%tf% $=D 4&k ?? 6     !$%(*446"((  {+  & & ( ? B    __"''(    " "#&'*  &!  ((*'*    __"''(    c 2((* -rc |dvrtjd|d|}d|}|rd|}|jd|tjd|dt j tt j td td d d gd d t j|||r|jd|tjd|dt j tt j td td d gd d t j|||jd|tjd|dt jtt j||yy)z7Adds a slurm partitions flag for the given API version.rr rxr)r*r+zL Parameters to define slurm cluster partitions. For e.g. --zO id={partitionId},nodesetIds=[{nodesetId1},{nodesetId2}],exclusive=false T)rrG)rh nodesetIds exclusiverhrrFr~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) rrrrr rrr!r[r2r1rs rAddSlurmPartitionsrcs #  " "%>{m#L MMtf%tf% $=D 4&k ?? 6   '11 4  |,  & & ( +.    __"''(    " ")33"T"  &!  ((*+.    __"''(    c 2((* 1rc|dvrtjd||jdtjdt |y)z>Adds a slurm default partition flag for the given API version.rr 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! ??    rc6|dvrtjd||jdtjdt j tttttt j tjdddgd || y ) z7Adds a slurm login node flag for the given API version.rr z--slurm-login-nodea7 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. )rjricount enableOSLoginenablePublicIPsr|r}rjriTr:rN) rrrrr rrr!rMr[rr0rs rAddSlurmLoginNoders #  " "%>{m#L MM ??    #!%*446"(( '/  A!r)NFF)NF)r}NFF)zon-demand-instancesNFF)zspot-instancesNFF)zreserved-instancesNFF)zdws-flex-instancesNFF)zslurm-node-setsNFFF)zslurm-partitionsNFFF)__doc__r*googlecloudsdk.api_lib.hypercomputeclusterrrHgooglecloudsdk.callioperr4googlecloudsdk.command_lib.cluster_director.clustersrrr"r'r7r<r>r@rQrTr\r^rbrdrmrprsrvrrrrrrrs -I/.KF  " $   "J.""+\")X"6&   ,b   ,b   ,b   -d    Ql    IX  %r