LJSrSSKJr SSKJr SSKJr SSKJr SSK J r SSK J r SSK J r SS KJr SS KJr SS KJr SS KJr SS KJr /S Qr/SQr/SQrSrSrSrSrSr Sr!Sr"Sr#S*Sjr$Sr%Sr&Sr'Sr(Sr)Sr*Sr+S r,S!r-S"r.S#r/S$r0S%r1S&r2S'r3S(r4g))+z;Utilities for flags for `gcloud colab-enterprise` commands.)runtime_templates)runtimes) arg_parsers)base)concepts)deps) completers)flags)iam_util)concept_parsers) properties)NVIDIA_TESLA_V100NVIDIA_TESLA_T4NVIDIA_TESLA_A100NVIDIA_A100_80GB NVIDIA_L4) PD_STANDARDPD_SSD PD_BALANCED PD_EXTREME)(POST_STARTUP_SCRIPT_BEHAVIOR_UNSPECIFIEDRUN_ONCERUN_EVERY_STARTDOWNLOAD_AND_RUN_EVERY_STARTc[R"SS[R"[R R R5/S9$)zeGet the attribute config for the region resource. Returns: The resource attribute for the region. regionz Cloud region for the {resource}.)name help_text fallthroughs)r ResourceParameterAttributeConfigrPropertyFallthroughr VALUEScolabr8lib/googlecloudsdk/command_lib/colab_enterprise/flags.pyGetRegionAttributeConfigr'.s@  2 2 2,,Z->->-D-D-K-KLM r%c S SjnU(a SnSnSnSnO SnSnSS0nSS /0n[RRUU"5S RU5SUUUS 9R U5 g) zAdd a resource argument for a runtime template to the parser. Args: parser: argparse parser for the command. verb: str, the verb to describe the resource, such as 'to update'. is_positional: bool, True for if arg is positional; False if flag. c^[R"SU[R[5SS9$)Nz6aiplatform.projects.locations.notebookRuntimeTemplatesT resource_name projectsId locationsIddisable_auto_completersr ResourceSpec DEFAULT_PROJECT_ATTRIBUTE_CONFIGr'r+s r&GetRuntimeTemplateResourceSpecEAddRuntimeTemplateResourceArg..GetRuntimeTemplateResourceSpecDs-  @#<<,. $  r%runtime_templateFNz--runtime-templateTr--regionzUnique name of the runtime template {}. This was optionally provided by setting --runtime-template-id in the create runtime-template command, or was system-generated if unspecified.)requiredprefixesflag_name_overridescommand_level_fallthroughs)zruntime templater ConceptParser ForResourceformat AddToParser)parserverb is_positionalr3arg_namer9r:rs r&AddRuntimeTemplateResourceArgrE;s!HHL#HH#R.zl+L++$&117-!-, Kr%cSSjn[RRSU"5SRU5SS9R U5 g)zAdd a resource argument for a runtime template to the parser. Args: parser: argparse parser for the command. verb: str, the verb to describe the resource, such as 'to update'. runtimec^[R"SU[R[5SS9$)Nz.aiplatform.projects.locations.notebookRuntimesTr*r/r2s r&GetRuntimeResourceSpec5AddRuntimeResourceArg..GetRuntimeResourceSpecls-  8#<<,. $  r%zUnique name of the runtime {}. This was optionally provided by setting --runtime-id in the create runtime command, or was system-generated if unspecified.Tr8N)rGr<)rArBrIs r&AddRuntimeResourceArgrLdsH++fTl ,Kr%c[R"SS[5[RSS9n[R R SUSRU5SS9RU5 g) zAdd a resource argument for a Vertex AI region to the parser. Args: parser: argparse parser for the command. verb: str, the verb to describe the resource, such as 'to update'. zaiplatform.projects.locationsrT)r+r-r,r.r7zkCloud region {}. Please see https://cloud.google.com/colab/docs/locations for a list of supported regions.rKN) rr0r'r1r r=r>r?r@)rArBregion_resource_specs r&AddRegionResourceArgrOsk"..%*,::" ++&, ,Kr%c|^U4Sjn[RRSU"5T5RU5 g)zAdds Resource arg for network to the parser. Args: help_text: str, the help text for the flag. parser: argparse parser for the command. cf>U4Sjn[R"SSU"5[RSS9$)z>Constructs and returns the Resource specification for network.cL>[R"ST[RS9$)Nnetworkrr completer)rr compute_network_flagsNetworksCompleterrsr&NetworkAttributeConfigUAddNetworkResourceArg..GetNetworkResourceSpec..NetworkAttributeConfigs%  6 6);;r%zcompute.networksrSF)r+rSprojectr.rr0r1)rYrs r&GetNetworkResourceSpec5AddNetworkResourceArg..GetNetworkResourceSpecs5  &(99 %  r%z --networkNr r=r>r@)rrAr]s` r&AddNetworkResourceArgr`s3$++)+YKr%c|^U4Sjn[RRSU"5T5RU5 g)zAdds Resource arg for subnetwork to the parser. Args: help_text: str, the help text for the flag. parser: argparse parser for the command. cv>U4SjnSn[R"SSU"5U"5[RS9$)z=Constructs and returns the Resource specification for Subnet.cL>[R"ST[RS9$)N subnetworkrT)rr compute_subnet_flagsSubnetworksCompleterrXsr&SubnetAttributeConfigVAddSubnetworkResourceArg..GetSubnetResourceSpec..SubnetAttributeConfigs%  6 6(==r%cJ[R"SS[RS9$)Nzsubnetwork-regionzfGoogle Cloud region of this subnetwork https://cloud.google.com/compute/docs/regions-zones/#locations.rT)rr r RegionCompleterr$r%r&RegionAttributeConfigVAddSubnetworkResourceArg..GetSubnetResourceSpec..RegionAttributeConfigs)  6 6"P.. r%zcompute.subnetworksrd)r+rdrr[r\)rgrkrs r&GetSubnetResourceSpec7AddSubnetworkResourceArg..GetSubnetResourceSpecs=  "(*$&99  r%z --subnetworkNr_)rrArms` r&AddSubnetworkResourceArgros38+++-yKr%crSn[RRSU"5USS9RU5 g)zAdds Resource arg for KMS key to the parser. Args: parser: argparse parser for the command. help_text: str, the help text for the flag. c tSnSnSnSn[R"SSU"5U"5U"5U"5S9$)Nc*[R"SS9$)Nzkms-key)rrr r$r%r&KmsKeyAttributeConfigRAddKmsKeyResourceArg..GetKmsKeyResourceSpec..KmsKeyAttributeConfigs  6 6I FFr%c,[R"SSS9$)Nz kms-keyringz!KMS keyring id of the {resource}.rrrsr$r%r&KmsKeyringAttributeConfigVAddKmsKeyResourceArg..GetKmsKeyResourceSpec..KmsKeyringAttributeConfigs  6 6(Kr%c,[R"SSS9$)Nz kms-locationz"Cloud location for the {resource}.rwrsr$r%r&KmsLocationAttributeConfigWAddKmsKeyResourceArg..GetKmsKeyResourceSpec..KmsLocationAttributeConfigs  6 6)Mr%c,[R"SSS9$)Nz kms-projectz$Cloud project id for the {resource}.rwrsr$r%r&KmsProjectAttributeConfigVAddKmsKeyResourceArg..GetKmsKeyResourceSpec..KmsProjectAttributeConfigs  6 6(Nr%z/cloudkms.projects.locations.keyRings.cryptoKeyskey)r+ cryptoKeysId keyRingsIdr-r,)rr0)rtrxr{r~s r&GetKmsKeyResourceSpec3AddKmsKeyResourceArg..GetKmsKeyResourceSpecsJG    9*,,..0,.  r%z --kms-keyFrKNr_)rArrs r&AddKmsKeyResourceArgrs=<++ , Kr%cB[RRU5 g)z+Adds the --async flags to the given parser.N)r ASYNC_FLAGr@rAs r& AddAsyncFlagrs//f%r%c U(aSOSnURSUS3S9nURSSSUS3S S 9 URS S US 3[SS9 URS[SS9 URSUS3S9nURSS[SS9 URSSUS3[SS9 URSUS3S9n[ SUS3US9 [ S US3US9 URS!S"S#S$S%US3S&9 g)'zConstruct args to provide a custom compute spec for notebook execution. Args: parser: argparse parser for the command. workbench_execution: bool, true if these flags are for creating a Workbench execution. zexecution environmentrGz!The machine configuration of the .helpz--machine-typeFz1The Compute Engine machine type selected for the z e2-standard-4)r8rdefaultz--accelerator-typez-The type of hardware accelerator used by the z;. If specified, --accelerator-count must also be specified.N)rchoicesrz--accelerator-countz/The number of accelerators used by the runtime.)typerz+The configuration for the data disk of the z --disk-typez The type of the persistent disk.rz--disk-size-gbzThe disk size of the zr in GB. If specified, the --disk-type must also be specified. The minimum size is 10GB and the maximum is 65536GB.drrrz"The network configuration for the zThe name of the VPC that this z is in.)rrAz%The name of the subnetwork that this z--enable-internet-access store_trueenable_internet_accessTz&Enable public internet access for the )actiondestrr) add_group add_argument_accelerator_choicesint _disk_choicesr`ro)rAworkbench_executionvm_namemachine_spec_groupdisk_spec_groupnetwork_spec_groups r&AddCustomEnvSpecFlagsrs"5) '' .wiq 9(!! >wiq I " !! :7)D@ @" "!!  <" $$ 7y B%/  -   "7),   '' /y :(0 A  1'' B  !!  # 3G9A > "r%cURSSS9nURSSSS9 URSSS S9 URS SS [S 9 URS SS[R"5SS9 g)z5Adds Resource arg for runtime template to the parser.Tz3The software configuration of the runtime template.)hiddenrz--post-startup-scriptFz?%% C     & %% ,    !& %%   ? &%%   F &%%     B & H vr%c0[USSS9 [U5 g)zEConstruct groups and arguments specific to runtime template deletion. to deleteTrCN)rErrs r&AddDeleteRuntimeTemplateFlagsrs 4Hvr%c[USSS9 g)zIConstruct groups and arguments specific to describing a runtime template. to describeTrNrErs r&AddDescribeRuntimeTemplateFlagsrs TJr%cn[US5 URR[R5 g)zEConstruct groups and arguments specific to listing runtime templates.z'for which to list all runtime templatesN)rO display_info AddUriFuncruntime_templates_utilGetRuntimeTemplateUrirs r&AddListRuntimeTemplatesFlagsrs(vHI  !7!M!MNr%cF[USSS9 [R"U5 g)zKConstruct arguments for adding an IAM policy binding to a runtime template.zto add IAM policy binding toTrN)rEr AddArgsForAddIamPolicyBindingrs r&AddFlagsToAddIamPolicyBindingrs" ,D ((0r%c[USSS9 g)zFConstruct arguments for getting the IAM policy for a runtime template.zto get IAM policy forTrNrrs r&AddGetIamPolicyFlagsrs %Tr%cF[USSS9 [R"U5 g)zFConstruct arguments for setting the IAM policy for a runtime template.zto set IAM policy forTrN)rEr AddArgForPolicyFilers r&AddSetIamPolicyBindingFlagsrs" %T v&r%cF[USSS9 [R"U5 g)zOConstruct arguments for removing an IAM policy binding from a runtime template.zto remove IAM policy fromTrN)rEr AddArgsForRemoveIamPolicyBindingrs r&AddRemoveIamPolicyBindingFlagsrs" ) ++F3r%c[USS9 [USSS9 URSSSS9 URS S S S9 URS SS S9 URSSSS9 URSS[R"5SS9 [ U5 g)z+Construct arguments for creating a runtime.zto create runtimerBzto configure the runtime withFrz --runtime-idzQThe id of the runtime to create. If not specified, a random id will be generated.rrTz*The display name of the runtime to create.rzThe descriptionz--runtime-userzUser email for the runtime owner. Runtimes can only be used by the owner. If a user is not provided, the gcloud user will be assumed to be the owner. The user cannot be a service account.rrrrN)rOrErrrrrs r&AddCreateRuntimeFlagsrsv$78 -U     7  orsB_M/(51BV]3D*G )% &R64>&R+\&L^%PDN K O 1'4&R4 >    r%