/_iFjUddlmZddlZddlZddlmZmZmZmZddl m Z ddl m Z m Z mZddlmZmZmZmZmZddlmZmZddlmZdd lmZmZdd lmZdd 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.m/Z/ddl0m1Z1ddl2m3Z3m4Z4m5Z5erddl6m7Z7dZ8de9d<e8dZ:de9d<e dGddZ;e dGddZefZ?d*d!Z@e Gd"d#ZAe Gd$d%ZBd+d&ZC d,d'ZDe Gd(d)e4ZEy)-) annotationsN)IteratorKeysViewMappingMutableMapping)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) CacheStatCacheStatsProvider group_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__d/var/www/html/land_sniper/venv/lib/python3.12/site-packages/streamlit/runtime/state/session_state.pyr-r-=s D r7r-ceZdZUdZded<y)Valuez0A widget value that's not serialized. Immutable.r r/Nr0r6r7r8r:r:Ds : Jr7r: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 r8__repr__zWStates.__repr__Xzz$r7c||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. Nrr/r json_valuestr) value_type)r@getKeyError isinstancer:r/rAr WhichOneof__getattribute__rdatajsonloads deserializerset_widget_metadatar )rGkwstatemetadatavalue_field_namer/ deserializeds r8 __getitem__zWStates.__getitem__[s' # >1+  fe $<<  ''++A.  1+   LL # #G ,    LL ) )*: ;  %%5 6&++E  -JJtE512E,,U3    +  |, Ar7c"||j|<yrCr@rGrXvs r8 __setitem__zWStates.__setitem__s Ar7c|j|=yrCr_rGrXs r8 __delitem__zWStates.__delitem__s KKNr7c,t|jSrC)lenr@rFs r8__len__zWStates.__len__s4;;r7c#8K|jEd{y7wrCr_rFs r8__iter__zWStates.__iter__s;;s c,t|jSrC)rr@rFs r8keysz WStates.keyss $$r7c6|Dchc] }|||f c}Scc}wrCr6rds r8itemsz WStates.itemss&*+DG +++sc2|Dchc]}|| c}Scc}wrCr6)rGwids r8valueszWStates.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)r@updaterArGothers r8rszWStates.updates4 5<<( ##E$9$9:r7c4t|||j<y)zGSet a widget's serialized value, overwriting any existing value it has.N)r-id)rG widget_states r8set_widget_from_protozWStates.set_widget_from_protos *< 8\__r7c t|||<y)zISet a widget's deserialized value, overwriting any existing value it has.N)r:r`s r8set_from_valuezWStates.set_from_values(Qr7c6||j|j<y)zBSet a widget's metadata, overwriting any existing metadata it has.N)rArw)rG widget_metas r8rWzWStates.set_widget_metadatas/:[^^,r7c |jjDcic].\}}t|jj |||s||0c}}|_ycc}}w)z&Remove widget state for stale widgets.N)r@rn_is_stale_widgetrArN)rGactive_widget_idsfragment_ids_this_runrXras r8remove_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>rKjson_trigger_valuefile_uploader_state_valuestring_trigger_valuechat_input_value)r@rNrPr-r/rAr.rwrM serializerrgetattrrSextendsetattrrTdumpsrCopyFromrr)rGrXitemrZwidgetr serializedarrs r8get_serializedzWStates.get_serializedse{{q! < dJ ':: ''++A.  !# ##((4 $U +&%(C HHOOJ '  : : FE4::j#9 : 1 1  , , 5 5j A , ,  ' ' 0 0 < ( (  # # , ,Z 8  :#9 FE: . r7c|jDcgc]$}|j|r|j|&}}td|Scc}w)GReturn a list of serialized widget values for each widget with a value.list[WidgetStateProto])r@rr)rG widget_idr@s r8as_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.r6TF) rArN RuntimeErrorcallback callback_argscallback_kwargsr fragment_idin_fragment_callback)rGrrZrargskwargsctxs r8 call_callbackzWStates.call_callbacks ''++I6   ;?@ @$$   %%+))/R " 8''3'+C $ d %f %',C $ d %f %r7NreturnrL)rXrLrr )rXrLrar;rNone)rXrLrrrint)rz Iterator[str])rz KeysView[str])rzset[tuple[str, Any]])rzset[Any])rur=rr)rxr.rr)rXrLrar rr)r}WidgetMetadata[Any]rr)rset[str]rlist[str] | Nonerr)rXrLrzWidgetStateProto | Nonerr)rrLrr)r1r2r3r4r dictr@r5rArHr]rbrerhrjrlrnrqrsryr{rWrrrrr6r7r8r=r=Ns !&d ;F ;6;D6QO3Q 1f  %,+;9; #  0    .`6&r7r=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#initializationr6keys r8_missing_key_error_messagers 'u-j kr7ceZdZUdZeeZded<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||jvSrC)rrGrs r8 __contains__zKeyIdMapper.__contains__0sd****r7c@||j|<||j|<yrCrrrGrrs r8rbzKeyIdMapper.__setitem__3s"$-S!*-Y'r7c&|j|yrC)deleters r8rezKeyIdMapper.__delitem__7s Cr7c|jSrCrrFs r8id_key_mappingzKeyIdMapper.id_key_mapping:s###r7cp||_|jDcic]\}}|| c}}|_ycc}}wrC)rrnr)rGkey_id_mappingrXras r8set_key_id_mappingzKeyIdMapper.set_key_id_mapping>s1-1?1E1E1GHA1HHs 2Nc:|jj||SrC)rrN)rGrdefaults r8get_id_from_keyzKeyIdMapper.get_id_from_keyBs##''W55r7c |j|SrCrrGrs r8get_key_from_idzKeyIdMapper.get_key_from_idEs##I..r7c|jj|j|jj|jyrC)rrsrrts r8rszKeyIdMapper.updateHs6 ##E$9$9: ##E$9$9:r7cl|jj|jjyrC)rclearrrFs r8rzKeyIdMapper.clearLs& ""$ ""$r7cV|j|}|j|=|j|=yrCrrs r8rzKeyIdMapper.deletePs.((-   %   +r7rrLrbool)rrLrr rrrrLrr)rr)rrrrrC)rrLr str | Nonerr)rrLrrL)rurrrrr)r1r2r3r4r rrr5rrrbrepropertyrrrrrsrrr6r7r8rr$si',D&AO^A&+D&AO^A+.$$I6/;%,r7rceZdZUdZeeZded<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-dHd*Z.d/d+Z/d/d,Z0y-)I 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|SrCrDrFs r8rHzSessionState.__repr__yrIr7c|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)rrOrrr)rG key_or_wids r8_compact_statezSessionState._compact_state}sg  J .2:.> +  %%' $$&   sA AAc|jj|jj|jj|jjy)z;Reset self completely, clearing all current and old values.N)rrrrrrFs r8rzSessionState.clearsH  %%' $$& !!#r7c|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!rO)rG wid_key_mapstaterXrs r8filtered_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.) ValueErrorrrOrrrGrrs r8rzSessionState._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)rGrr/rr widget_idsform_idss r8rbzSessionState.__setitem__s !" ?++;;HdKI00J,,HJ&(h*>+( 3""*+=? -2)r7c|j|}||vs||vstt|||jvr |j|=||jvr |j|=||j vr|j j |||jvr |j|=||jvr|j|=yyrC)rrOrrrrrrrs r8rezSessionState.__delitem__#s'', t yD05c:; ; $)) )'', $// !$ $%% %    & &s + .. .&&y1  ' * (r7c\|jD]}|jj|y)zHSet the value of all widgets represented in the given WidgetStatesProto.N)widgetsrry)rG widget_statesrs r8set_widgets_from_protoz#SessionState.set_widgets_from_proto8s-"** @E  " " 8 8 ? @r7c|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)rGlatest_widget_statess r8on_script_will_rerunz!SessionState.on_script_will_rerun=s7   ##$89 r7cddlm}|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.r6rK)streamlit.runtime.scriptrunnerr r_widget_changedrArNrrrmain_dgwarninglistr@rl callbacksrr_dispatch_trigger_callbacksrM_dispatch_json_change_callbacks)rGr rprZ&changed_widget_ids_for_single_callbackwidget_ids_to_processrrs r8rzSessionState._call_callbacksIsA --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) rrrrOrPrrrrNrLrr) rGrprZrrwidget_proto_stater\payloadspayload event_namecbs r8rz(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)Nrr/)rPrrrrlrN)objr/s r8unwrapz.unwraps^c4( '-C388:7)+(eT*;&9 r7)r(objectrzdict[str, object])rrrNrOrrlr)rGrprZrrnew_valold_valr)new_mapold_mapall_keysrX changed_keysrr%s r8rz,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. )rrNr)rGr new_value old_valuechangeds r8rzSessionState._widget_changeds=**..y9 OO'' 2 !Y.r7cF|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)rGrs r8on_script_finishedzSessionState.on_script_finished s  ""#67r7c|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)rrArNrMr:r)rGstate_idrZs r8rzSessionState._reset_triggers.s.. CH--==AA(KH#&&/97>qA%-- qD    sAB-cL|jjj|S)zBReturn the metadata for a widget id from the current widget state.)rrArNrs r8_get_widget_metadataz!SessionState._get_widget_metadatabs%%5599)DDr7cN|j}||jj|<y)zSet a widget's metadata.N)rwrrA)rGrArs r8_set_widget_metadataz!SessionState._set_widget_metadatafs##&& >q!DEEr7c"||j|<yrC)rrs r8_set_key_widget_mappingz$SessionState._set_key_widget_mappingws(1H%r7cd|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) rwr?rDrVrrr{rrr)rGrZrrrVinitial_widget_value widget_valuewidget_value_changeds r8register_widgetzSessionState.register_widgetzsKK  !!(+    ( (H = D h&6($:N$00L#+L,>#?  " " 1 1)=Q R Ci1  -  (t3 8O8O 9 $L2FGGr7c. ||y#t$rYywxYw)NTF)rOrs r8rzSessionState.__contains__s'  I  s  c:ddlm}tdd||}|gS)Nr)asizeofst_session_state) streamlit.vendor.pympler.asizeofrLr$)rGrLstats r8 get_statszSessionState.get_statss<+R?v r7c|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)rGrXeerr_msgs r8_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_optionrXrFs r8maybe_check_serializablez%SessionState.maybe_check_serializables%   E F  $ $ & Gr7Nrr)rr)rr)rrLrr)rrLr/z Any | Nonerr)rz Iterator[Any]r)rrLrr )rrrrrr )rrLr/r rrr)rWidgetStatesProtorr)r r\rr) rrrrrrrrrr) rprLrZrrrrrrr)rrLrr)rrrr)rrrr)rrLrWidgetMetadata[Any] | None)rArrrr)rXrLrrL)rrLrrLrr)rZzWidgetMetadata[T]rrrzRegisterWidgetResult[T]rrzlist[CacheStat])1r1r2r3r4r rrr5rr=rrrr#rrHrrrrrrrrjrhr]rrbrerr rrrrrr7rr6r=r?rArrDrIrrQrXr[r6r7r8rrVs""'t!V >V&>V >V  >V  >V@6N 6N&6N 6N  6N  6Np 854 4EL 9F2%H)%H5?%H %HNF&'r7rc,|jtSrC) startswithr(rs r8rrs >>7 88r7cN|sy|j|vxs|xr|j|v S)NT)rwr)rZrrs r8rrs=   (( Y ! Wh&:&:BW&W r7c eZdZUded<ddZy)SessionStateStatProviderr' _session_mgrcg}|jjD]7}|jj}|j |j 9t |SrC)rdlist_active_sessionssession session_staterrQr&)rGstats session_inforhs r8rQz"SessionStateStatProvider.get_statssX!# --BBD 4L(00>>M LL002 3 45!!r7Nr^)r1r2r3r5rQr6r7r8rcrcs   "r7rc)rrLrrLr)rZr]rrrrrr)F __future__rrTrScollections.abcrrrrcopyr dataclassesr r r typingr r rrr streamlitrr$streamlit.delta_generator_singletonsrstreamlit.errorsrr streamlit.proto.WidgetStates_pb2rr.rr\7streamlit.runtime.scriptrunner_utils.script_run_contextrstreamlit.runtime.state.commonrrrrrrrr r!$streamlit.runtime.state.presentationr"$streamlit.runtime.state.query_paramsr#streamlit.runtime.statsr$r%r&!streamlit.runtime.session_managerr'r(r5r*r-r:r;rLr=rrrrrrcr6r7r8rzs# GG11#JSLNV   A<NN@(BuA$%%?@u  $  $ J& & K&nS#X&K& K&\ .,., .,b s 's ' s 'l9(, " "1" "r7