o h0@sddlmZddlZddlmZmZmZddlmZddl m Z m Z ddl m Z ddlmZddlmZmZe r>dd lmZGd d d eZed d GdddZed d GdddZd ddZd!ddZGdddZdS)") annotationsN) dataclassfieldreplace)Enum) TYPE_CHECKINGcast)util)ChatInputValue) WidgetState WidgetStates) ContextInfoc@seZdZdZdZdZdS)ScriptRequestTypeCONTINUESTOPRERUNN)__name__ __module__ __qualname__rrrrrh/var/www/vscode/kcb/lib/python3.10/site-packages/streamlit/runtime/scriptrunner_utils/script_requests.pyrsrT)frozenc@seZdZUdZdZded<dZded<dZded<dZded <dZ d ed <e e d Z d ed<dZ ded<dZded<dZded<dddZdS) RerunDataz+Data attached to RERUN requests. Immutable.str query_stringNWidgetStates | None widget_statespage_script_hash page_namez str | None fragment_id)default_factory list[str]fragment_id_queueFboolis_fragment_scoped_rerun is_auto_rerunzContextInfo | None context_inforeturncC t|SNr repr_selfrrr__repr__? zRerunData.__repr__r(r)rrr__doc__r__annotations__rrrr rlistr#r%r&r'r/rrrrr,s         rc@s>eZdZUdZded<dZded<edd d Zdd d ZdS) ScriptRequestz,A STOP or RERUN request and associated data.rtypeNzRerunData | None _rerun_datar(rcCs |jtjur tdtd|jS)Nz)RerunData is only set for RERUN requests.r)r6rr RuntimeErrorrr7r-rrr rerun_dataJs  zScriptRequest.rerun_datarcCr)r*r+r-rrrr/Pr0zScriptRequest.__repr__)r(rr1) rrrr2r3r7propertyr9r/rrrrr5Cs   r5r#r"r%r$r(cCst|o| S)a7Returns whether the currently running script should be preempted due to a fragment rerun. Reruns corresponding to fragment runs that weren't caused by calls to `st.rerun(scope="fragment")` should *not* cancel the current script run as doing so will affect elements outside of the fragment. )r$)r#r%rrr'_fragment_run_should_not_preempt_scriptTs r; old_statesr new_statesc Cs|s|sdS|s |S|s|Sdd|jD}ddtddfg}|jD]2}|D]-\}}|d|krSt|||krS||j}|rN|d|krSt|||krS|||j<q&q"t}|j||S)adCoalesce an older WidgetStates into a newer one, and return a new WidgetStates containing the result. For most widget values, we just take the latest version. However, any trigger_values (which are set by buttons) that are True in `old_states` will be set to True in the coalesced result, so that button presses don't go missing. NcSsi|]}|j|qSr)id).0wstaterrr usz+_coalesce_widget_states..) trigger_valueFchat_input_value)datavalue) widgetsChatInputValueProto WhichOneofgetattrgetr>r extendvalues) r<r= states_by_idtrigger_value_types old_statetrigger_value_type unset_valuenew_trigger_val coalescedrrr_coalesce_widget_statesbs4      rTc@s@eZdZdZddZdddZdd d ZdddZdddZdS)ScriptRequestszAn interface for communicating with a ScriptRunner. Thread-safe. AppSession makes requests of a ScriptRunner through this class, and ScriptRunner handles those requests. cCst|_tj|_t|_dSr*) threadingLock_lockrr_staterr7r-rrr__init__s  zScriptRequests.__init__r(NonecCs4|j tj|_WddS1swYdS)zRequest that the ScriptRunner stop running. A stopped ScriptRunner can't be used anymore. STOP requests succeed unconditionally. N)rXrrrYr-rrr request_stops "zScriptRequests.request_stopnew_datarr$c Cs|j|jtjkr WddS|jtjkr5tj|_|jr)t|d|jgd}||_ WddS|jtjkr}t |jj |j }|jrXg|jj }|j|vrW| |jn |j r_|j }ng}t |j||j|j||j|j|jd|_ WddStd|j1swYdS)a5Request that the ScriptRunner rerun its script. If the ScriptRunner has been stopped, this request can't be honored: return False. Otherwise, record the request and return True. The ScriptRunner will handle the rerun request as soon as it reaches an interrupt point. NF)r r#T)rrrrr#r%r&r'z Unrecognized ScriptRunnerState: )rXrYrrrrr rr7rTrr#appendrrrrr%r&r'r8)r.r]coalesced_statesr#rrr request_rerunsR        =zScriptRequests.request_rerunScriptRequest | NonecCs|jtjks|jtjkrt|jj|jjrdS|j?|jtjkrEt|jj|jjr3 WddStj|_t tj|jWdS|jtj ksMJt tj WdS1s\wYdS)aCalled by the ScriptRunner when it's at a yield point. If we have no request or a RERUN request corresponding to one or more fragments (that is not a fragment-scoped rerun), return None. If we have a (full script or fragment-scoped) RERUN request, return the request and set our internal state to CONTINUE. If we have a STOP request, return the request and remain stopped. N) rYrrrr;r7r#r%rXr5rr-rrron_scriptrunner_yields,     $z$ScriptRequests.on_scriptrunner_yieldr5cCsl|j)|jtjkrtj|_ttj|jWdStj|_ttjWdS1s/wYdS)aYCalled by the ScriptRunner when it's about to run its script for the first time, and also after its script has successfully completed. If we have a RERUN request, return the request and set our internal state to CONTINUE. If we have a STOP request or no request, set our internal state to STOP. N)rXrYrrrr5r7rr-rrron_scriptrunner_readys  $z$ScriptRequests.on_scriptrunner_readyN)r(r[)r]rr(r$)r(ra)r(r5) rrrr2rZr\r`rbrcrrrrrUs   I(rU)r#r"r%r$r(r$)r<rr=rr(r) __future__rrV dataclassesrrrenumrtypingrr streamlitr streamlit.proto.Common_pb2r rG streamlit.proto.WidgetStates_pb2r r streamlit.proto.ClientState_pb2r rrr5r;rTrUrrrrs$       9