tSrSSKJr SSKJr SrSrSrSrSrS r S r S r S r S r SrSrSrSrg)z%colab-enterprise runtimes api helper.) properties) resourcescjURRR5nUR5$)zGet the parent Location resource name for the runtime. Args: args: Argparse object from Command.Run Returns: The resource name in the form projects/{project}/locations/{location}. )CONCEPTSregionParse RelativeName)argsrs 7lib/googlecloudsdk/api_lib/colab_enterprise/runtimes.pyGetParentForRuntimer s+ ==   % % '&    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/notebookRuntimes/z9aiplatform.projects.locations.notebookRuntimes.operations) collectionz(aiplatform.projects.locations.operations)rREGISTRYParseRelativeName WrongResourceCollectionException)operation_names r ParseRuntimeOperationr#su^+    1 1 I2    - -!K .   5 5   sAAAcr[RRURSS9nUR 5$)z(Transform a runtime resource into a URL.z.aiplatform.projects.locations.notebookRuntimes) relative_namer)rrrnameSelfLink)resourceruntimes r GetRuntimeUrir;s8    0 0MMA 1 '    r c 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. labels)keyvalue)additionalPropertiesN) IsSpecified+GoogleCloudAiplatformV1beta1NotebookRuntime LabelsValueritemsAdditionalProperty)r messageslabels_messagerrs r GetLabelsFromArgsr(Ds| h<<HH #kk//1 1   - -# - C1     s A. cURS5(a UR$[RRR R 5$)zGets runtime user from command line args if provided; else default to caller. Args: args: Argparse object from Command.Run Returns: Runtime user of the runtime. runtime_user)r!r*rVALUEScoreaccountGetr s r "GetRuntimeUserFromArgsOrPropertiesr0[sB n%%       ! ! ) ) - - //r cfURRR5R5$)zGet the resource name for the notebook runtime. Args: args: Argparse object from Command.Run Returns: The resource name in the form projects/{project}/locations/{location}/notebookRuntimes/{runtime_id}. )rrrr r/s r GetRuntimeResourceNamer2ks&    $ $ & 3 3 55r c URUR[U5URUR[ X5S9$)zCreates the NotebookRuntime message for the AssignNotebookRuntime request. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the NotebookRuntime message. )r runtimeUser displayName descriptionr)r" runtime_idr0 display_namer6r(r r&s r CreateRuntimeMessager:xsE  = = ??4T:##"" t . > r c[U5n[X5nURURUURUR R R5R5S9US9$)zBuilds a AssignNotebookRuntimeRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the AssignNotebookRuntimeRequest message. )notebookRuntimenotebookRuntimeIdnotebookRuntimeTemplate)8googleCloudAiplatformV1beta1AssignNotebookRuntimeRequestparent) r r:8AiplatformProjectsLocationsNotebookRuntimesAssignRequest8GoogleCloudAiplatformV1beta1AssignNotebookRuntimeRequestr7rruntime_templaterr )r r&r@rs r !CreateRuntimeAssignRequestMessagerDs} t $&  0'  J J?G@A@A! OO"&--"@"@"F"F"H"U"U"W@A@  K r c2UR[U5S9$)zBuilds a GetNotebookRuntimeRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the GetNotebookRuntimeRequest message. r)5AiplatformProjectsLocationsNotebookRuntimesGetRequestr2r9s r CreateRuntimeGetRequestMessagerHs%  G G !$ ' H r c2UR[U5S9$)zBuilds a ListNotebookRuntimeRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the ListNotebookRuntimeRequest message. )r@)6AiplatformProjectsLocationsNotebookRuntimesListRequestr r9s r CreateRuntimeListRequestMessagerKs%  H H  & I r c2UR[U5S9$)zBuilds a DeleteNotebookRuntimeRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the DeleteNotebookRuntimeRequest message. rF)8AiplatformProjectsLocationsNotebookRuntimesDeleteRequestr2r9s r !CreateRuntimeDeleteRequestMessagerNs%  J J !$ ' K r cPURUR5[U5S9$)zBuilds a UpgradeNotebookRuntimeRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the UpgradeNotebookRuntimeRequest message. )9googleCloudAiplatformV1beta1UpgradeNotebookRuntimeRequestr)9AiplatformProjectsLocationsNotebookRuntimesUpgradeRequest9GoogleCloudAiplatformV1beta1UpgradeNotebookRuntimeRequestr2r9s r "CreateRuntimeUpgradeRequestMessagerSs=  K K@HACACAE !$ ' L r cPURUR5[U5S9$)zBuilds a StartNotebookRuntimeRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the StartNotebookRuntimeRequest message. )7googleCloudAiplatformV1beta1StartNotebookRuntimeRequestr)7AiplatformProjectsLocationsNotebookRuntimesStartRequest7GoogleCloudAiplatformV1beta1StartNotebookRuntimeRequestr2r9s r CreateRuntimeStartRequestMessagerXs4  I I>F>~>~?A !$ ' J r cPURUR5[U5S9$)zBuilds a StopNotebookRuntimeRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the StopNotebookRuntimeRequest message. )6googleCloudAiplatformV1beta1StopNotebookRuntimeRequestr)6AiplatformProjectsLocationsNotebookRuntimesStopRequest6GoogleCloudAiplatformV1beta1StopNotebookRuntimeRequestr2r9s r CreateRuntimeStopRequestMessager]s1  H H=E=|=|=~ !$ ' I r N)__doc__googlecloudsdk.corerrr rrr(r0r2r:rDrHrKrNrSrXr]r r rasV,*) 0. 0 6&0    ""r