.SrSSKJr SSKJr SSKJr SSKrSSKJr SSKJ r SSK J r SS K J r "S S \ R5rg) zAbstract waiter utility for api_lib.util.waiter.py. Abstract waiter utility class for api_lib.util.waiter.WaitFor. This class is the base class for poller that need to stream output and poll on dataproc's operation. )absolute_import)division)unicode_literalsN)storage_helpers)waiter)log) console_attrc\rSrSrSrSr\RS5r\RS5r \RS5r \RS5r Sr S r S rS rS rg )AbstractOperationStreamerPoller$aBase abstract poller class for dataproc operation. Base abstract poller class for dataproc operation. The class is designed to stream remote output from GCS. Pass TrackerUpdateFunction to waiter.WaitFor's tracker_update_func parameter to stream remote output. c,SUlSUlXlg)zTPoller for batch workload. Args: dataproc: A api_lib.dataproc.Dataproc instance. N)saved_stream_uridriver_log_streamerdataproc)selfrs Plib/googlecloudsdk/api_lib/dataproc/poller/abstract_operation_streamer_poller.py__init__(AbstractOperationStreamerPoller.__init__.s !D#DMcg)a\Determines if the poll result is done. Determines if the poll result is done. This is a null implementation that simply returns False. Sub class should override this and provide concrete checking logic. Overrides. Args: poll_result: Poll result returned from Poll function. Returns: True if the remote resource is done, or False otherwise. Fr poll_results rIsDone&AbstractOperationStreamerPoller.IsDone8s rcg)adFetches remote resource. Fetches remote resource. This is a null implementation that simply returns None. Sub class should override this and provide concrete fetching logic. Overrides. Args: ref: Resource reference. The same argument passed to waiter.WaitFor. Returns: None. Sub class should override this and return the actual fetched resource. Nr)rrefs rPoll$AbstractOperationStreamerPoller.PollJ rcg)aGets output uri from poll result. Gets output uri from poll result. This is a null implementation that returns None. Sub class should override this and return actual output uri for output streamer, or returns None if something goes wrong and there is no output uri in the poll result. Args: poll_result: Poll result returned by Poll. Returns: None. Sub class should override this and returns actual output uri, or None when something goes wrong. Nrrs r _GetOutputUri-AbstractOperationStreamerPoller._GetOutputUri\r rcg)aaReturns operation result to caller. This function is called after GetResult streams remote output. This is a null implementation that simply returns None. Sub class should override this and provide actual _GetResult logic. Args: poll_result: Poll result returned from Poll. Returns: None. Sub class should override this and return actual result. Nrrs r _GetResult*AbstractOperationStreamerPoller._GetResultns rcJURSUS5 URU5$)zReturns result for remote resource. This function first stream remote output to user, then returns operation result by calling _GetResult. Overrides. Args: poll_result: Poll result returned by Poll. Returns: Wahtever returned from _GetResult. N)TrackerUpdateFunctionr%rs r GetResult)AbstractOperationStreamerPoller.GetResult~s%" t[$7 ??; ''rcFURU5 UR5 g)aCustom tracker function which gets called after every tick. This gets called whenever progress tracker gets a tick. However we want to stream remote output to users instead of showing a progress tracker. Args: tracker: Progress tracker instance. Not being used. poll_result: Result from Poll function. status: Status argument that is supposed to pass to the progress tracker instance. Not being used here as well. N)_CheckStreamer _StreamOutput)rtrackerrstatuss rr(5AbstractOperationStreamerPoller.TrackerUpdateFunctions  $rcUR(aFURR(a*URR[R5 ggg)N)ropenReadIntoWritablererr)rs rr--AbstractOperationStreamerPoller._StreamOutputs: D$<$<$A$A //8%BrcHSnURU5nU(dgUR(aBURU:wa2SUlSUlU"5 [R"S5 U"5 UR(d"X0l[ R "U5Ulgg)aChecks if need to init a new output streamer. Checks if need to init a new output streamer. Remote may fail; switch to new output uri. Invalidate the streamer instance and init a new one if necessary. Args: poll_result: Poll result returned from Poll. c[R"5n[RR SUR 5S-5 g)N=r)r GetConsoleAttrrr4Print GetTermSize)attrs r_PrintEqualsLineAccrossScreenUAbstractOperationStreamerPoller._CheckStreamer.._PrintEqualsLineAccrossScreens3  ( ( *d ggmmC$**,Q//0rNz/Attempt failed. Streaming new attempt's output.)r"rrrwarningrStorageObjectSeriesStream)rrr=uris rr,.AbstractOperationStreamerPoller._CheckStreamers1   [ )C   !6!6#!=!%d"d#% kkCD#%  # #!!0!J!J3!Od $r)rrrN)__name__ __module__ __qualname____firstlineno____doc__rabcabstractmethodrrr"r%r)r(r-r,__static_attributes__rrrr r $s"""  (( 9"Prr )rG __future__rrrrHgooglecloudsdk.api_lib.dataprocrgooglecloudsdk.api_lib.utilrgooglecloudsdk.corergooglecloudsdk.core.consoler OperationPollerr rrrrQs: '' ;.#4cPf&<&<cPr