U
vi @ s d dl Z d dlZd dlmZ d dlmZmZmZ d dlm Z mZ d dl
m Z m Z m
Z
ddlmZ edZed ZG d
d d eZdS ) N)Future)ListOptionalTypeVar)abctyping)CompositeDisposable
DisposableSingleAssignmentDisposable )AsyncIOScheduler_TStateZRxc @ s e Zd ZdZdeje ee ej dddZ
dej eje ee ej dddZ dej
eje ee ej dd d
Zed d d
ZdS )AsyncIOThreadSafeSchedulerzA scheduler that schedules work via the asyncio mainloop. This is a
subclass of AsyncIOScheduler which uses the threadsafe asyncio methods.
N)actionstatereturnc sL t dd fdd }j|ddfdd }tt|S )a! Schedules an action to be executed.
Args:
action: Action to be executed.
state: [Optional] state to be given to the action function.
Returns:
The disposable object used to cancel the scheduled action
(best effort).
Nr c s j d_d S Nr Z
invoke_actionZ
disposable r sadselfr r i/opt/alt/python38/lib/python3.8/site-packages/reactivex/scheduler/eventloop/asynciothreadsafescheduler.pyinterval( s z5AsyncIOThreadSafeScheduler.schedule..intervalc sF r d S t d d fdd } j| d S )Nr c s d d S Nr )cancel
set_resultr )futurehandler r
cancel_handle4 s zKAsyncIOThreadSafeScheduler.schedule..dispose..cancel_handle) _on_self_loop_or_not_runningr r _loopcall_soon_threadsaferesultr! r r )r r dispose- s z4AsyncIOThreadSafeScheduler.schedule..dispose)r
r# r$ r r )r r r r r( r )r r r r r r schedule s
z#AsyncIOThreadSafeScheduler.schedule)duetimer r r c s |dkr j dS t dd fdd g ddfdd }j| ddfd d
}tt|S ) au Schedules an action to be executed after duetime.
Args:
duetime: Relative time after which to execute the action.
action: Action to be executed.
state: [Optional] state to be given to the action function.
Returns:
The disposable object used to cancel the scheduled action
(best effort).
r r Nr c s j d_d S r r r r r r r T s z>AsyncIOThreadSafeScheduler.schedule_relative..intervalc s j d S N)appendr#
call_laterr )r r secondsr r r stage2[ s z.stage2c sV d dfdd r$ d S t d d fdd } j| d S )Nr c s6 z W n tk
r0 Y nX d S r+ )popr Exceptionr )r r r do_cancel_handlesa s
zXAsyncIOThreadSafeScheduler.schedule_relative..dispose..do_cancel_handlesc s d d S r )r r r2 r r r r! n s zTAsyncIOThreadSafeScheduler.schedule_relative..dispose..cancel_handle)r" r r# r$ r% r& r' r3 r r( ` s z=AsyncIOThreadSafeScheduler.schedule_relative..dispose)Z
to_secondsr) r
r, r# r$ r r )r r* r r r/ r( r )r r r r r. r r r schedule_relative= s
z,AsyncIOThreadSafeScheduler.schedule_relativec C s | |}| j|| j ||dS )ao Schedules an action to be executed at duetime.
Args:
duetime: Absolute time at which to execute the action.
action: Action to be executed.
state: [Optional] state to be given to the action function.
Returns:
The disposable object used to cancel the scheduled action
(best effort).
r ) to_datetimer4 now)r r* r r r r r schedule_absolutew s
z,AsyncIOThreadSafeScheduler.schedule_absoluter c C s> | j sdS d}z t }W n tk
r2 Y nX | j |kS )z
Returns True if either self._loop is not running, or we're currently
executing on self._loop. In both cases, waiting for a future to be
resolved on the loop would result in a deadlock.
TN)r#
is_runningasyncioget_event_loop RuntimeError)r current_loopr r r r" s
z7AsyncIOThreadSafeScheduler._on_self_loop_or_not_running)N)N)N)__name__
__module__ __qualname____doc__r ZScheduledActionr
r r ZDisposableBaser) Z RelativeTimer4 Z AbsoluteTimer7 boolr" r r r r r s, ( > r )r9 loggingconcurrent.futuresr r r r r Z reactivexr Zreactivex.disposabler r r
Zasyncioschedulerr r
getLoggerlogr r r r r s