qidZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z ddl mZeGddZGd d eZeGd d ZGd deZy)zAsyncio task management. This module provides task management functionality. Includes both abstract base classes and concrete implementations for managing asyncio tasks with comprehensive monitoring and cleanup capabilities. N)ABCabstractmethod) dataclass) CoroutineDictOptionalSequence)loggerc0eZdZUdZej ed<y)TaskManagerParamszConfiguration parameters for task manager initialization. Parameters: loop: The asyncio event loop to use for task management. loopN)__name__ __module__ __qualname____doc__asyncioAbstractEventLoop__annotations__T/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/utils/asyncio/task_manager.pyr r s  # ##rr ceZdZdZedefdZedejfdZ ede de dejfdZ edd ejd eefd Zedeejfd Zy )BaseTaskManagerzAbstract base class for asyncio task management. Provides the interface for creating, monitoring, and managing asyncio tasks. paramscyzInitialize the task manager with configuration parameters. Args: params: Configuration parameters for task management. Nrselfrs rsetupzBaseTaskManager.setup( rreturncy)zuGet the event loop used by this task manager. Returns: The asyncio event loop instance. Nrrs rget_event_loopzBaseTaskManager.get_event_loop1r r coroutinenamecy)a]Creates and schedules a new asyncio Task that runs the given coroutine. The task is added to a global set of created tasks. Args: coroutine: The coroutine to be executed within the task. name: The name to assign to the task for identification. Returns: The created task object. Nr)rr%r&s r create_taskzBaseTaskManager.create_task:s rNtasktimeoutc Kyw)XCancels the given asyncio Task and awaits its completion with an optional timeout. This function removes the task from the set of registered tasks upon completion or failure. Args: task: The task to be cancelled. timeout: The optional timeout in seconds to wait for the task to cancel. Nr)rr)r*s r cancel_taskzBaseTaskManager.cancel_taskIs  scy)Returns the list of currently created/registered tasks. Returns: Sequence of currently managed asyncio tasks. Nrr#s r current_taskszBaseTaskManager.current_tasksVr rN)rrrrrr rrrr$rstrTaskr(rfloatr-r r0rrrrr"s  -    9 9    Y  c  gll      gll  Xe_     x 5  rrc0eZdZUdZej ed<y)TaskDatazwInternal data structure for tracking task metadata. Parameters: task: The asyncio Task being managed. r)N)rrrrrr3rrrrr6r6`s ,,rr6ceZdZdZddZdefdZdejfdZ de d e dejfd Z dd ejd eefd ZdeejfdZdefdZd ejfdZy) TaskManagerzConcrete implementation of BaseTaskManager. Manages asyncio tasks. Provides comprehensive task lifecycle management including creation, monitoring, cancellation, and cleanup. r!Nc i|_d|_y)z5Initialize the task manager with empty task registry.N)_tasks_paramsr#s r__init__zTaskManager.__init__ss+- 48 rrc,|js||_yyr)r;rs rrzTaskManager.setupxs ||!DLrc\|js td|jjS)zGet the event loop used by this task manager. Returns: The asyncio event loop instance. Raises: Exception: If the task manager is not properly set up. 2TaskManager is not setup: unable to get event loop)r; Exceptionr r#s rr$zTaskManager.get_event_loops'||PQ Q||   rr%r&cVfd}|js td|jjj|}|j |j |j |jt|tjd|S)aCreates and schedules a new asyncio Task that runs the given coroutine. The task is added to a global set of created tasks. Args: coroutine: The coroutine to be executed within the task. name: The name to assign to the task for identification. Returns: The created task object. Raises: Exception: If the task manager is not properly set up. c XK d{S7#tj$rtjdt$rb}t j |j}|d}tjd|jd|jd|Yd}~yd}~wwxYww)Nz: task cancelledz unexpected exception (:): ) rCancelledErrorr tracer@ traceback extract_tb __traceback__errorfilenamelineno)etblastr%r&s r run_coroutinez.TaskManager.create_task..run_coroutines b&&))  v%567 b))!//:"v v%## L NNdV#IJ K%%   %%aoo6Bb6D LL&Edmm_TUVZVaVaUbbefgehi   %%aoo6Bb6D OO&Edmm_TUVZVaVaUbbefgehi    s!E(AAAE(A A AE(AA+E%E(E%E(E%AC;6E(; E%AE  E%%E(cp|jjDcgc]}|jc}Scc}w)r/)r:valuesr))rdatas rr0zTaskManager.current_taskss) '+kk&8&8&:;d ;;;s3 task_datacV|jj}||j|<y)zfAdd a task to the internal registry. Args: task_data: The task metadata. N)r)rWr:)rrar&s rrUzTaskManager._add_tasks$ ~~&&(% Drc|j} |j|=y#t$r$}tj|d|Yd}~yd}~wwxYw)zHandle task completion by removing the task from the registry. Args: task: The completed asyncio task. z1: unable to remove task data (already removed?): N)rWr:KeyErrorr rG)rr)r&rNs rrTzTaskManager._task_done_handlersP }} X D! X LLD6!RSTRUV W W Xs A AA )r!Nr1)rrrrr<r rrrr$rr2r3r(rr4r-r r0r6rUrTrrrr8r8ks9 "-" ! 9 9 !$Y$c$gll$L"gll"Xe_"Hrisp#!66 $$ $; c; |  HX/HXr