o ^f? @sxdZddlZddlmZddlmZmZddlmZm Z m Z m Z ddl m Z ddlmZmZeeZGd d d ZdS) a A minimal scheduler to schedule tasks to run in the future (sync version). Inspired to the standard library `sched.scheduler`, but designed for multi-thread usage from the ground up, not as an afterthought. Tasks can be scheduled in front of the one currently running and `Scheduler.run()` can be left running without any tasks scheduled. Tasks are called "Task", not "Event", here, because we actually make use of `[threading/asyncio].Event` and the two would be confusing. N) monotonic)heappushheappop)AnyCallableListOptional)Task)LockEventc@sheZdZdddZdZdedeegefde fdd Z d edeegefde fd d Z dd dZ dS) SchedulerreturnNcCsg|_t|_t|_dS)N)_queuer _lockr _event)selfrK/home/ertert/spirit/venv/lib/python3.10/site-packages/psycopg_pool/sched.py__init__s zScheduler.__init__g@delayactioncCst|}|||S)zpEnter a new task in the queue delayed in the future. Schedule a `!None` to stop the execution. )renterabs)rrrtimerrrenter%s  zScheduler.enterrcCs\t||}|jt|j||jd|u}Wdn1s wY|r,|j|S)znEnter a new task in the queue at an absolute time. Schedule a `!None` to stop the execution. rN)r rrrrset)rrrtaskfirstrrrr-s   zScheduler.enterabsc Cs|j} |j/t}|r|dnd}|r'|j|krt|n |j|}d}n|j}|jWdn1s9wY|ri|jsEdSz|Wn#t yh}zt d|j|j j |WYd}~n d}~ww|j|q)zExecute the events scheduled.TrNz$scheduled task run %s failed: %s: %s)rrrrrEMPTY_QUEUE_TIMEOUTrclearr Exceptionloggerwarning __class____name__wait)rqnowrrerrrrun<s:       z Scheduler.run)rN) r$ __module__ __qualname__rrfloatrrrr rrr)rrrrr s  ""r )__doc__loggingrrheapqrrtypingrrrr_taskr _acompatr r getLoggerr$r!r rrrrs