:SrSSKJr SSKJr SSKJr SSKJr Sr Sr Sr S r S r S rS rS rSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrg)z.colab-enterprise runtime-templates api helper.) exceptions)iam_util) arg_utils) resourcescjURRR5nUR5$)zGet the parent Location resource name for the runtime template. Args: args: Argparse object from Command.Run Returns: The resource name in the form projects/{project}/locations/{location}. )CONCEPTSregionParse RelativeName)argsr s @lib/googlecloudsdk/api_lib/colab_enterprise/runtime_templates.pyGetParentForRuntimeTemplaters+ ==   % % '&    cSU;a[RRUSS9$[RRUSS9$![Ra N4f=f)zParse operation relative resource name to the operation reference object. Args: operation_name: The operation resource name Returns: The operation reference object z/notebookRuntimeTemplates/zAaiplatform.projects.locations.notebookRuntimeTemplates.operations) collectionz(aiplatform.projects.locations.operations)rREGISTRYParseRelativeName WrongResourceCollectionException)operation_names r ParseRuntimeTemplateOperationr%su"^3    1 1 X2    - -!K .   5 5   sAAAcr[RRURSS9nUR 5$)z1Transform a runtime template resource into a URL.z6aiplatform.projects.locations.notebookRuntimeTemplates) relative_namer)rrrnameSelfLink)resourceruntime_templates r GetRuntimeTemplateUrir;s9''99MMI:  " " $$rcURnSnURS5(aN[R"SURSS9R [R "UR55nU"URUURS9$)zConstructs the machine spec from command line args. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the aiplatform API. Returns: Machine spec config for the runtime template. Naccelerator_typezaccelerator-typec,USs=:g=(a U;$s $)N UNSPECIFIEDxs r (GetMachineSpecFromArgs..Tsm!.>{.K.K  . .B  ,DNN;I &'8'8 rcfURnSnSnURS5(a2URRR 5R 5nURS5(a2URR R 5R 5nU"UUURS9$)zConstructs the network spec from command line args. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the aiplatform API. Returns: Network spec config for the runtime template. Nnetwork subnetwork)rJrKenableInternetAccess)'GoogleCloudAiplatformV1beta1NetworkSpecr/rrJr r rKenable_internet_access)r r6network_spec_confignetwork_relative_namesubnetwork_relative_names r GetNetworkSpecFromArgsrRs!HH! i   MM11779FFH l###}}77==?LLN #)66 rc URS5(aVURRnU"URR 5VVs/sHup4UR X4S9PM snnS9$gs snnf)zConstructs the labels from command line args. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the aiplatform API. Returns: Labels for the runtime template. labels)keyvalue)additionalPropertiesN)r/3GoogleCloudAiplatformV1beta1NotebookRuntimeTemplate LabelsValuerTitemsAdditionalProperty)r r6labels_messagerUrVs r GetLabelsFromArgsr]s| hDDPP #kk//1 1   - -# - C1     s A. cURnURS5(a-URnUS:XaU"SS9$[U5S-nU"SUS9$g) zConstructs the idle shutdown config from command line args. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the aiplatform API. Returns: Idle shutdown config for the runtime template. idle_shutdown_timeoutrT)idleShutdownDisabledsF)r` idleTimeoutN)6GoogleCloudAiplatformV1beta1NotebookIdleShutdownConfigr/r_str)r r6idle_shutdown_configduration_secondsduration_serializeds r GetIdleShutdownConfigFromArgsrhsnEE -..111 !t <<./#5 "0C  rc>URUR(+S9$)zConstructs the euc config from command line args. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the aiplatform API. Returns: Euc config for the runtime template. ) eucDisabled)-GoogleCloudAiplatformV1beta1NotebookEucConfig enable_eucr r6s r GetEucConfigFromArgsrns'  ? ?' @ rc4URURS9$)zConstructs the shielded vm config from command line args. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the aiplatform API. Returns: Shielded vm config for the runtime template. )enableSecureBoot),GoogleCloudAiplatformV1beta1ShieldedVmConfigenable_secure_bootrms r GetShieldedVmConfigFromArgsrss%  > >.. ? rcJURS5(a UR$/$)N network_tags)r/rur s r GetNetworkTagsFromArgsrws#"..~>>  FBFrc URnURnURnSnURS5(aN[R "SUR SS9R[R"UR55n/nURS5(a,URVs/sHor"XpRUS9PM nnU"UU"UURURS9S 9$s snf) z6Constructs the software config from command line args.Npost_startup_script_behaviorzpost-startup-script-behaviorc US:g$)N(POST_STARTUP_SCRIPT_BEHAVIOR_UNSPECIFIEDr"r#s r r%+GetSoftwareConfigFromArgs..s  5"6rr' set_env_vars)rrV)postStartupScriptBehaviorpostStartupScriptpostStartupScriptUrl)envpostStartupScriptConfig) "GoogleCloudAiplatformV1beta1EnvVar3GoogleCloudAiplatformV1beta1PostStartupScriptConfig2GoogleCloudAiplatformV1beta1NotebookSoftwareConfigr/rr0(PostStartupScriptBehaviorValueValuesEnumr2r3ryr}post_startup_scriptpost_startup_script_url)r r6env_varpost_startup_script_configsoftware_config!post_startup_script_behavior_enumenv_varsks r GetSoftwareConfigFromArgsrs  7 7'BBOO/&*# 455(1(B(B//XX6) ""4#D#DE &( n%%=A=N=N=NQ//23=N   8$E 44#;; s1C0cURnURS5(a6U"URRR 5R 5S9$g)zConstructs the encryption spec from the kms key resource arg. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the aiplatform API. Returns: Encryption spec for the notebook resource. kms_key) kmsKeyNameN)*GoogleCloudAiplatformV1beta1EncryptionSpecr/rrr r )r r6encryption_specs r CreateEncryptionSpecConfigrsQGG/ i  ==((..0==?  rcfURRR5R5$)zGet the resource name for the runtime template. Args: args: Argparse object from Command.Run Returns: The resource name in the form projects/{project}/locations/{location}/notebookRuntimeTemplates/{runtime_template_id}. )rrr r rvs r GetRuntimeTemplateResourceNamer&s&  ' ' - - / < < >>rc(URURURUR[ X5[ X5[ X5[X5[X5[X5[X5[U5[X5[X5S9 $)zCreates the RuntimeTemplate message for the create request. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the RuntimeTemplate message. ) r displayName description machineSpecdataPersistentDiskSpec networkSpecrTidleShutdownConfig eucConfigshieldedVmConfig networkTagsencryptionSpecsoftwareConfig)rXruntime_template_id display_namerr9rHrRr]rhrnrsrwrrrms r CreateRuntimeTemplater3s  E E  # ###""(8:4J(8 t .6tF$T424B(./?.t> F rcd[U5n[X5nURUURUS9$)zBuilds a CreateNotebookRuntimeTemplateRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the CreateNotebookRuntimeTemplateRequest message. )3googleCloudAiplatformV1beta1NotebookRuntimeTemplatenotebookRuntimeTemplateIdparent)rr@AiplatformProjectsLocationsNotebookRuntimeTemplatesCreateRequestr)r r6rrs r "CreateRuntimeTemplateCreateRequestrNsA 't ,&*4:OO>N$($<$<Prc2UR[U5S9$)zBuilds a DeleteNotebookRuntimeTemplateRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the DeleteNotebookRuntimeTemplateRequest message. r)@AiplatformProjectsLocationsNotebookRuntimeTemplatesDeleteRequestrrms r "CreateRuntimeTemplateDeleteRequestrds%OO-d3Prc2UR[U5S9$)zBuilds a GetNotebookRuntimeTemplateRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the GetNotebookRuntimeTemplateRequest message. r)=AiplatformProjectsLocationsNotebookRuntimeTemplatesGetRequestrrms r CreateRuntimeTemplateGetRequestrvs%LL-d3Mrc2UR[U5S9$)zBuilds a ListNotebookRuntimeTemplatesRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the ListNotebookRuntimeTemplatesRequest message. )r)>AiplatformProjectsLocationsNotebookRuntimeTemplatesListRequestrrms r CreateRuntimeTemplateListRequestrs%MM,T2Nrc2UR[U5S9$)zBuilds a RuntimeTemplatesGetIamPolicyRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the RuntimeTemplatesGetIamPolicyRequest message. )r)FAiplatformProjectsLocationsNotebookRuntimeTemplatesGetIamPolicyRequestrrms r (CreateRuntimeTemplateGetIamPolicyRequestrs%  X X-d3 Y 55rcRURUS9nURU[U5S9$)aBuilds a RuntimeTemplatesSetIamPolicyRequest message. Args: iam_policy: The IAM policy to set. args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the RuntimeTemplatesSetIamPolicyRequest message. )policy)googleIamV1SetIamPolicyRequestr)GoogleIamV1SetIamPolicyRequestFAiplatformProjectsLocationsNotebookRuntimeTemplatesSetIamPolicyRequestr) iam_policyr r6google_iam_set_policy_requests r (CreateRuntimeTemplateSetIamPolicyRequestrsB#+"I"I #J#  X X%B-d3 Y rcp[R"URUR5n[ X U5$)aReads policy file from args to build a RuntimeTemplatesSetIamPolicyRequest. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the RuntimeTemplatesSetIamPolicyRequest message. )rParsePolicyFile policy_fileGoogleIamV1Policyr)r r6rs r 0CreateRuntimeTemplateSetIamPolicyRequestFromFilers2  # # 22 4& 1& IIrN) __doc__googlecloudsdk.calliopergooglecloudsdk.command_lib.iamr$googlecloudsdk.command_lib.util.apisrgooglecloudsdk.corerrrrr9r@rHrRr]rhrnrsrwrrrrrrrrrrrr"rr rs5.3:) ,%2 -82.2  G@& ?6,$$$ 5& Jr