8j#ddlmZddlZddlZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z ddlmZe rddlZdd lmZdd lmZdd lmZmZGd d Ze dGddeZe dGddeZy)) annotationsN)Task) timedelta)Thread) TYPE_CHECKINGSelf) docs_group) to_seconds) TracebackType)Run) RunClientRunClientAsyncc<eZdZdZdZdZeddd dZd dZy ) StatusMessageWatcherBasez=Base class for polling and logging Actor run status messages.Fseconds check_periodcf|jrd|_||_t||_d|_y)NT)_force_propagate propagate _to_loggerr _check_period_last_status_message)self to_loggerrs [/var/www/html/GAP/venv/lib/python3.12/site-packages/apify_client/_status_message_watcher.py__init__z!StatusMessageWatcherBase.__init__s/  "&I #' 5$&!c|j|jxsd}|jxsd}d|d|}||jk7r"||_|jj ||j xsd Sy)aGet relevant run data, log them if changed and return `True` if more data is expected. Args: run_data: The Run model that contains the run data. Returns: `True` if more data is expected, `False` otherwise. zUnknown statusrzStatus: z , Message: FT)statusstatus_messagerrinfois_status_message_terminal)rrun_datar$r%new_status_messages r _log_run_dataz&StatusMessageWatcherBase._log_run_data&s}  __8(8F%44:N#+F8;~>N!O !T%>%>>,>)$$%78 ;;DuE Er"N)rlogging.LoggerrrreturnNone)r(z Run | Noner,bool) __name__ __module__ __qualname____doc__r_final_sleep_time_srr!r*r"r rrs&GOXabOc'r"rOtherc~eZdZdZedd d fdZd dZd dZddZ dd Z d d Z xZ S)StatusMessageWatcherAsynca/Polls and logs Actor run status messages in an asyncio task. The status message and status of the Actor run are polled at a fixed interval and forwarded to the provided logger whenever they change. There is no guarantee that every intermediate status message will be captured, especially when messages change rapidly. Can be used as an async context manager, which automatically starts and cancels the polling task. Alternatively, call `start` and `stop` manually. Obtain an instance via `RunClientAsync.get_status_message_watcher`. rrcDt|||||_d|_y)a.Initialize `StatusMessageWatcherAsync`. Args: run_client: The run client used to poll the Actor run status and status message. to_logger: The logger to which the status messages will be forwarded. check_period: How often to poll the status message. rrN)superr! _run_client _logging_taskr run_clientrr __class__s r r!z"StatusMessageWatcherAsync.__init__Hs' 9<H%*.r"c|jr%|jjs tdtj|j |_|jS)zwStart the polling task. The caller is responsible for cleanup by calling the `stop` method when done. zLogging task already active)r=done RuntimeErrorasyncio create_task_log_changed_status_messagers r startzStatusMessageWatcherAsync.startVsR   d&8&8&=&=&?<= =$001Q1Q1ST!!!r"cK|js td|jj |jd{d|_y7 #tj$rYwxYw#d|_wxYww)zStop the logging task.zLogging task is not activeN)r=rCcancelrDCancelledErrorrGs r stopzStatusMessageWatcherAsync.stop`sr!!;< < !!# &$$ $ $"&D  %%%   "&D sF2A8AAA A8AA)&A,(A))A,, A55A8c.K|j|Sw)z\Start the logging task within the context. Exiting the context will cancel the logging task.rHrGs r __aenter__z$StatusMessageWatcherAsync.__aenter__ms  scK|'tj|jd{|jd{y77w)zCancel the logging task.N)rDsleepr3rLrexc_typeexc_valexc_tbs r __aexit__z#StatusMessageWatcherAsync.__aexit__rs>  -- 8 89 9 9iik :s %A AA AA A cK |jjd{}|j|sytj|j d{\7>7#t $rk}|jjj|r|jjdn |jjdYd}~yYd}~yd}~wwxYwwNzAStatus message redirection stopped: the status request timed out.z;Status message redirection stopped due to unexpected error:) r<getr*rDrQr Exception _http_clientis_timeout_errorrwarning exception)rr(excs r rFz5StatusMessageWatcherAsync._log_changed_status_messagezs i!%!1!1!5!5!77))(3mmD$6$6777 78 i,,==cB''(kl))*ghhm isOCA%A!A%C"A%A#A%#A%% C.AC CCC)r?rrr+rrr,r-)r,rr,r-r,rrSztype[BaseException] | NonerTzBaseException | NonerUzTracebackType | Noner,r-) r/r0r1r2rr!rHrLrOrVrF __classcell__r@s@r r7r7<stcltubv /+ /8F /V_ /  /" & 2=Q[o  ir"r7c~eZdZdZedd d fdZd dZd dZddZ dd Z d d Z xZ S)StatusMessageWatchera'Polls and logs Actor run status messages in a background thread. The status message and status of the Actor run are polled at a fixed interval and forwarded to the provided logger whenever they change. There is no guarantee that every intermediate status message will be captured, especially when messages change rapidly. Can be used as a context manager, which automatically starts and stops the polling thread. Alternatively, call `start` and `stop` manually. Obtain an instance via `RunClient.get_status_message_watcher`. r8rrcRt|||||_d|_d|_y)a)Initialize `StatusMessageWatcher`. Args: run_client: The run client used to poll the Actor run status and status message. to_logger: The logger to which the status messages will be forwarded. check_period: How often to poll the status message. r:NF)r;r!r<_logging_thread _stop_loggingr>s r r!zStatusMessageWatcher.__init__s/ 9<H%.2"r"c |jr%|jjr tdd|_t j |j d|_|jj|jS)zyStart the polling thread. The caller is responsible for cleanup by calling the `stop` method when done. zLogging thread already activeFT)targetdaemon)rhis_aliverCri threadingrrFrHrGs r rHzStatusMessageWatcher.startsl   D$8$8$A$A$C>? ?"(//t7W7W`de ""$###r"c|js tdd|_|jjd|_d|_y)zDSignal the logging thread to stop logging and wait for it to finish.zLogging thread is not activeTNF)rhrCrijoinrGs r rLzStatusMessageWatcher.stopsA##=> >! !!##"r"c&|j|S)z^Start the logging thread within the context. Exiting the context will stop the logging thread.rNrGs r __enter__zStatusMessageWatcher.__enter__s  r"cf|tj|j|jy)zStop the logging thread.N)timerQr3rLrRs r __exit__zStatusMessageWatcher.__exit__s%   JJt// 0 r"c |j|jjsy|jryt j |j W#t$rk}|jjj|r|jjdn |jjdYd}~yYd}~yd}~wwxYwrX) r*r<rYrirtrQrrZr[r\rr]r^)rr_s r rFz0StatusMessageWatcher._log_changed_status_messages i))$*:*:*>*>*@A%% 4--.   i,,==cB''(kl))*ghhm is"*A A A C#AC  C)r?r rr+rrr,r-)r,rr`rarb) r/r0r1r2rr!rHrLrrrurFrcrds@r rfrfst^gop]q #& #3A #QZ #  # $# 2=Q[o ir"rf) __future__rrDrnrtrdatetimerrtypingrrapify_client._docsr apify_client._utils.timer loggingtypesr apify_client._modelsr apify_client._resource_clientsr rrr7rfr4r"r rs" &)/#(H##L GJi 8JiJiZ GKi3KiKir"