dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z d Ze jj fd Ze jj fd ZddZdZdZdZddZ ddZy )z1A library that is used to support trace commands.)absolute_import)division)unicode_literals)apis)waiter)base) properties) resourcesc~y)zXReturns the correct API version. Args: release_track: base.ReleaseTrack object v2 release_tracks ,lib/googlecloudsdk/api_lib/workbench/util.pyApiVersionSelectorrs  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)2sV $$,,00$0?G    ! !G $/ " 11rc|jdrR|jjj}t |j |j jSy)Ninstance) IsSpecifiedCONCEPTSr+r'r) locationsIdr RelativeName)argsr+s rGetParentForInstancer1;sS j!}}%%++-H x33'22 44@LNC"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 rGetProjectResourcer3BsK    ! ! B""''//33T3B% " ''rctjjjj d}|j dr$t |j|jStjjjjrLt tjjjj d|jStd)a Gets the parent resource for the command. Args: args: The args object containing the command arguments. Returns: The resource name of the parent resource. Raises: ValueError: If neither `--location` nor the `location` property is specified. Trr(z9One of --location or location property must be specified.) r r"r#r$r%r,r)r(r/rIsExplicitlySet ValueError)r0r$s rGetParentFromArgsr7Is    " " * * . . . =' j! t}}g 6 C C EE""++;;= ##,,00$0? ln C rcXtjj|dt|S)Nz'notebooks.projects.locations.operations)r! api_version)r r&ParseRelativeNamer)namers rGetOperationResourcer<cs.    - - :$]3 . rct|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. )r<r;rrCloudOperationPollerNoResourcesprojects_locations_operationsCloudOperationPollerWaitFor) operationmessageservicer is_delete operation_refclientpollers rWaitForOperationrIksj&'y~~}E- ] #&  3 3,,.F ( (557F  w 77r)N)F)__doc__ __future__rrrgooglecloudsdk.api_lib.utilrrrgooglecloudsdk.calliopergooglecloudsdk.corer r r ReleaseTrackGArrr)r1r3r7r<rIr rrrQsz8&'9.(*)!--00H #//22H 1C'4;@8r