3 \o@sdZddlZddlZddlmZy ddlZWnek rHddlZYnXddlmZ dgZ GdddeddZ d e j_d e j _d e j _d e j_d e j_eZGdddZdS)aA generally useful event scheduler class. Each instance of this class manages its own queue. No multi-threading is implied; you are supposed to hack that yourself, or use a single instance per application. Each instance is parametrized with two functions, one that is supposed to return the current time, one that is supposed to implement a delay. You can implement real-time scheduling by substituting time and sleep from built-in module time, or you can implement simulated time by writing your own functions. This can also be used to integrate scheduling with STDWIN events; the delay function is allowed to modify the queue. Time can be expressed as integers or floating point numbers, as long as it is consistent. Events are specified by tuples (time, priority, action, argument, kwargs). As in UNIX, lower priority numbers mean higher priority; in this way the queue can be maintained as a priority queue. Execution of the event means calling the action function, passing it the argument sequence in "argument" (remember that in Python, multiple function arguments are be packed in a sequence) and keyword parameters in "kwargs". The action function may be an instance method so it has another way to reference private data (besides global variables). N) namedtuple) monotonic schedulerc@s8eZdZgZddZddZddZddZd d Zd S) EventcCs|j|jf|j|jfkS)N)timepriority)sor /usr/lib64/python3.6/sched.py__eq__'sz Event.__eq__cCs|j|jf|j|jfkS)N)rr)rr r r r __lt__(sz Event.__lt__cCs|j|jf|j|jfkS)N)rr)rr r r r __le__)sz Event.__le__cCs|j|jf|j|jfkS)N)rr)rr r r r __gt__*sz Event.__gt__cCs|j|jf|j|jfkS)N)rr)rr r r r __ge__+sz Event.__ge__N) __name__ __module__ __qualname__ __slots__r r rrrr r r r r%s rz(time, priority, action, argument, kwargszaNumeric type compatible with the return value of the timefunc function passed to the constructor.zSEvents scheduled for the same time will be executed in the order of their priority.z?Executing the event means executing action(*argument, **kwargs)zGargument is a sequence holding the positional arguments for the action.zDkwargs is a dictionary holding the keyword arguments for the action.c@s^eZdZeejfddZfefddZfefddZ ddZ d d Z dd d Z e ddZdS)rcCs g|_tj|_||_||_dS)zGInitialize a new instance, passing the time and delay functionsN)_queue threadingRLock_locktimefunc delayfunc)selfrrr r r __init__<s zscheduler.__init__c Cs@|tkr i}t|||||}|jtj|j|WdQRX|S)zEnter a new event in the queue at an absolute time. Returns an ID for the event which can be used to remove it, if necessary. N) _sentinelrrheapqheappushr)rrractionargumentkwargseventr r r enterabsDs zscheduler.enterabscCs|j|}|j|||||S)z{A variant that specifies the time as a relative time. This is actually the more commonly used interface. )rr$)rdelayrr r!r"rr r r enterRs zscheduler.enterc Cs.|j|jj|tj|jWdQRXdS)zRemove an event from the queue. This must be presented the ID as returned by enter(). If the event is not in the queue, this raises ValueError. N)rrremoverheapify)rr#r r r cancel[s zscheduler.cancelc Cs|j |j SQRXdS)z!Check whether the queue is empty.N)rr)rr r r emptyfszscheduler.emptyTc Cs|j}|j}|j}|j}tj}x||>|s,P|d\}}} } } |} || krRd} n d} ||WdQRX| r|sx|| S||| q | | | |dq WdS)aExecute events until the queue is empty. If blocking is False executes the scheduled events due to expire soonest (if any) and then return the deadline of the next scheduled call in the scheduler. When there is a positive delay until the first event, the delay function is called and the event is left in the queue; otherwise, the event is removed from the queue and executed (its action function is called, passing it the argument). If the delay function returns prematurely, it is simply restarted. It is legal for both the delay function and the action function to modify the queue or to raise an exception; exceptions are not caught but the scheduler's state remains well-defined so run() may be called again. A questionable hack is added to allow other threads to run: just after an event is executed, a delay of 0 is executed, to avoid monopolizing the CPU when other threads are also runnable. rTFN)rrrrrheappop)rZblockinglockqrrpoprrr r!r"Znowr%r r r runks* z scheduler.runc Cs:|j|jdd}WdQRXtttj|gt|S)zAn ordered list of upcoming events. Events are named tuples with fields for: time, priority, action, arguments, kwargs N)rrlistmaprr+len)rZeventsr r r queues zscheduler.queueN)T)rrr_timerZsleeprrr$r&r)r*r/propertyr3r r r r r:s   2)__doc__rr collectionsrr ImportErrorZdummy_threadingrr4__all__rrr r!r"objectrrr r r r s"