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 (async 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)ALockAEventc@sheZdZdddZdZdedeegefde fdd Z d edeegefde fd d Z dd dZ dS)AsyncSchedulerreturnNcCsg|_t|_t|_dS)N)_queuer _lockr _event)selfrQ/home/ertert/spirit/venv/lib/python3.10/site-packages/psycopg_pool/sched_async.py__init__s zAsyncScheduler.__init__g@delayactioncst|}|||IdHS)zpEnter a new task in the queue delayed in the future. Schedule a `!None` to stop the execution. N)renterabs)rrrtimerrrenter"s zAsyncScheduler.enterrc srt||}|j4IdHt|j||jd|u}WdIdHn 1IdHs+wY|r7|j|S)znEnter a new task in the queue at an absolute time. Schedule a `!None` to stop the execution. Nr)r rrrrset)rrrtaskfirstrrrr*s  ( zAsyncScheduler.enterabsc s|j} |j4IdH2t}|r|dnd}|r,|j|kr$t|n |j|}d}n|j}|jWdIdHn 1IdHsDwY|rw|jsPdSz |IdHWn&t yv}zt d|j|j j |WYd}~nd}~ww|j|IdHq)zExecute the events scheduled.TNrz$scheduled task run %s failed: %s: %s)rrrrrEMPTY_QUEUE_TIMEOUTrclearr Exceptionloggerwarning __class____name__ wait_timeout)rqnowrrerrrrun9s<    (  zAsyncScheduler.run)rN) r$ __module__ __qualname__rrfloatrrrr rrr)rrrrr s  ""r )__doc__loggingrrheapqrrtypingrrrr_taskr _acompatr r getLoggerr$r!r rrrrs