U
vi
@ s d dl Z d dlZd dlmZmZ d dlmZmZmZ d dlm Z mZ d dl
m Z d dl m
Z
mZ ddlmZ dd lmZ e d
Zd Zed ZG d
d deZdS ) N)datetime timedelta)AnyOptionalTypeVar)abctyping) AbsoluteTime)ArgumentOutOfRangeException
PriorityQueue )PeriodicScheduler)
ScheduledItemZRxd _TStatec s, e Zd ZdZd#edd fdd
Zejddd Zeed
Z ee
dd d Z d$ej e
ee
ejd
ddZd%ejej e
ee
ejdddZd&ejej e
ee
ejdddZedddZddddZejddddZejddd d Zejddd d Zeejejejd d!d"Z Z S )'VirtualTimeSchedulerzcVirtual Scheduler. This scheduler should work with either
datetime/timespan or ticks as int/intr N)
initial_clockreturnc s, t || _d| _t | _t | _dS )zCreates a new virtual time scheduler with the specified
initial clock value.
Args:
initial_clock: Initial value for the clock.
FN) super__init___clock _is_enabled threadingLock_lockr _queue)selfr __class__ Y/opt/alt/python38/lib/python3.8/site-packages/reactivex/scheduler/virtualtimescheduler.pyr s
z VirtualTimeScheduler.__init__)r c
C s$ | j | jW 5 Q R S Q R X d S )N)r r r r r r
_get_clock& s z VirtualTimeScheduler._get_clock)fgetc C s | | jS )zRepresents a notion of time for this scheduler. Tasks being
scheduled on a scheduler will adhere to the time denoted by this
property.
Returns:
The scheduler's current time, as a datetime instance.
) to_datetimer r! r r r now, s
zVirtualTimeScheduler.now)actionstater c C s | j | j||dS )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).
r' )schedule_absoluter )r r& r' r r r schedule8 s z VirtualTimeScheduler.schedule)duetimer& r' r c C s | | j|}| j|||dS )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( )addr r) )r r+ r& r' timer r r schedule_relativeH s z&VirtualTimeScheduler.schedule_relativec C s< | |}t| |||}| j | j| W 5 Q R X |jS )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$ r r r enqueueZ
disposable)r r+ r& r' dtsir r r r) ] s
z&VirtualTimeScheduler.schedule_absolutec C s | j | jr W 5 Q R dS d| _W 5 Q R X d}| j | jrD| jsPW 5 Q R q| j }|j| jkrt| jtr||j| _n| |j| _d}n<|t
krt| jtr| j t dd7 _ n| jd7 _d}W 5 Q R X |
s| |d7 }q0| dS )z"Starts the virtual time scheduler.NTr i ) microsecondsg ?r )r r r dequeuer+ r%
isinstancer r
to_seconds MAX_SPINNINGclockr is_cancelledinvokestop)r Zspinningitemr r r startu s.
zVirtualTimeScheduler.startc C s | j d| _W 5 Q R X dS )z!Stops the virtual time scheduler.FN)r r r! r r r r: s zVirtualTimeScheduler.stop)r- r c C s$ | |}| j: | j|kr"t | j|ks2| jr@W 5 Q R dS d| _W 5 Q R X | jx | jrd| jspW 5 Q R q| j }|j|krW 5 Q R q|j| jkrt| j t
r|j| _ n| |j| _ | j W 5 Q R X |
sP| qP| j. d| _t| j t
r
|| _ n | || _ W 5 Q R X dS )zAdvances the schedulers clock to the specified absolute time,
running all work til that point.
Args:
time: Absolute time to advance the schedulers clock to.
NTF)r$ r r% r
r r peekr+ r4 r r r5 r3 r8 r9 )r r- r0 r; r r r
advance_to s2
z VirtualTimeScheduler.advance_toc C s* t d| | | | j| | dS )zAdvances the schedulers clock by the specified relative time,
running all work scheduled for that timespan.
Args:
time: Relative time to advance the schedulers clock by.
z(VirtualTimeScheduler.advance_by(time=%s)N)logdebugr> r, r% to_timedelta)r r- r r r
advance_by s z VirtualTimeScheduler.advance_byc C sd | | j| |}| |}| j|kr.t | j& t| jtrJ|| _n | || _W 5 Q R X dS )zAdvances the schedulers clock by the specified relative time.
Args:
time: Relative time to advance the schedulers clock by.
N)
r, r% rA r$ r
r r4 r r r5 )r r- absoluter0 r r r sleep s
zVirtualTimeScheduler.sleep)rC relativer c C s | || | S )a Adds a relative time value to an absolute time value.
Args:
absolute: Absolute virtual time value.
relative: Relative virtual time value to add.
Returns:
The resulting absolute virtual time sum value.
)r$ rA )clsrC rE r r r r, s zVirtualTimeScheduler.add)r )N)N)N) __name__
__module__ __qualname____doc__r r r r" propertyr7 r r% ZScheduledActionr r r ZDisposableBaser* Z RelativeTimer. r) r r< r: r> rB rD classmethodr,
__classcell__r r r r r sH
%. r )loggingr r r r r r r Z reactivexr Zreactivex.abc.schedulerr Zreactivex.internalr
r Zperiodicschedulerr
Z
scheduleditemr getLoggerr? r6 r r r r r r s