idZddlmZddlZddlZddlZddlmZerddlm Z ddl Z ddl m Z ejeZGdd ej"Zy) zAConsumer thread for processing live data with callback functions.) annotationsN) TYPE_CHECKING)Callable)SeiscbeZdZdZ d fd Zd dZd dZd dZd dZd ddZ d dZ xZ S)ConsumeraKThreaded consumer for processing Seis data via callback. Runs in a separate thread to process incoming data bars from a Seis by calling a user-provided callback function. Uses a queue for thread-safe data passing. Args: seis: Seis instance this consumer processes data from callback: Function(seis, data) to call with new data bars Example: >>> def my_callback(seis, data): ... print(f"Received bar: {data}") >>> consumer = Consumer(seis, my_callback) >>> consumer.start() >>> consumer.put(new_data) cLt|dtj|_||_||_|j jd|j jd|j jd|j jj|_ y)zInitialize consumer thread. Args: seis: Seis to consume data from callback: Callback function for new data T)daemon_N) super__init__queueQueue_bufferseiscallback__name__symbolexchangeintervalvaluename)selfrr __class__s L/home/work/apex_v16/venv/lib/python3.12/site-packages/tvDatafeed/consumer.pyrzConsumer.__init__%s %9>    }}%%&ayy yy!!"!yy!!''( * cPd|jd|jjdS)zReturn detailed representation.z Consumer(z, )rrrrs r__repr__zConsumer.__repr__>s&499-r$--*@*@)ACCrcL|jd|jjS)zReturn string representation.z , callback=r r!s r__str__zConsumer.__str__Bs"))K (>(>'?@@rcDtjd|j |jj }|!tjd|jn |j |j |[tjd|jd|_d|_d|_y#t$rc}tjd|j j|j |d |jn#t$rYnwxYwYd}~d}~wwxYw)zMain thread loop - processes data from queue. Continuously pulls data from the queue and calls the callback function. Exits when None is received (shutdown signal). zConsumer thread %s startedTNz+Consumer thread %s received shutdown signalz'Callback %s raised exception for %s: %s)exc_infozConsumer thread %s exiting) loggerdebugrrgetrr Exceptionerrorr del_consumer)rdataes rrunz Consumer.runFs  1499=<<##%D| JDIIV  dii.4  1499=   '  =MM**II! %%'  s< B33 D<8D5DD DDDDDcT|j|jj|yy)z{Add data to processing queue. Args: data: DataFrame with bar data, or None to signal shutdown Nrput)rr-s rr2z Consumer.putms% << # LL  T " $rcT|jy|jj||S)zStop thread and remove from Seis. Args: timeout: Maximum wait time in seconds (-1 for blocking) Returns: True if successful, False if timeout T)rr,)rtimeouts rr,zConsumer.del_consumervs( 99 yy%%dG44rcT|j|jjdyy)zcSignal thread to stop processing. Sends None to queue which triggers thread exit. Nr1r!s rstopz Consumer.stops% << # LL  T " $r)rrrz$Callable[[Seis, pd.DataFrame], None]returnNone)r7str)r7r8)r-zpd.DataFrame | Noner7r8))r4intr7bool) r __module__ __qualname____doc__rr"r$r/r2r,r6 __classcell__)rs@rr r sJ$  7    2DA%N# 5#rr )r? __future__rloggingr threadingtypingrcollections.abcrpandaspdrr getLoggerrr'Threadr rrrKsIG"  (   8 $w#yw#r