U {_@sdZddlmZddlmZddlmZddlZddlZddlZddlZddl m Z Gddde Z d d Z dd e dfd d ZddZGdddeZdS)zk Timeout decorator. :copyright: (c) 2012-2013 by PN. :license: MIT, see LICENSE for more details. )print_function)unicode_literals)divisionNwrapsc@s"eZdZdZdddZddZdS) TimeoutErrorz>Thrown when a timeout occurs in the `timeout` context manager. Timed OutcCs ||_dSN)value)selfr r T/opt/alt/python38/lib/python3.8/site-packages/timeout_decorator/timeout_decorator.py__init__szTimeoutError.__init__cCs t|jSr )reprr r r r r __str__"szTimeoutError.__str__N)r)__name__ __module__ __qualname____doc__rrr r r r rs rcCs|dkr|n||dS)z This function checks if a exception message is given. If there is no exception message, the default behaviour is maintained. If there is an exception message, the message is passed to the exception with the 'value' keyword. Nr ) exceptionexception_messager r r _raise_exception&srTcsfdd}|S)aAdd a timeout parameter to a function and return it. :param seconds: optional time limit in seconds or fractions of a second. If None is passed, no timeout is applied. This adds some flexibility to the usage: you can disable timing out depending on the settings. :type seconds: float :param use_signals: flag indicating whether signals should be used for timing function out or the multiprocessing When using multiprocessing, timeout granularity is limited to 10ths of a second. :type use_signals: bool :raises: TimeoutError if time limit is reached It is illegal to pass anything other than a function as the first parameter. The function is wrapped and returned to the caller. csPr.fddtfdd}|Stfdd}|SdS)NcstdSr )r)signumframe)rtimeout_exceptionr r handlerDsz*timeout..decorate..handlerc sp|d}|r,ttj}ttj|s:||Sz||WS|rjttjdttj|XdS)Ntimeoutr)popsignalSIGALRM setitimer ITIMER_REAL)argskwargsZ new_secondsold)functionrsecondsr r new_functionGs  z/timeout..decorate..new_functioncst}|||Sr )_Timeout)r#r$Ztimeout_wrapper)rr&r'rr r r(Ysr)r&r(rr'r use_signals)r&rr decorateAsztimeout..decorater )r'r+rrr,r r*r r2srcOs@z|d|||fWn"|dtdfYnXdS)alRun a function with arguments and return output via a queue. This is a helper function for the Process created in _Timeout. It runs the function with positional arguments and keyword arguments and then returns the function's output by way of a queue. If an exception gets raised, it is returned to _Timeout to be raised by the value property. TFN)putsysexc_info)queuer&r#r$r r r _targetbsr2c@s@eZdZdZddZddZddZedd Zed d Z d S) r)aWrap a function and add a timeout (limit) attribute to it. Instances of this class are automatically generated by the add_timeout function defined above. Wrapping a function allows asynchronous calls to be made and termination of execution after a timeout has passed. cCsJ||_||_||_||_|j|_|j|_t|_t |_ t |_ dS)z4Initialize instance in preparation for being called.N) _Timeout__limit_Timeout__function_Timeout__timeout_exception_Timeout__exception_messagerrtime_Timeout__timeoutmultiprocessingProcess_Timeout__processQueue_Timeout__queue)r r&rrlimitr r r rys  z_Timeout.__init__cOs|d|j|_td|_|j|jf|}tjt||d|_d|j_ |j |jdk rj|jt |_ |j s|t dqj|jS)aExecute the embedded function object asynchronously. The function given to the constructor is transparently called and requires that "ready" be intermittently polled. If and when it is True, the "value" property may then be checked for returned data. rr-)targetr#r$TNg{Gz?)rr3r9r<r=r4r:r2r;daemonstartr7r8readysleepr )r r#r$r r r __call__s    z_Timeout.__call__cCs&|jr|jt|j|jdS)z:Terminate any possible execution of the embedded function.N)r;is_alive terminaterr5r6rr r r cancels  z_Timeout.cancelcCs2|jr|jtkr||jo0|j S)z9Read-only property indicating status of "value" property.)r3r8r7rGr=fullemptyrr r r rBsz_Timeout.readycCs(|jdkr$|j\}}|r |S|dS)z:Read-only property containing data returned from function.TN)rBr=get)r flagloadr r r r s  z_Timeout.valueN) rrrrrrDrGpropertyrBr r r r r r)ps  r))r __future__rrrr/r7r9r functoolsrAssertionErrorrrrr2objectr)r r r r s      0