tdZddlmZddlmZdZdZdZdZdZd Z d Z d Z d Z d Z dZdZdZdZy)z%colab-enterprise runtimes api helper.) properties) resourcescj|jjj}|jS)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+ ==   % % '&    cd|vr" tjj|dStjj|dS#tj$rY6wxYw)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   s AAAcxtjj|jd}|j S)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 |jdr[|jj}||jj Dcgc]\}}|j ||c}}Sycc}}w)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 U  - -#U - C1     sA. c|jdr |jStjjj j S)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[s? n%       ! ! ) ) - - //r cf|jjjjS)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 |j|jt||j|jt ||S)aCreates 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:xsG  = = ??4T:##"" tX . > r ct|}t||}|j|j||j|j j jj|S)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 $& x 0'  J J?G@A@A! OO"&--"@"@"F"F"H"U"U"W@A@  K r c8|jt|S)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 c8|jt|S)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 c8|jt|S)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 cV|j|jt|S)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 cV|j|jt|S)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 cV|j|jt|S)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