g|/ddlmZddlZddlmZmZmZddlmZddl m Z ddl m Z ddl mZddlmZmZGd d eZed Gd dZed GddZ ddZ ddZGddZy)) annotationsN) dataclassfieldreplace)Enum)cast)util)StringTriggerValue) WidgetState WidgetStatesceZdZdZdZdZy)ScriptRequestTypeCONTINUESTOPRERUNN)__name__ __module__ __qualname__rrri/var/www/openai/venv/lib/python3.12/site-packages/streamlit/runtime/scriptrunner_utils/script_requests.pyrrsH D ErrT)frozenceZdZUdZdZded<dZded<dZded<dZded <dZ d ed <e e Z d ed<dZ ded<dZded<ddZy) 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_rerunc,tj|SNr repr_selfs r__repr__zRerunData.__repr__:zz$rreturnr)rrr__doc__r__annotations__rr r!r"rlistr%r'r(r/rrrrr)se5L#)-M&-cIs#K"#(#>y>%*d*M4 rrcDeZdZUdZded<dZded<ed dZd dZy) ScriptRequestz,A STOP or RERUN request and associated data.rtypeNzRerunData | None _rerun_datac|jtjur tdt t |j S)Nz)RerunData is only set for RERUN requests.)r8rr RuntimeErrorrrr9r-s r rerun_datazScriptRequest.rerun_dataEs4 99-33 3JK KIt//00rc,tj|Sr*r+r-s rr/zScriptRequest.__repr__Kr0r)r2rr1) rrrr3r4r9propertyr<r/rrrr7r7>s-6 $(K!( 11  rr7c"t|xr| 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's r'_fragment_run_should_not_preempt_scriptr@Os ! " C+C'CCrc|s|sy|s|S|s|S|jDcic]}|j|}}ddtdfg}|jD]}|D]{\}}|jd|k(st |||k7s+|j |j}|r%|jd|k(s]t |||k(sm|||j<}t } | jj|j| Scc}w)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. N) trigger_valueFstring_trigger_value)datavalue) widgetsidStringTriggerValueProto WhichOneofgetattrgetr extendvalues) old_states new_stateswstate states_by_idtrigger_value_types old_statetrigger_value_type unset_valuenew_trigger_val coalesceds r_coalesce_widget_statesrX]s/ j  )3(:(:,(:f 6(:, ! !8d!CD '' /B +  $$W-1CCI'9:kI"."2"29<<"@'$..w7;MM 1CD S1:L.+0C(0I \0023 G,sDc6eZdZdZdZddZd dZd dZd dZy) ScriptRequestszAn interface for communicating with a ScriptRunner. Thread-safe. AppSession makes requests of a ScriptRunner through this class, and ScriptRunner handles those requests. c~tj|_tj|_t |_yr*) threadingLock_lockrr_staterr9r-s r__init__zScriptRequests.__init__s'^^% '00 $;rcp|j5tj|_dddy#1swYyxYw)zRequest that the ScriptRunner stop running. A stopped ScriptRunner can't be used anymore. STOP requests succeed unconditionally. N)r^rrr_r-s r request_stopzScriptRequests.request_stops"ZZ+00DKZZs,5c |j5|jtjk(r dddy|jtjk(rKtj |_|j rt|d|j g}||_ dddy|jtj k(rt|jj|j}|j rBg|jj}|j |vr7|j|j n|jr |j}ng}t|j||j|j ||j"|j$|_ dddyt'd|j#1swYyxYw)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)rrr r!r%r'r(z Unrecognized ScriptRunnerState: )r^r_rrrrr"rr9rXrr%appendrrr r!r'r(r;)r.new_datacoalesced_statesr%s r request_rerunzScriptRequests.request_rerunsZZ{{/444Z {{/888055 ''& $(+3+?+?*@ H $, )Z,{{/555$;$$22H4J4J$ '')N$*:*:*L*L(M%++3DD)001E1EF//(0(B(B% )+%#,!)!6!6"2%-%>%>&00&7-5-N-N"*"8"8$ sZx!A$++OP PyZsF8AF8C:F8 F88Gc|jtjk(sQ|jtjk(r5t |j j |j jry|j5|jtjk(rt |j j |j jr dddytj|_ttj|j cdddS|jtjk(sJttjcdddS#1swYyxYw)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) r_rrrr@r9r%r'r^r7rr-s ron_scriptrunner_yieldz$ScriptRequests.on_scriptrunner_yields ;;+44 4 KK,22 27  22  99 ZZ{{/555;$$66$$== Z088 $%6%<%N>NOZ;;"3"8"88 88 !2!7!78ZZsN>NOZ,00DK !2!7!78ZZsAB$--B$$B-N)r2None)rerr2r&)r2zScriptRequest | None)r2r7) rrrr3r`rbrgrirkrrrrZrZs$ ' 1FQP&9P9rrZ)r%r$r'r&r2r&)rNrrOrr2r) __future__rr\ dataclassesrrrenumrtypingr streamlitr streamlit.proto.Common_pb2r rH streamlit.proto.WidgetStates_pb2r r rrr7r@rXrZrrrrts#11TF   $   ( $     D  D" D  D6#61D66rU9U9r