dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z d Ze jj fd Ze jj fd ZddZdZdZdZdZdZddZ ddZy )z1A library that is used to support trace commands.)absolute_import)division)unicode_literals)apis)waiter)base) properties) resourcesc@|tjjk(ryy)zXReturns the correct API version. Args: release_track: base.ReleaseTrack object v1v1beta1)r ReleaseTrackGA release_tracks ,lib/googlecloudsdk/api_lib/notebooks/util.pyApiVersionSelectorrs d''***  c@tjdt|S)z%Returns the client for the trace API. notebooks) core_apisGetClientInstancerrs r GetClientr'!  $ $[%7 %F HHrc@tjdt|S)z'Returns the messages for the trace API.r)rGetMessagesModulerrs r GetMessagesr-rrNc|s4tjjjj d}t j j|d|idS)NTrequired projectsIdznotebooks.projects.locations)params collection)r VALUEScoreprojectGetr REGISTRYParse)locationr&s rGetLocationResourcer+3sV $$,,00$0?G    ! !G $/ " 11rc|jdrR|jjj}t |j |j jSy)Nruntime) IsSpecifiedCONCEPTSr-r)r+ locationsIdr! RelativeName)argsr-s rGetParentForRuntimer3<sS i mm##))+G w22&11 33?<>B!rc|jdrR|jjj}t |j |j jSy)Ninstance)r.r/r5r)r+r0r!r1)r2r5s rGetParentForInstancer6CsS j!}}%%++-H x33'22 44@LNC"rc|jdrR|jjj}t |j |j jSy)N environment)r.r/r8r)r+r0r!r1)r2r8s rGetParentForEnvironmentr9JsS m$--++113K {66*55 77C|~F%rctjj|xs4tjj j jddS)z-Returns the resource for the current project.Trznotebooks.projects)r#)r r(r)r r$r%r&r')r&s rGetProjectResourcer;QsK    ! ! B""''//33T3B% " ''rctjjjj d}|j dr$t |j|jStjjjjrLt tjjjj d|jSy)NTrr*) r r$r%r&r'r.r+r*r1rIsExplicitlySet)r2r&s rGetParentFromArgsr>Xs    " " * * . . . =' j! t}}g 6 C C EE""++;;= ##,,00$0?  >rcXtjj|dt|S)Nz'notebooks.projects.locations.operations)r# api_version)r r(ParseRelativeNamer)namers rGetOperationResourcerCbs.    - - :$]3 . 55rct|j|}t|}|r tj|j }n tj ||j }tj|||S)a Waits for the given google.longrunning.Operation to complete. Args: operation: The operation to poll. message: String to display for default progress_tracker. service: The service to get the resource after the long running operation completes. release_track: base.ReleaseTrack object. is_delete: Bool indicating is Poller should fetch resource post operation. Raises: apitools.base.py.HttpError: if the request returns an HTTP error Returns: The created Environment resource. )rCrBrrCloudOperationPollerNoResourcesprojects_locations_operationsCloudOperationPollerWaitFor) operationmessageservicer is_delete operation_refclientpollers rWaitForOperationrPisj*'y~~}E- ] #&  3 3,,.F ( (557F  w 77r)N)F)__doc__ __future__rrrgooglecloudsdk.api_lib.utilrrrgooglecloudsdk.calliopergooglecloudsdk.corer r rrrrrr+r3r6r9r;r>rCrPrrrWs8&'9.(*)!--00H #//22H 1BCF' 5 % 8r