&elHddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z ddl mZddlmZmZddlZddlmZmZddlmZmZdd lmZdd lmZdd lmZmZdd l m!Z!dd l"m#Z#ddl$m%Z%m&Z&ddl'm(Z(ddl)m*Z*e rddl+m,Z,ee-Z.eGddeZ/edGddZ0edGddZ1edGddZ2edGddZ3e e2e3fZ4 eGddZ5eGd d!Z6 Gd"d#ej7Z8d0d+Z9d1d/Z:dS)2) annotationsN) dataclass) TYPE_CHECKINGAnyIteratorUnion)Message)Protocolruntime_checkable)runtimeutil)NONWIDGET_ELEMENTSWIDGETS) get_logger)Block)CachedStFunctionWarningCacheReplayClosureError) CacheType) update_hash)ScriptRunContextget_script_run_ctx)WidgetMetadata)HASHLIB_KWARGSDeltaGeneratorceZdZUded<dS)WidgetstridN__name__ __module__ __qualname____annotations__Olib/python3.11/site-packages/streamlit/runtime/caching/cached_message_replay.pyrr3s GGGGGr&rT)frozenc2eZdZUdZded<ded<ded<dS) WidgetMsgMetadatazEverything needed for replaying a widget and treating it as an implicit argument to a cached function, beyond what is stored for all elements. r widget_idr widget_valueWidgetMetadata[Any]metadataNr!r"r#__doc__r$r%r&r'r*r*8s?NNN!!!!!!r&r*c.eZdZUded<ded<ded<dS) MediaMsgData bytes | strmediarmimetypemedia_idNr r%r&r'r2r2Cs.MMMMMMMMr&r2cXeZdZUdZded<ded<ded<ded<dZd ed <dZd ed <dS) ElementMsgDatazAn element's message and related metadata for replaying that element's function call. widget_metadata is filled in if and only if this element is a widget. media_data is filled in iff this is a media element (image, audio, video). r delta_typer messageid_of_dg_called_onreturned_dgs_idNzWidgetMsgMetadata | Nonewidget_metadatazlist[MediaMsgData] | None media_data)r!r"r#r0r$r=r>r%r&r'r8r8JsmOOO04O4444,0J000000r&r8c.eZdZUded<ded<ded<dS) BlockMsgDatarr:rr;r<Nr r%r&r'r@r@[s3NNNr&r@c<eZdZUdZded<ded<ded<ded<d S) CachedResultz}The full results of calling a cache-decorated function, enough to replay the st functions called while executing it. rvaluez list[MsgData]messagesrmain_id sidebar_idNr/r%r&r'rBrBsCJJJLLLOOOOOr&rBc0eZdZUdZded<ded<dd Zd S)MultiCacheResultszWidgets called by a cache-decorated function, and a mapping of the widget-derived cache key to the final results of executing the function. zset[str] widget_idszdict[str, CachedResult]resultsctxr cache_typerreturnrcv|jfdt|jD}t||}|S)Nc,g|]}|v||fSr%r%).0widstates r' z.s.   "%C5LLS%* LLLr&) session_statesortedrI_make_widget_key)selfrKrL widget_values widget_keyrRs @r'get_current_widget_keyz(MultiCacheResults.get_current_widget_keysV!    )/)@)@   &mZ@@ r&N)rKrrLrrMr)r!r"r#r0r$rZr%r&r'rHrHsN$$$$r&rHceZdZdZd,dZd-dZejd.d Zd/dZ d0dZ d1dZ d2dZ d3d#Z ejd4d$Zd5d(Zd6d*Zd+S)7CachedMessageReplayContextzA utility for storing messages generated by `st` commands called inside a cached function. Data is stored in a thread-local object, so it's safe to use an instance of this class across multiple threads. rLrcg|_d|_g|_g|_g|_d|_g|_||_d|_dS)Nr) _cached_func_stack_suppress_st_function_warning_cached_message_stack_seen_dg_stack_most_recent_messages_registered_metadata _media_data _cache_type_allow_widgets)rWrLs r'__init__z#CachedMessageReplayContext.__init__sP<>-.*:<".046"@D!/1%#$r&rMrc*tj|SN)r repr_rWs r'__repr__z#CachedMessageReplayContext.__repr__sz$r&functypes.FunctionType allow_widgetsboolIterator[None]c#K|j||jg|jt |r|xjdz c_ dV|j|j|_|j|r|xjdzc_dSdS#|j|j|_|j|r|xjdzc_wxYw)zContext manager that should wrap the invocation of a cached function. It allows us to track any `st.foo` messages that are generated from inside the function for playback during cache retrieval. N)r^appendr`rasetrfpoprb)rWrmros r'calling_cached_functionz2CachedMessageReplayContext.calling_cached_functionsT &&t,,, "))"--- ""355)))  %   1 $   ) EEE  # ' ' ) ) ))-)C)G)G)I)ID &   # # % % % )##q(#### ) )  # ' ' ) ) ))-)C)G)G)I)ID &   # # % % % )##q(######s 0CA$D>r9 element_protor invoked_dg_id used_dg_idreturned_dg_idNonec6tjsdSt|jdkr|||}t |t rN|jrG|j}|js!t dtt|d|j}nd}|j } t|||||| } |jD] } |js|| | !g|_ d|_|jD]} | |dS)zRecord the element protobuf as having been produced during any currently executing cached functions, so they can be replayed any time the function's execution is skipped because they're in the cache. NrszRTrying to save widget message that wasn't registered. This should not be possible.)r.)r existslenr`select_dg_to_save isinstancerrrc_LOGGERerrorAttributeErrorr*rdr8rfrtraadd) rWr9rxryrzr{ id_to_saverQ widget_metar>element_msg_datamsgsss r'save_element_messagez/CachedMessageReplayContext.save_element_messagesb~  F t) * *a / /// zJJJ-00 #]5E ##&0)MMl)(/(A # )J-    2 2 2&2+*=KK 0111$(!$ " "A EE. ! ! ! ! " "r& block_protorc|||}|jD]&}|t|||'|jD]}||dSri)rr`rtr@rar)rWrryrzr{rrrs r'save_block_messagez-CachedMessageReplayContext.save_block_messageEs++M:FF . O OD KK [*nMM N N N N$ " "A EE. ! ! ! ! " "r& invoked_id acting_on_idcXt|jdkr||jdvr|S|S)aSelect the id of the DG that this message should be invoked on during message replay. See Note [DeltaGenerator method invocation] invoked_id is the DG the st function was called on, usually `st._main`. acting_on_id is the DG the st function ultimately runs on, which may be different if the invoked DG delegated to another one because it was in a `with` block. r)rra)rWrrs r'rz,CachedMessageReplayContext.select_dg_to_saveRs9 t" # #a ' 'LDCachedMessageReplayContext.suppress_cached_st_function_warningis **a/** ; EEE  . .! 3 . .5::::::  . .! 3 . .5::::::::s 7Adg'DeltaGenerator' st_func_namec|tvrdS|tvr |jdkrdSt|jdkr1|jdkr(|jd}||||dSdSdS)aIf appropriate, warn about calling st.foo inside @memo. DeltaGenerator's @_with_element and @_widget wrappers use this to warn the user when they're calling st.foo() from within a function that is wrapped in @st.cache. Parameters ---------- dg : DeltaGenerator The DeltaGenerator to publish the warning to. st_func_name : str The name of the Streamlit function that was called. Nrr)rrrfrr^r_ _show_cached_st_function_warning)rWrr cached_funcs r'%maybe_show_cached_st_function_warningz@CachedMessageReplayContext.maybe_show_cached_st_function_warningrs. - - - F 7 " "t':Q'>'> F t& ' '! + +0RVW0W0W1"5K  1 1"lK P P P P P , +0W0Wr&rc|5t|j||}||ddddS#1swxYwYdSri)rrre exception)rWrrres r'rz;CachedMessageReplayContext._show_cached_st_function_warnings 5 5 7 7  '(8, TTA LLOOO                  s,AAAN)rLr)rMr)rmrnrorprMrq) r9rrxr ryrrzrr{rrMr|) rrryrrzrr{rrMr|)rrrrrMr)r.r-rMr|)rr3r5rrrrMr|)rMrq)rrrrrMr|)rrrrrrnrMr|)r!r"r#r0rgrl contextlibcontextmanagerrwrrrrrrrrr%r&r'r\r\s$ % % % %    ))))*5"5"5"5"n " " " "    ----NNNN ;;;;QQQQ@      r&r\resultrLrrrnrMr|c ddlm}ddlm}i}tj||j<tj||j<t} |j D]}t|tr|j ||j j|d|j|jF|jD]>}t#jj|j|j|j?||j} | |j|j} t| |r | ||j<t|t8r1||j} | |j} | ||j<dS#t<$rt?||wxYw)aReplay the st element function calls that happened when executing a cache-decorated function. When a cache function is executed, we record the element and block messages produced, and use those to reproduce the DeltaGenerator calls, so the elements will appear in the web app even when execution of the function is skipped because the result was cached. To make this work, for each st function call we record an identifier for the DG it was effectively called on (see Note [DeltaGenerator method invocation]). We also record the identifier for each DG returned by an st function call, if it returns one. Then, for each recorded message, we get the current DG instance corresponding to the DG the message was originally called on, and enqueue the message using that, recording any new DGs produced in case a later st function call is on one of them. rr)register_widget_from_metadataN) streamlit.delta_generatorrstreamlit.runtime.state.widgetsrst_mainrEsidebarrFrrDrr8r=r.r9r>r get_instancemedia_file_mgrrr4r5r6r;_enqueuer:r<r@_blockKeyErrorr) rrLrrr returned_dgsrKmsgdatarmaybe_dgnew_dgs r'replay_cached_messagesrs&988888MMMMMM/1L#%8L &(jL"#   C?? ; ;C#~.. ;&211+4  >- #,..=AA J t}"#"89;;s~s{CCh77A8@L!45C.. ;!#"893;//4: S01+ ; ;, ???%j+>>>?s D E((Fwidgetslist[tuple[str, Any]]rctjdit}|D]}t||||S)zGenerate a key for the given list of widgets used in a cache-decorated function. Keys are generated by hashing the IDs and values of the widgets in the given list. md5)r)hashlibnewrr hexdigest)rrL func_hasher widget_id_vals r'rVrVsO +66~66K << M; ;;;;  " ""r&)rrBrLrrrnrMr|)rrrLrrMr); __future__rrr threadingtypes dataclassesrtypingrrrrgoogle.protobuf.messager typing_extensionsr r streamlitrr r streamlit.elementsrrstreamlit.loggerrstreamlit.proto.Block_pb2r&streamlit.runtime.caching.cache_errorsrr$streamlit.runtime.caching.cache_typer!streamlit.runtime.caching.hashingr1streamlit.runtime.scriptrunner.script_run_contextrrstreamlit.runtime.state.commonrstreamlit.utilrrrr!rrr*r2r8r@MsgDatarBrHlocalr\rrVr%r&r'rs#""""" !!!!!!666666666666++++++99999999########::::::::''''''++++++;:::::999999:99999))))))9888888 *X       X    $"""""""" $  $ 1 1 1 1 1 1 1 1  $  , -6r    2>zzzzzzzzz4?4?4?4?n # # # # # #r&