mxi1*ddlmZddlZddlmZmZmZddlmZddl m Z m Z ddl m Z ddlmZddlmZmZe rdd lmZGd d eZed GddZed GddZ ddZ ddZGddZy)) annotationsN) dataclassfieldreplace)Enum) TYPE_CHECKINGcast)util)ChatInputValue) WidgetState WidgetStates) ContextInfoceZdZdZdZdZy)ScriptRequestTypeCONTINUESTOPRERUNN)__name__ __module__ __qualname__rrrk/home/htdocs/ttos/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<e e Zded<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_rerunzset[str]cached_message_hasheszContextInfo | None context_infoc,tj|SNr repr_selfs r__repr__zRerunData.__repr__Azz$rreturnr)rrr__doc__r__annotations__r!r"r#r$rlistr'r)r*setr+r,r3rrrrr,s5L#)-M&-cIs#K"#(#>y>%*d*M4&+C&@8@'+L$+ rrcDeZdZUdZded<dZded<ed dZd dZy) ScriptRequestz,A STOP or RERUN request and associated data.rtypeNzRerunData | None _rerun_datac||jtjur tdt d|j S)Nz)RerunData is only set for RERUN requests.r)r=rr RuntimeErrorr r>r1s r rerun_datazScriptRequest.rerun_dataLs4 99-33 3JK KK!1!122rc,tj|Sr.r/r1s rr3zScriptRequest.__repr__Rr4r)r6rr5) rrrr7r8r>propertyrAr3rrrr<r<Es-6 $(K!( 33  rr<c"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_scriptrEVs ! " 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_valueFchat_input_value)datavalue) widgetsidChatInputValueProto WhichOneofgetattrgetr extendvalues) old_states new_stateswstate states_by_idtrigger_value_types old_statetrigger_value_type unset_valuenew_trigger_val coalesceds r_coalesce_widget_statesr]ds5 j  )3(:(:,$ 6,L, ! 0d;< ''; /B ; +  $$W-1CCI'9:kI"."2"29<<"@'$..w7;MM 1CD S1:L.+ ;;0I \0023 G,sDc8eZdZdZddZddZd 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_staterr>r1s 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)rcrrrdr1s r request_stopzScriptRequests.request_stops-ZZ 1+00DK 1 1 1s,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$|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) rr!r"r#r'r+r)r*r, Unrecognized ScriptRunnerState: )rcrdrrrrr$rr>r]r!r'appendrrr"r#r+r)r*r,r@)r2new_datacoalesced_statesr's r request_rerunzScriptRequests.request_rerunsZZ> Q{{/444> Q> Q {{/888055 ''& $(+3+?+?*@ H $, )> Q> Q,{{/555$;$$22H4J4J$ '')N$*:*:*L*L(M%++3DD)001E1EF//(0(B(B% )+%#,!)!6!6"2%-%>%>&00&7*2*H*H-5-N-N"*"8"8!)!6!6 $ w> Q> Q|!A$++OP P}> Q> QsGAGDG6GGc|jtjk(sQ|jtjk(r5t |j j |j jry|j5|jtjk(rt |j j |j jr dddytj|_ttj|j cdddS|jtjk7rtd|jdttjcdddS#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. Nriz. This should never happen.) rdrrrrEr>r'r)rcr<rr@r1s ron_scriptrunner_yieldz$ScriptRequests.on_scriptrunner_yields/ ;;+44 4 KK,22 27  22  99 ZZ 9{{/555;$$66$$==  9 9088 $%6%<%N>NO 9 9{{/444"6t{{mC^_!!2!7!78# 9 9 9srr1s ron_scriptrunner_readyz$ScriptRequests.on_scriptrunner_ready%sZZ 9{{/555/88 $%6%<%N>NO 9 9,00DK !2!7!78 9 9 9sAB$--B$$B-N)r6None)rkrr6r()r6zScriptRequest | None)r6r<) rrrr7rergrmrorqrrrr_r_s$ ' 1HQT)9V9rr_)r'r&r)r(r6r()rSr rTr r6r ) __future__rra dataclassesrrrenumrtypingrr streamlitr streamlit.proto.Common_pb2r rM streamlit.proto.WidgetStates_pb2r r streamlit.proto.ClientState_pb2rrrr<rEr]r_rrrr{s#11&LF;   $   0 $     D  D" D  D6#61D66rZ9Z9r