g6&xddlmZddlmZmZmZddlmZerddlm Z GddZ d dZ d dZ d d Z y )) annotations) TYPE_CHECKINGAnyCallable) ForwardMsg)DeltaceZdZUdZdZded<e d dZdZddZ ddZ dd Z d dd Z dd Z dd Zy)ForwardMsgQueuea:Accumulates a session's outgoing ForwardMsgs. Each AppSession adds messages to its queue, and the Server periodically flushes all session queues and delivers their messages to the appropriate clients. ForwardMsgQueue is not thread-safe - a queue should only be used from a single thread. N#Callable[[ForwardMsg], None] | None_before_enqueue_msgc|t_y)zjSet a callback to be called before a message is enqueued. Used in static streamlit app generation.N)r r )before_enqueue_msgs X/var/www/openai/venv/lib/python3.12/site-packages/streamlit/runtime/forward_msg_queue.pyon_before_enqueue_msgz%ForwardMsgQueue.on_before_enqueue_msg&s /A+c g|_i|_yN)_queue_delta_index_mapselfs r__init__zForwardMsgQueue.__init__.s(* =?rcddlm}|jDcgc] }|| c}t|jj dScc}w)Nr) MessageToDict)queueids)google.protobuf.json_formatrrlistrkeys)rrms r get_debugzForwardMsgQueue.get_debug7sF=15 < 1mA& <--2245   new_sessionparent_messagescript_finishedsession_status_changedN)r WhichOneofr- fragment_id_update_script_finished_messager)rretain_lifecycle_msgsfragment_ids_this_runr/s rclearzForwardMsgQueue.clearbs$%DK@!#9 ;;&C>>&)*5 ) II1 # 5 5 =#&99#8#8@U#U-05JRV5VW&DK<!#=sA0B c>|j}|j|S)zeClear the queue and return a list of the messages it contained before being cleared. )rrA)rrs rflushzForwardMsgQueue.flushs   rc,t|jSrr#rs r__len__zForwardMsgQueue.__len__s4;;r)rr returnNone)rFzdict[str, Any])rFbool)r/rrFrG)FN)r?rHr@zlist[str] | NonerFrG)rFzlist[ForwardMsg])rFint)__name__ __module__ __qualname____doc__r __annotations__ staticmethodrrr!r%r5rArCrErrr r s@D<CA?A AA? % D',263##3# 03#  3#j rr ct|jdsy|jjd}|dk7xr|dk7S)zHTrue if the ForwardMsg is potentially composable with other ForwardMsgs.r-Fr7add_rowsarrow_add_rows)HasFieldr-r<)r/ delta_types rr'r's= <<  %%f-J  # F 6F(FFrcp|jd}|dk(ry|jd}|dk(r|S|dk(r|Sy)aJCombines new_delta onto old_delta if possible. If the combination takes place, the function returns a new Delta that should replace old_delta in the queue. If the new_delta is incompatible with old_delta, the function returns None. In this case, the new_delta should just be appended to the queue as normal. r7 add_blockN new_element)r<) old_delta new_deltaold_delta_typenew_delta_types rr,r,sP))&1N$))&1N&$ rc|jddk(rJ|dus'|jtjjk7rtjj |_|S)a- When we are here, the message queue is cleared from non-lifecycle messages before they were flushed to the browser. If there were no non-lifecycle messages in the queue, changing the type here should not matter for the frontend anyways, so we optimistically change the `script_finished` message to `FINISHED_EARLY_FOR_RERUN`. This indicates to the frontend that the previous run was interrupted by a new script start. Otherwise, a `FINISHED_SUCCESSFULLY` message might trigger a reset of widget states on the frontend. r7r:F)r<r:rScriptFinishedStatusFINISHED_SUCCESSFULLYFINISHED_EARLY_FOR_RERUN)r/is_fragment_runs rr>r>sS ~~f!22 5   *"A"A"W"W W(==VV JrN)r/rrFrH)rYrrZrrFz Delta | None)r/rrarHrFr) __future__rtypingrrrstreamlit.proto.ForwardMsg_pb2rstreamlit.proto.Delta_pb2rr r'r,r>rPrrrfsO#//5/G G T G!H &*r