o ,&]i1@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__rrrrrr/var/www/html/IGF-ODF-V3/venv/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<e ed Zded<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_rerunzset[str]cached_message_hasheszContextInfo | None context_inforeturncC t|SNr repr_selfrrr__repr__A zRerunData.__repr__r)r)rrr__doc__r__annotations__rrrr rlistr#r%r&setr'r(r0rrrrr,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)r8rr RuntimeErrorrr9r.rrr rerun_dataLs  zScriptRequest.rerun_datarcCr*r+r,r.rrrr0Rr1zScriptRequest.__repr__)r)rr2) rrrr3r4r9propertyr;r0rrrrr7Es   r7r#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_scriptVs 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 wsz+_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_statesds4      rVc@sBeZdZdZdddZdddZdd 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. r)NonecCst|_tj|_t|_dSr+) threadingLock_lockrr_staterr9r.rrr__init__s  zScriptRequests.__init__cCs4|j tj|_WddS1swYdS)zRequest that the ScriptRunner stop running. A stopped ScriptRunner can't be used anymore. STOP requests succeed unconditionally. N)r[rrr\r.rrr request_stops "zScriptRequests.request_stopnew_datarr$c Cs"|j|jtjkr WddS|jtjkr5tj|_|jr)t|d|jgd}||_ WddS|jtjkrt |jj |j }|jrXg|jj }|j|vrW| |jn |j r_|j }ng}t |j||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&r( Unrecognized ScriptRunnerState: )r[r\rrrrr rr9rVrr#appendrrrrr'r%r&r(r:)r/r_coalesced_statesr#rrr request_rerunsT        >zScriptRequests.request_rerunScriptRequest | NonecCs|jtjks|jtjkrt|jj|jjrdS|jF|jtjkrEt|jj|jjr3 WddStj|_t tj|jWdS|jtj krTt d|jdt tj WdS1scwYdS)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. Nr`z. This should never happen.) r\rrrr=r9r#r%r[r7rr:r.rrron_scriptrunner_yields2      $z$ScriptRequests.on_scriptrunner_yieldr7cCsl|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)r[r\rrrr7r9rr.rrron_scriptrunner_ready%s  $z$ScriptRequests.on_scriptrunner_readyN)r)rX)r_rr)r$)r)rd)r)r7) rrrr3r]r^rcrerfrrrrrWs    J+rW)r#r"r%r$r)r$)r>rr?rr)r) __future__rrY dataclassesrrrenumrtypingrr streamlitr streamlit.proto.Common_pb2r rI streamlit.proto.WidgetStates_pb2r r streamlit.proto.ClientState_pb2r rrr7r=rVrWrrrrs$       9