mxirUddlmZddlZddlZddlmZmZmZmZm Z ddl m Z ddl m Z mZmZddlmZmZmZmZmZddlmZmZddlmZdd lmZmZdd lmZ dd lm!Z"dd l#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4m5Z5m6Z6m7Z7erddl8m9Z9dZ:de;d<e:dZe>e=zZ?de;d<e Gdd ee@efZAd*d!ZBe Gd"d#ZCe Gd$d%ZDd+d&ZE d,d'ZFe Gd(d)e6ZGy)-) annotationsN)IteratorKeysViewMappingMutableMappingSequence)deepcopy) dataclassfieldreplace) TYPE_CHECKINGAnyFinal TypeAliascast)configutil)get_dg_singleton_instance)StreamlitAPIExceptionUnserializableSessionStateError) WidgetState) WidgetStates)get_script_run_ctx) RegisterWidgetResultTValueFieldName WidgetArgsWidgetCallbackWidgetMetadatais_array_value_field_name is_element_idis_keyed_element_id)apply_presenter) QueryParams)CACHE_MEMORY_FAMILY CacheStat StatsProvidergroup_cache_stats)SessionManagerz$$STREAMLIT_INTERNAL_KEYrSTREAMLIT_INTERNAL_KEY_PREFIX_SCRIPT_RUN_WITHOUT_ERRORSSCRIPT_RUN_WITHOUT_ERRORS_KEYT)frozenceZdZUdZded<y) Serializedz:A widget value that's serialized to a protobuf. Immutable.WidgetStateProtovalueN__name__ __module__ __qualname____doc____annotations__\/home/htdocs/ttos/venv/lib/python3.12/site-packages/streamlit/runtime/state/session_state.pyr/r/Bs D r9r/ceZdZUdZded<y)Valuez0A widget value that's not serialized. Immutable.rr1Nr2r8r9r:r<r<Is : Jr9r<rWStateceZdZUdZeeZded<eeZded<ddZ ddZ dd Z dd Z dd Z dd Zdd Zd dZd!dZd"dZd#dZd$dZd%dZ d&dZd'dZd(dZd)dZy)*WStateszA mapping of widget IDs to values. Widget values can be stored in serialized or deserialized form, but when values are retrieved from the mapping, they'll always be deserialized. default_factoryzdict[str, WState]stateszdict[str, WidgetMetadata[Any]]widget_metadatac,tj|SNrrepr_selfs r:__repr__zWStates.__repr__]zz$r9c||jj|}| t|t|tr |j S|j j|}| t|td|j jd}|r|j j|nd}t|rtd|j}n$|dk(rtjtd|}|j|}|jt!||t ||j|<|S)zReturn the value of the widget with the given key. If the widget's value is currently stored in serialized form, it will be deserialized first. Nrr1r json_valuestr) value_type)rBgetKeyError isinstancer<r1rCr WhichOneof__getattribute__r datajsonloads deserializerset_widget_metadatar )rIkwstatemetadatavalue_field_namer1 deserializeds r: __getitem__zWStates.__getitem__`s' # >1+  fe $<<  ''++A.  1+   LL # #G ,    LL ) )*: ;  %%5 6&++E  -JJtE512E,,U3    +  |, Ar9c"||j|<yrErBrIrZvs r: __setitem__zWStates.__setitem__s Ar9c|j|=yrErarIrZs r: __delitem__zWStates.__delitem__s KKNr9c,t|jSrE)lenrBrHs r:__len__zWStates.__len__s4;;r9c#8K|jEd{y7wrErarHs r:__iter__zWStates.__iter__s;;s c,t|jSrE)rrBrHs r:keysz WStates.keyss $$r9c6|Dchc] }|||f c}Scc}wrEr8rfs r:itemsz WStates.itemss&*+DG +++sc2|Dchc]}|| c}Scc}wrEr8)rIwids r:valueszWStates.valuess%)*cS ***s c|jj|j|jj|jy)zCopy all widget values and metadata from 'other' into this mapping, overwriting any data in this mapping that's also present in 'other'. N)rBupdaterCrIothers r:ruzWStates.updates4 5<<( ##E$9$9:r9c4t|||j<y)zGSet a widget's serialized value, overwriting any existing value it has.N)r/id)rI widget_states r:set_widget_from_protozWStates.set_widget_from_protos *< 8\__r9c t|||<y)zISet a widget's deserialized value, overwriting any existing value it has.N)r<rbs r:set_from_valuezWStates.set_from_values(Qr9c6||j|j<y)zBSet a widget's metadata, overwriting any existing metadata it has.N)rCry)rI widget_metas r:rYzWStates.set_widget_metadatas/:[^^,r9c |jjDcic].\}}t|jj |||s||0c}}|_ycc}}w)z&Remove widget state for stale widgets.N)rBrp_is_stale_widgetrCrP)rIactive_widget_idsfragment_ids_this_runrZrcs r:remove_stale_widgetszWStates.remove_stale_widgetssZ ))+ 1#$$((+!% qD   s3Ac|jj|}|yt|tr |jS|j j|}|yt }||_|j}|j|j}t|r)t||}|jj||S|dvr"t||tj ||S|dk(r|j"j%||S|dk(r|j&j%||S|dk(r|j(j%||S|| t||||S)zGet the serialized value of the widget with the given id. If the widget doesn't exist, return None. If the widget exists but is not in serialized form, it will be serialized first. N>rMjson_trigger_valuefile_uploader_state_valuestring_trigger_valuechat_input_value)rBrPrRr/r1rCr0ryrO serializerr getattrrUextendsetattrrVdumpsrCopyFromrr)rIrZitemr\widgetr serializedarrs r:get_serializedzWStates.get_serializedse{{q! < dJ ':: ''++A.  !# ##((4 $U +&%(C HHOOJ '  : : FE4::j#9 : 1 1  , , 5 5j A , ,  ' ' 0 0 < ( (  # # , ,Z 8  :#9 FE: . r9c|jDcgc]$}|j|r|j|&}}td|Scc}w)GReturn a list of serialized widget values for each widget with a value.list[WidgetStateProto])rBrr)rI widget_idrBs r:as_widget_stateszWStates.as_widget_statessS"[[ ""9-    *  ,f55  s)Ac(|jj|}|td|d|j}|y|jxsd}|j xsi}t }|r#|jd|_||i|d|_y||i|y)zCall the given widget's callback and return the callback's return value. If the widget has no callback, return None. If the widget doesn't exist, raise an Exception. NzWidget z not found.r8TF) rCrP RuntimeErrorcallback callback_argscallback_kwargsr fragment_idin_fragment_callback)rIrr\rargskwargsctxs r: call_callbackzWStates.call_callbacks ''++I6   ;?@ @$$   %%+))/R " 8''3'+C $ d %f %',C $ d %f %r9NreturnrN)rZrNrr)rZrNrcr=rNone)rZrNrrrint)rz Iterator[str])rz KeysView[str])rzset[tuple[str, Any]])rzset[Any])rwr?rr)rzr0rr)rZrNrcrrr)rWidgetMetadata[Any]rr)rset[str]rlist[str] | Nonerr)rZrNrzWidgetStateProto | Nonerr)rrNrr)r3r4r5r6r dictrBr7rCrJr_rdrgrjrlrnrprsrur{r}rYrrrrr8r9r:r?r?Ss !&d ;F ;6;D6QO3Q 1f  %,+;9; #  0    .`6&r9r?cd|dS)Nzst.session_state has no key "z". Did you forget to initialize it? More info: https://docs.streamlit.io/develop/concepts/architecture/session-state#initializationr8keys r:_missing_key_error_messager"s 'u-j kr9ceZdZUdZeeZded<eeZded<ddZ ddZ ddZ e dd Z dd Zddd Zdd ZddZddZddZy ) KeyIdMappera#A mapping of user-provided keys to element IDs. It also maps element IDs to user-provided keys so that this reverse mapping does not have to be computed ad-hoc. All built-in dict-operations such as setting and deleting expect the key as the argument, not the element ID. r@dict[str, str]_key_id_mapping_id_key_mappingc||jvSrE)rrIrs r: __contains__zKeyIdMapper.__contains__5sd****r9c@||j|<||j|<yrErrrIrrs r:rdzKeyIdMapper.__setitem__8s"$-S!*-Y'r9c&|j|yrE)deleters r:rgzKeyIdMapper.__delitem__<s Cr9c|jSrErrHs r:id_key_mappingzKeyIdMapper.id_key_mapping?s###r9cp||_|jDcic]\}}|| c}}|_ycc}}wrE)rrpr)rIkey_id_mappingrZrcs r:set_key_id_mappingzKeyIdMapper.set_key_id_mappingCs1-1?1E1E1GHA1HHs 2Nc:|jj||SrE)rrP)rIrdefaults r:get_id_from_keyzKeyIdMapper.get_id_from_keyGs##''W55r9c |j|SrErrIrs r:get_key_from_idzKeyIdMapper.get_key_from_idJs##I..r9c|jj|j|jj|jyrE)rrurrvs r:ruzKeyIdMapper.updateMs6 ##E$9$9: ##E$9$9:r9cl|jj|jjyrE)rclearrrHs r:rzKeyIdMapper.clearQs& ""$ ""$r9cV|j|}|j|=|j|=yrErrs r:rzKeyIdMapper.deleteUs.((-   %   +r9rrNrbool)rrNrrrrrrNrr)rr)rrrrrE)rrNr str | Nonerr)rrNrrN)rwrrrrr)r3r4r5r6r rrr7rrrdrgpropertyrrrrrurrr8r9r:rr)si',D&AO^A&+D&AO^A+.$$I6/;%,r9rceZdZUdZeeZded<eeZded<ee Z ded<ee Z ded <ee Zd ed <d.d Zd/d Zd/dZed0dZd1dZd2dZd3dZd4dZd5dZd6dZd7dZd8dZd9dZd:dZd;dZd/dZ ddZ#d?d Z$d/d!Z%d@d"Z&dAd#Z'dBd$Z(dCd%Z)dDd&Z*dEd'Z+ dFd(Z,dGd)Z- dH dId+Z.d/d,Z/d/d-Z0y*)J SessionStateaSessionState allows users to store values that persist between app reruns. Example ------- >>> if "num_script_runs" not in st.session_state: ... st.session_state.num_script_runs = 0 >>> st.session_state.num_script_runs += 1 >>> st.write(st.session_state.num_script_runs) # writes 1 The next time your script runs, the value of st.session_state.num_script_runs will be preserved. >>> st.session_state.num_script_runs += 1 >>> st.write(st.session_state.num_script_runs) # writes 2 r@dict[str, Any] _old_state_new_session_stater?_new_widget_stater_key_id_mapperr$ query_paramsc,tj|SrErFrHs r:rJzSessionState.__repr__~rKr9c|D]} |||j|<|jj|jjy#t$rYXwxYw)zCopy all current session_state and widget_state values into our _old_state dict, and then clear our current session_state and widget_state. N)rrQrrr)rI key_or_wids r:_compact_statezSessionState._compact_statesg  J .2:.> +  %%' $$&   sA AAc|jj|jj|jj|jjy)z;Reset self completely, clearing all current and old values.N)rrrrrrHs r:rzSessionState.clearsH  %%' $$& !!#r9c|jj}i}|jD]G}t|st |s ||||<"t |s.t |r: ||}||||<I|S#t $rYWwxYw)zVThe combined session and widget state, excluding keyless widgets and internal widgets.)rr_keysr!_is_internal_keyr"rQ)rI wid_key_mapstaterZrs r:filtered_statezSessionState.filtered_states))88 "  A #,| | td| |j|S| |j|S| |j|S| |j|St#t$rYGwxYw#t$rYDwxYw#t$rYAwxYw#t$rYtwxYw)zGet the value of an entry in Session State, using either the user-provided key or a widget id as appropriate for the internal dict being accessed. At least one of the arguments must have a value. zEuser_key and widget_id cannot both be None. This should never happen.) ValueErrorrrQrrrIrrs r:rzSessionState._getitems   1W    ..x88   --i88   y11   x00 ;           sEAA-A< B  A*)A*- A98A9< BB BBct}|N|jj|d}|j}|j}||vs||vrt d|d|d||j |<y)zSet the value of the session_state entry with the given user_key. If the key corresponds to a widget or form that's been instantiated during the current script run, raise a StreamlitAPIException instead. Nz`st.session_state.z0` cannot be modified after the widget with key `z` is instantiated.)rrrwidget_ids_this_runform_ids_this_runrr)rIrr1rr widget_idsform_idss r:rdzSessionState.__setitem__s !" ?++;;HdKI00J,,HJ&(h*>+( 3""*+=? -2)r9c|j|}||vs||vstt|||jvr |j|=||jvr |j|=||j vr|j j |||jvr |j|=||jvr|j|=yyrE)rrQrrrrrrrs r:rgzSessionState.__delitem__(s'', t yD05c:; ; $)) )'', $// !$ $%% %    & &s + .. .&&y1  ' * (r9c\|jD]}|jj|y)zHSet the value of all widgets represented in the given WidgetStatesProto.N)widgetsrr{)rI widget_statesrs r:set_widgets_from_protoz#SessionState.set_widgets_from_proto=s-"** @E  " " 8 8 ? @r9c|j|j|j||jy)zCalled by ScriptRunner before its script re-runs. Update widget data and call callbacks on widgets whose value changed between the previous and current script runs. N)_reset_triggersrr_call_callbacks)rIlatest_widget_statess r:on_script_will_rerunz!SessionState.on_script_will_rerunBs7   ##$89 r9cddlm}|jDcgc]I}|j|r6|jjj |x} |j |K}}|D]} |jj| t|jjj}|D]}|jjj |}|r |j7|jxsd}|j xsi}|j#|||||j$dk(s{|j'||||ycc}w#|$r'tjjdYwxYw)zFCall callbacks for widgets whose value changed or whose trigger fired.rRerunExceptionN0Calling st.rerun() within a callback is a no-op.r8rM)streamlit.runtime.scriptrunnerrr_widget_changedrCrPrrrmain_dgwarninglistrBrn callbacksrr_dispatch_trigger_callbacksrO_dispatch_json_change_callbacks)rIrrrr\&changed_widget_ids_for_single_callbackwidget_ids_to_processrrs r:r zSessionState._call_callbacksNsA --2 ##C(!33CCGGLL!!- 2 .2 : C &&44S9 !%T%;%;%B%B%G%G%I J( RC--==AA#FHx119))/RD--3F  , ,S(D& I""l244S(D&Q R)2 " )+33;;F sAE +E(E:9E:cRddlm}t}|r$|jd|_ ||i|d|_y ||i|y#|$r&t j jdY>> metadata.callbacks = {"submit": on_submit} The frontend can send a single event payload: >>> {"event": "submit", "value": "payload"} Or a list of event payloads to be processed in order: >>> [{"event": "edit", ...}, {"event": "submit", ...}] Parameters ---------- wid : str The widget ID. metadata : WidgetMetadata[Any] Metadata for the widget, including registered callbacks. args : WidgetArgs Positional arguments forwarded to the callback. kwargs : dict[str, Any] Keyword arguments forwarded to the callback. NzMapping[str, object]event) rrrrQrRrrrrPrNrr ) rIrrr\rrwidget_proto_stater^payloadspayload event_namecbs r:rz(SessionState._dispatch_trigger_callbackssN"33BB3G!   0 0 $#55c: ,-'(># Vgt,!%& 99"hfU  V 1 $#  $sC C$#C$c|jsy |jj|}|jj|}dd}||}||} |s| r|j | j z} | D chc](} | j| |j| k7s'| *} } | D]4} |jj| }|!|j ||||6yy#t$rd}YwxYwcc} w)a8Dispatch change callbacks for JSON-valued widgets. Computes a shallow diff between the new and old JSON maps and invokes callbacks for keys that changed or were added/removed. Parameters ---------- wid : str The widget ID. metadata : WidgetMetadata[Any] Metadata for the widget, including registered callbacks. args : WidgetArgs Positional arguments forwarded to the callback. kwargs : dict[str, Any] Keyword arguments forwarded to the callback. Nct|tsiStd|}t|j dhk(r,|j d}t|tr t|St|S)Nrr1)rRrrrrnrP)objr1s r:unwrapz.unwraps^c4( '-C388:7)+(eT*;&9 r9)r*objectrzdict[str, object])rrrPrQrrnr )rIrrr\rrnew_valold_valr+new_mapold_mapall_keysrZ changed_keysrr's r:rz,SessionState._dispatch_json_change_callbackss.!!  ,,005G//%%c* // g||~ 6H'/T!7;;q>W[[QR^3SATLT# N''++C0>11"hfM N ' G *UsC,(C=+C=, C:9C:c||jj|}|jj|}||k7}|S)zuTrue if the given widget's value changed between the previous script run and the current script run. )rrPr)rIr new_value old_valuechangeds r:rzSessionState._widget_changeds=**..y9 OO'' 2 !Y.r9cF|j|j|y)aCalled by ScriptRunner after its script finishes running. Updates widgets to prepare for the next script run. Parameters ---------- widget_ids_this_run: set[str] The IDs of the widgets that were accessed during the script run. Any widget state whose ID does *not* appear in this set is considered "stale" and will be removed. N)r _remove_stale_widgets)rIrs r:on_script_finishedzSessionState.on_script_finished%s  ""#67r9c|jD]y}|jjj|}|+|jdk(rt d|j|<S|jdvsbt d|j|<{|j D]g}|jjj|}|+|jdk(rd|j |<J|jdvsYd|j |<iy)z8Set all trigger values in our state dictionary to False.N trigger_valueF>rrr)rrCrPrOr<r)rIstate_idr\s r:r zSessionState._reset_triggers3s.. CH--==AA(KH#&&/97>qA%-- qD    sAB-cL|jjj|S)zBReturn the metadata for a widget id from the current widget state.)rrCrPrs r:_get_widget_metadataz!SessionState._get_widget_metadatags%%5599)DDr9cN|j}||jj|<y)zSet a widget's metadata.N)ryrrC)rIrCrs r:_set_widget_metadataz!SessionState._set_widget_metadataks##&& >q!DEEr9c"||j|<yrE)rrs r:_set_key_widget_mappingz$SessionState._set_key_widget_mapping|s(1H%r9cd|j}|j|||j||||vr?|||vr9|j}t |d}|j j ||td||}t |}|duxr|j|}t||S)aRegister a widget with the SessionState. Returns ------- RegisterWidgetResult[T] Contains the widget's current value, and a bool that will be True if the frontend needs to be updated with the current value. Nr) ryrArFrXr rr}rrr)rIr\rrrXinitial_widget_value widget_valuewidget_value_changeds r:register_widgetzSessionState.register_widgetsKK  !!(+    ( (H = D h&6($:N$00L#+L,>#?  " " 1 1)=Q R Ci1  -  (t3 8O8O 9 $L2FGGr9c. ||y#t$rYywxYw)NTF)rQrs r:rzSessionState.__contains__s'  I  s  NcFddlm}tdd||}t|giS)Nr)asizeofst_session_state) streamlit.vendor.pympler.asizeofrNr&r%)rI _family_namesrNstats r: get_statszSessionState.get_statss) =+R?$dV,,r9c|D]} tj||y#t$r&}dt||d|d}t ||d}~wwxYw)zVerify that everything added to session state can be serialized. We use pickleability as the metric for serializability, and test for pickleability by just trying it. z%Cannot serialize the value (of type `z`) of 'aV' in st.session_state. Streamlit has been configured to use [pickle](https://docs.python.org/3/library/pickle.html) to serialize session_state values. Please convert the value to a pickle-serializable type. To learn more about this behavior, see [our docs](https://docs.streamlit.io/knowledge-base/using-streamlit/serializable-session-state).N)pickler Exceptiontyper)rIrZeerr_msgs r:_check_serializablez SessionState._check_serializablesw  FA F T!W% F F;DaM?'RSQTU{{6g>AE Fs" A!A  AcPtjdr|jyy)zVerify that session state can be serialized, if the relevant config option is set. See `_check_serializable` for details. z&runner.enforceSerializableSessionStateN)r get_optionr[rHs r:maybe_check_serializablez%SessionState.maybe_check_serializables%   E F  $ $ & Gr9rr)rr)rr)rrNrr)rrNr1z Any | Nonerr)rz Iterator[Any]r)rrNrr)rrrrrr)rrNr1rrrr)rWidgetStatesProtorr)r r_rr) rrrrrrrrrr) rrrNr\rrrrrrr)rrNrr)rrrr)rrrr)rrNrWidgetMetadata[Any] | None)rCrrrr)rZrNrrN)rrNrrNrr)r\zWidgetMetadata[T]rrrzRegisterWidgetResult[T]rrErRzSequence[str] | Nonerzdict[str, list[CacheStat]])1r3r4r5r6r rrr7rr?rrrr$rrJrrrrrrrrlrjr_rrdrgrr r r rrrr9r r8r?rArCrrFrKrrTr[r^r8r9r:rr[s""'t!V >V&>V >V  >V  >V@6N 6N&6N 6N  6N  6Np 854 4EL 9F2%H)%H5?%H %HN59 -1 - # -F&'r9rc,|jtSrE) startswithr*rs r:rrs >>7 88r9cN|sy|j|vxs|xr|j|v S)NT)ryr)r\rrs r:rrs=   (( Y ! Wh&:&:BW&W r9c<eZdZUded<eddZ d ddZy) SessionStateStatProviderr) _session_mgrctfSrE)r%rHs r:stats_familiesz'SessionStateStatProvider.stats_familiess #%%r9Ncg}|jjD]N}|jj}|j }|j D]}|j |P|siStt|iSrE) rglist_active_sessionssession session_staterTrsrr%r()rIrRstats session_inform session_stats family_statss r:rTz"SessionStateStatProvider.get_statss"$ --BBD +L(00>>M)335M - 4 4 6 +  \* + + I$%6u%=>>r9)rz Sequence[str]rEra)r3r4r5r7rrirTr8r9r:rfrfs4   &&59?1? #?r9rf)rrNrrNr)r\r`rrrrrr)H __future__rrVrVcollections.abcrrrrrcopyr dataclassesr r r typingr rrrr streamlitrr$streamlit.delta_generator_singletonsrstreamlit.errorsrr streamlit.proto.WidgetStates_pb2rr0rr_7streamlit.runtime.scriptrunner_utils.script_run_contextrstreamlit.runtime.state.commonrrrrrrr r!r"$streamlit.runtime.state.presentationr#$streamlit.runtime.state.query_paramsr$streamlit.runtime.statsr%r&r'r(!streamlit.runtime.session_managerr)r*r7r,r/r<r=rNr?rrrrrrfr8r9r:rs# QQ11#JSLNV   A<@(BuA$%%?@u  $  $ J& & K&nS#X&K& K&\ .,., .,b x 'x ' x 'v9(, " ?}? ?r9