o i'@sRddlZddlZddlmZmZmZGdddZdejdefddZd d Z dS) N)AnyCallable Coroutinec @seZdZdZddZddZddZded efd d Z d ed efd dZ d edede e e e fffddZ dejded efddZdejd ed efddZdejd ed efddZdS)BackgroundSchedulerzf Schedules background tasks execution either in separate thread or in the running event loop. cCs d|_g|_t|_d|_dS)NF) _next_timer _event_loops threadingLock_lock_stoppedselfrD/home/ubuntu/.local/lib/python3.10/site-packages/redis/background.py__init__ s  zBackgroundScheduler.__init__cCs |dSN)stopr rrr__del__s zBackgroundScheduler.__del__cCs|j8|jr WddSd|_|jr|jd|_|jD] }|r-||jq!|jWddS1s>wYdS)zB Stop all scheduled tasks and clean up resources. NT) r r rcancelr is_runningcall_soon_threadsaferclear)r looprrrrs    "zBackgroundScheduler.stopdelaycallbackcG|j|jr WddSWdn1swYt}|j|j|Wdn1s7wYtjt||j ||g|Rdd}| dS)zI Runs callable task once after certain delay in seconds. NTtargetargsdaemon) r r asyncionew_event_looprappendrThread_start_event_loop_in_thread _call_laterstart)r rrrrthreadrrrrun_once(  zBackgroundScheduler.run_onceintervalcGr)zN Runs recurring callable task with given interval in seconds. NTr) r r r r!rr"rr#r$_call_later_recurringr&)r r*rrrr'rrr run_recurring=r)z!BackgroundScheduler.run_recurringcoro.csjjr WddSWdn1swYtt|g|Rfdd_dS)z Runs recurring coroutine with given interval in seconds in the current event loop. To be used only from an async context. No additional threads are created. NcsVjjr WddSWdn1swY_dSrr r call_laterrrr*rr tickwrappedrrr1`sz5BackgroundScheduler.run_recurring_async..tick)r r r get_running_loop_async_to_sync_wrapperr/r)r r*r-rrr0rrun_recurring_asyncRs z'BackgroundScheduler.run_recurring_asyncrcGsX|j|jr WddSWdn1swY|j||g|R|_dSrr.)r rrrrrrrr%lszBackgroundScheduler._call_latercGs`|j|jr WddSWdn1swY|j|||j|||g|RdSr)r r r%_execute_recurringr rr*rrrrrr+ts z)BackgroundScheduler._call_later_recurringcGs|j|jr WddSWdn1swYz||Wn ty.Ynw|j|jr? WddSWdn1sIwY|j|||j|||g|RdS)zR Executes recurring callable task with given interval in seconds. N)r r Exceptionr%r6r7rrrr6s,    z&BackgroundScheduler._execute_recurringN)__name__ __module__ __qualname____doc__rrrfloatrr(r,rrr5r AbstractEventLoopr%r+r6rrrrrsB   r event_loop call_soon_cbcGs t||j||g|Rz>|Wz3zt|}|D]}|q|tj|ddiWn ty:YnwW| dSW| dS| wz1zt|}|D]}|qW|tj|ddiWn tysYnwW| wW| w| w)z Starts event loop in a thread and schedule callback as soon as event loop is ready. Used to be able to schedule tasks using loop.call_later. :param event_loop: :return: return_exceptionsTN) r set_event_loop call_soon run_forever all_tasksrrun_until_completegatherr8close)r?r@rpendingtaskrrrr$s@          r$csfdd}|S)a Wraps an asynchronous function so it can be used with loop.call_later. :param loop: The event loop in which the coroutine will be executed. :param coro_func: The coroutine function to wrap. :param args: Positional arguments to pass to the coroutine function. :param kwargs: Keyword arguments to pass to the coroutine function. :return: A regular function suitable for loop.call_later. cstjiddS)N)r)r ensure_futurerr coro_funckwargsrrrr2sz'_async_to_sync_wrapper..wrappedr)rrMrrNr2rrLrr4s r4) r rtypingrrrrr>r$r4rrrrs