jKjddlmZddlZddlZddlZddlmZddlmZddl m Z m Z m Z m Z mZmZddlmZdZdZd ZeGd d eZy) )AnyN)Thread)_configure_posthog_logging)AI_EVENTS_ENDPOINTEVENTS_ENDPOINTAPIErrorDatetimeSerializer batch_post is_ai_event)EmptyiiiPcxeZdZdZej dZ d dZdZdZ dZ dZ d Z d Z d Zy) Consumerz.Consumes the messages from the client's queue.posthogNc tj|d|_||_||_||_||_||_||_||_ | |_ d|_ ||_ | |_ | |_y)zCreate a consumer thread.TN)r__init__daemonflush_atflush_intervalapi_keyhoston_errorqueuegzipdedicated_ai_endpointrunningretriestimeouthistorical_migration) selfrrrrrrrrrrrs a/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/posthog/consumer.pyrzConsumer.__init__(sp    ,      %:"    $8!c|jjd|jr|j|jr|jjdy)zRuns the consumer.zconsumer is running...zconsumer exited.N)logdebugruploadrs r runz Consumer.runKs? /0ll KKMll )*r!cd|_y)zPause the consumer.FN)rr&s r pausezConsumer.pauseSs  r!c d}|j}t|dk(ry |j|d}|D]}|jj|S#t$rz}|jj d|d}|j rF |j ||n2#t$r&}|jj d|Yd}~nd}~wwxYwYd}~d}~wwxYw#|D]}|jjwxYw)z:Upload the next batch of items, return whether successful.FrTzerror uploading: %szon_error handler failed: %sN) nextlenrequest Exceptionr#errorrr task_done)rsuccessbatcheitems r r%zConsumer.uploadWs  u:? ' LL G $$& E HHNN0! 4G}}EMM!U+ EHHNN#@!DDE  E $$&sSA C"*C B C C)C C CCCCC#Dc|j}g}tj}d}t||jkr%tj|z }||j k\r |S |j d|j |z }ttj|tj}|j|}||kDr:|jjd|dzt||j|j!|||z }|t"k\r|jj%d| |S t||jkr%|S#t&$rY|SwxYw)z)Return the next batch of items to upload.rT)blockr)clsz(Item exceeds %dKiB limit, dropping. (%s)izhit batch size limit (size: %d))rtime monotonicr,rrgetjsondumpsr encode _max_msg_sizer#r/strr0appendBATCH_SIZE_LIMITr$r ) rritems start_time total_sizeelapsedr4 item_size max_msg_sizes r r+z Consumer.nextpsW ^^%  %j4==(nn&3G$---* ) yytT5H5H75RyS 45G H O O QR #11$7 |+HHNNB$,D  OO% T"i' !11HHNN#DjQ 2%%j4==(0    s(B!E% ;E%% E21E2c|js|j|tyg}g}|D]1}t|j dr|n|}|j |3d}|tf|t ffD]\}}|s |j||||y#t$r,} || }n|jjd|| Yd} ~ Sd} ~ wwxYw)aUpload the batch, routing `$ai_*` events to their own endpoint when enabled. Each destination is attempted independently so a failure on one does not skip the other. The first failure is raised (so `upload()` logs it and invokes `on_error`); a second is logged here so it isn't silently lost. The batch was already dequeued in `upload()`, so unsent events are dropped after retries, same as the single-endpoint path. Neventzerror uploading to %s: %s) r_sendrr r:r@rr.r#r/) rr2 ai_eventsanalytics_eventsr4target first_exceventspathr3s r r-zConsumer.requests)) JJuo . ! &(D"-dhhw.?"@YFVF MM$   / * + LFD I 64(   O !  I$ !IHHNN#>aH  Is:B C"CCc d}d}t|jdzD]H} t|j|j|j |j ||j|y|r|y#t$rp}|}||s||jkrLt|dd}|r|dkDrtj|n"tjtd|zdYd}~d}~wwxYw) zLAttempt to upload a single batch to `path`, retrying before raising an errorct|trFt|jtr+d|jcxkxrdkncxr|jdv Syy)Nii)iiFT) isinstancerstatusint)excs r is_retryablez$Consumer._send..is_retryablesO#x(cjj#. 0S0Rcjj 6Rr!N)rrr2rrP retry_afterr) rangerr rrrrrr.getattrr8sleepmin)rr2rPrWlast_excattemptr3rYs r rJzConsumer._sends T\\A-.G 8LLII LL)-)B)B/0 N  8#AT\\)")!]D"AK"{Q ;/ 3q'z2#67 8sAA.. C'7A&C""C'cf|jr t|jdrtStS)NrI)rr r:AI_MAX_MSG_SIZE MAX_MSG_SIZE)rr4s r r>zConsumer._max_msg_sizes'  % %+dhhw6G*H" "r!) dNNg@F FF)__name__ __module__ __qualname____doc__logging getLoggerr#rr'r)r%r+r-rJr>r!r rr#s`8 '  I &C   "#!9F+2 D#J+Zr!r)typingrr;rlr8 threadingrposthog._loggingrposthog.requestrrrr r r rr rdrcrArrnr!r rssR  7 "#DvDr!