&eem~UddlmZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZm Z dd l!m"Z"ddl#m$Z$m%Z%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6m7Z7m8Z8m9Z9m:Z:ddl;mm?Z?ddl@mAZAddlBmCZCddlDmEZEe rddlFmGZGdZHdeId<eeJZKdeId <Gd!d"eLZMed#$Gd%d&ZNGd'd(e ZOGd)d*eZPGd+d,ZQdS)-) annotationsN) dataclassfield)Enum) TYPE_CHECKING AwaitableDict NamedTupleOptionalTupleType)Final)config) get_logger)BackMsg) ForwardMsg) AppSession)get_data_cache_stats_provider!get_resource_cache_stats_provider)LocalDiskCacheStorageManager)ForwardMsgCachecreate_reference_msgpopulate_hash_if_needed) _mem_caches)MediaFileManager)MediaFileStorage)MemorySessionStorage)is_cacheable_msg) ScriptData) ScriptCache)ActiveSessionInfo SessionClientSessionClientDisconnectedErrorSessionManagerSessionStorage)SCRIPT_RUN_WITHOUT_ERRORS_KEYSessionStateStatProvider) StatsManager)UploadedFileManager)WebsocketSessionManager)LocalSourcesWatcher)CacheStorageManager<rSCRIPT_RUN_CHECK_TIMEOUTLOGGERceZdZdZdS)RuntimeStoppedErrorz;Raised by operations on a Runtime instance that is stopped.N)__name__ __module__ __qualname____doc__9lib/python3.11/site-packages/streamlit/runtime/runtime.pyr1r1KsEEEEr7r1T)frozenceZdZUdZded<ded<ded<ded <ee Zd ed <eZ d ed<ee Z ded<dZ ded<dS) RuntimeConfigz$Config options for StreamlitRuntime.str script_path Optional[str] command_linermedia_file_storager)uploaded_file_manager)default_factoryr,cache_storage_managerzType[SessionManager]session_manager_classr%session_storageFboolis_helloN) r2r3r4r5__annotations__rrrCr*rDrrErGr6r7r8r;r;Os.. )(((/...274222 3JIIII',eZ!dKd?Z"dS)VRuntimeNzOptional[Runtime] _instancereturnc<|jtd|jS)znReturn the singleton Runtime instance. Raise an Error if the Runtime hasn't been created yet. NzRuntime hasn't been created!)rZ RuntimeErrorclss r8instancezRuntime.instances# = =>> >}r7rFc|jduS)a2True if the singleton Runtime instance has been created. When a Streamlit app is running in "raw mode" - that is, when the app is run via `python app.py` instead of `streamlit run app.py` - the Runtime will not exist, and various Streamlit functions need to adapt. N)rZr^s r8existszRuntime.existss}D((r7rr;ctjtd|t_d|_d|_|j|_|j|_tj |_ t|_ |j|_t!|j|_|j|_t+|_|j|j|j|j|j|_t7|_|jt=|jt?|jt@|j|j |j|j|jtC|jdS)a Create a Runtime instance. It won't be started yet. Runtime is *not* thread-safe. Its public methods are generally safe to call only on the same thread that its event loop runs on. Parameters ---------- config Config options. Nz Runtime instance already exists!)storage)rErA script_cachemessage_enqueued_callback)"rYrZr] _async_objs_loop_coroutine_taskr=_main_script_pathrG _is_hellorJrK_stater_message_cacherA_uploaded_file_mgrrr@_media_file_mgrrC_cache_storage_managerr _script_cacherDrE_enqueued_some_message _session_mgrr( _stats_mgrregister_providerrrrr')selfrs r8__init__zRuntime.__init__s   (ABB B 48CG!!'!3"* .//"(">/8QRRR&,&B#(]]8F8"2"&"9+&*&A    '.. ))*G*I*IJJJ ))*K*M*MNNN ))+666 ))$*=>>> ))$*ABBB ))*B4CT*U*UVVVVVr7rJc|jSN)rkrus r8statez Runtime.states {r7rc|jSrx)rlrys r8 message_cachezRuntime.message_caches ""r7r)c|jSrx)rmrys r8uploaded_file_mgrzRuntime.uploaded_file_mgrs &&r7r,c|jSrx)rorys r8rCzRuntime.cache_storage_managers **r7rc|jSrx)rnrys r8media_file_mgrzRuntime.media_file_mgrs ##r7r(c|jSrx)rsrys r8 stats_mgrzRuntime.stats_mgrs r7Awaitable[None]c4|jS)z?A Future that completes when the Runtime's run loop has exited.)_get_async_objsrWrys r8rWzRuntime.stoppeds##%%--r7 session_idr<Optional[SessionClient]cL|j|}|dS|jS)zGet the SessionClient for the given session_id, or None if no such session exists. Notes ----- Threading: SAFE. May be called on any thread. N)rrget_active_session_infoclientrur session_infos r8 get_clientzRuntime.get_clients.(@@LL  4""r7Nonec Kttjtjtjtjtjtj}||_tj|d|_|j d{VdS)aStart the runtime. This must be called only once, before any other functions are called. When this coroutine returns, Streamlit is ready to accept new sessions. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. )rRrSrTrUrVrWzRuntime.loop_coroutine)nameN) rQasyncioget_running_loopEventFuturerg create_task_loop_coroutinerhrVru async_objss r8startz Runtime.starts".00moo"=??"=??N$$N$$    &$+$7  " ")A% % % !          r7crfd}j|dS)zRequest that Streamlit close all sessions and stop running. Note that Streamlit won't stop running immediately. Notes ----- Threading: SAFE. May be called from any thread. cjtjtjfvrdStdtjjdS)NzRuntime stopping...) rkrJrNrOr/debug _set_staterSset)rrusr8stop_on_eventloopz'Runtime.stop..stop_on_eventloop/sc{|4l6JKKK LL. / / / OOL1 2 2 2  $ $ & & & & &r7N)rrRcall_soon_threadsafe)rurrs` @r8stopz Runtime.stop$sU))++  ' ' ' ' ' ' 112CDDDDDr7c6|j|S)zTrue if the session_id belongs to an active session. Notes ----- Threading: SAFE. May be called on any thread. )rris_active_session)rurs r8rzRuntime.is_active_session9s 22:>>>r7rr" user_infoDict[str, Optional[str]]existing_session_idr>session_id_overridec|r |r Jd|jtjtjfvrt d|jd|j|t|j|j |||}| tj | j |S)aCreate a new session (or connect to an existing one) and return its unique ID. Parameters ---------- client A concrete SessionClient implementation for communicating with the session's client. user_info A dict that contains information about the session's user. For now, it only (optionally) contains the user's email address. { "email": "example@example.com" } existing_session_id The ID of an existing session to reconnect to. If one is not provided, a new session is created. Note that whether the Runtime's SessionManager supports reconnecting to an existing session depends on the SessionManager that this runtime is configured with. session_id_override The ID to assign to a new session being created with this method. Setting this can be useful when the service that a Streamlit Runtime is running in wants to tie the lifecycle of a Streamlit session to some other session-like object that it manages. Only one of existing_session_id and session_id_override should be set. Returns ------- str The session's unique string ID. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. zFOnly one of existing_session_id and session_id_override should be set!zCan't connect_session (state=))r script_datarrr)rkrJrNrOr1rrconnect_sessionrrirjrrMrrTr)rurrrrrs r8rzRuntime.connect_sessionBsV  T$7 T T S T T T ;<0,2FG G G%&Tdk&T&T&TUU U&66"4#94>JJ 3 3 7    CDDD -11333r7chtd|||||S)zCreate a new session (or connect to an existing one) and return its unique ID. Notes ----- This method is simply an alias for connect_session added for backwards compatibility. z:create_session is deprecated! Use connect_session instead.)rrrr)r/warningr)rurrrrs r8create_sessionzRuntime.create_sessions@ STTT## 3 3 $   r7c|j|}|r9|j|j|j||dS)a9Close and completely shut down a session. This differs from disconnect_session in that it always completely shuts down a session, permanently losing any associated state (session state, uploaded files, etc.). This function may be called multiple times for the same session, which is not an error. (Subsequent calls just no-op.) Parameters ---------- session_id The session's unique ID. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. N)rrget_session_inforlremove_refs_for_sessionsession close_session_on_session_disconnectedrs r8rzRuntime.close_sessionsk&(99*EE  8   7 7 8L M M M   + +J 7 7 7 %%'''''r7c|j|}|r9|j|j|j||dS)aZDisconnect a session. It will stop producing ForwardMsgs. Differs from close_session because disconnected sessions can be reconnected to for a brief window (depending on the SessionManager/SessionStorage implementations used by the runtime). This function may be called multiple times for the same session, which is not an error. (Subsequent calls just no-op.) Parameters ---------- session_id The session's unique ID. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. N)rrrrlrrdisconnect_sessionrrs r8rzRuntime.disconnect_sessionsm&(@@LL  =   7 7 8L M M M   0 0 < < < %%'''''r7msgrc|jtjtjfvrt d|jd|j|}|td|dS|j |dS)a+Send a BackMsg to an active session. Parameters ---------- session_id The session's unique ID. msg The BackMsg to deliver to the session. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. zCan't handle_backmsg (state=rNz3Discarding BackMsg for disconnected session (id=%s)) rkrJrNrOr1rrrr/rrhandle_backmsg)rurrrs r8rzRuntime.handle_backmsgs ;<0,2FG G G%&ST[&S&S&STT T(@@LL   LLEz    F++C00000r7exc BaseExceptionc|jtjtjfvrt d|jd|j|}|td|dS|j |dS)a.Handle an Exception raised during deserialization of a BackMsg. Parameters ---------- session_id The session's unique ID. exc The Exception. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. z6Can't handle_backmsg_deserialization_exception (state=rNz=Discarding BackMsg Exception for disconnected session (id=%s)) rkrJrNrOr1rrrr/rrhandle_backmsg_exception)rurrrs r8(handle_backmsg_deserialization_exceptionz0Runtime.handle_backmsg_deserialization_exceptions ;<0,2FG G G%WWWW (@@LL   LLO    F55c:::::r7Tuple[bool, str]cbK|jtjtjtjfvrdSdS)N)Tok)F unavailable)rkrJrKrNrOrys r8is_ready_for_browser_connectionz'Runtime.is_ready_for_browser_connections9 ;   !     :##r7c Ktt|j|j|j|j|jt|jddi} |dtj }t|j vrftj |z tkrGtjdd{Vt|j vrtj |z tkGt|j vr |dS|j t}|rdnd}||f|S#|wxYw) aJLoad and execute the app's script to verify it runs without an error. Returns ------- (True, "ok") if the script completes without error, or (False, err_msg) if the script raises an exception. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. emailz test@test.com)rrArerflocal_sources_watcherrNg?)Ftimeoutrerror)rrrirjrmrprqr+ request_reruntime perf_counterr& session_stater.rsleepshutdown)rurnowrrs r8does_script_run_without_errorz%Runtime.does_script_run_without_errorsp"4#94>JJ"&"9+&*&A"5d6L"M"M0       ! !$ ' ' '#%%C-W5JJJ&((3.2JJJmC(((((((((.W5JJJ&((3.2JJJ-G4III'         &'DEB)$$'Cs7       G      sB*EEE new_statecVtd|j|||_dS)NzRuntime state: %s -> %s)r/rrk)rurs r8rzRuntime._set_state>s& . YGGG r7c,K|} |jtjkr |tjn-|jtjkrntd|j|j d|j s|jtjkr{tj tj|j tj|j ftjd{Vn|jtjkr|j|jD]}|j}|D]h} |||n4#t0$r'|j|jjYnwxYwtjdd{Vitjdd{Vnntj tj|j tj|j ftjd{V|j |jD]&}|j|jj'|tj|j ddS#t@$rR}|j!|tEj#tH%dYd}~dSd}~wwxYw)zThe main Runtime loop. This function won't exit until `stop` is called. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. zBad Runtime state at start: N) return_whenrg{Gz?zJ Please report this bug at https://github.com/streamlit/streamlit/issues. )&rrkrJrKrrLrMr]rV set_resultrSis_setrwaitrrTFIRST_COMPLETEDrUclearrrlist_active_sessionsrflush_browser_queue _send_messager#ridr list_sessionsrrOrW Exception set_exception traceback print_excr/info)ruractive_session_infomsg_listrres r8rzRuntime._loop_coroutineBs))++ H {l222 BCCCC KKK"#O$+#O#OPPP   ) )$ / / / *1133* ;,"DDD",#/ 0D0I0I0K0KLL#/ 0I0N0N0P0PQQ%,$; [L$OOO-33555/3/@/U/U/W/W 3 3+#6#>#R#R#T#T#+ 3 3C" $ 2 23F L L L L#A""" $ 1 D D$7$?$B!"!"!"!"!"" #*-"2"222222222 3"-----------l+J,@,E,E,G,GHH+J,E,J,J,L,LMM!( 7 I!*1133* Z!% 1 ? ? A A I I !// 0D0GHHHH OOL0 1 1 1   ) )$ / / / / /      , ,Q / / /   ! ! ! KK          s>F L7%F<;L7<.G-*L7,G--EL77 NANNrr!rct||j_|}|jjrt||j||j|jr/t d|j t|}t d|j |j ||j|j| ddkrw|jtjkrbt dt#jd|xjdz c_|j|j|j|j|dS) aSend a message to a client. If the client is likely to have already cached the message, we may instead send a "reference" message that contains only the hash of the message. Parameters ---------- session_info : ActiveSessionInfo The ActiveSessionInfo associated with websocket msg : ForwardMsg The message to send to the client Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. z$Sending cached message ref (hash=%s)zCaching message (hash=%s)typescript_finishedzYScript run finished successfully; removing expired entries from MessageCache (max_age=%s)zglobal.maxCachedMessageAgeN)rmetadata cacheablerrlhas_message_referencerscript_run_countr/rhashr add_message WhichOneofrrFINISHED_SUCCESSFULLYr get_option"remove_expired_entries_for_sessionrwrite_forward_msg)rurr msg_to_sends r8rzRuntime._send_messagesw$"2#!6!6  < !  #C ( ( ("88\)<+H 8  CSXNNN2377 LL4ch ? ? ?   + +\)<+H    NN6 " "&7 7 7#z'GGG LL!">??      ) )Q . ) )   B B$l&C    --k:::::r7cv|}|j|jjdS)aCallback called by AppSession after the AppSession has enqueued a message. Sets the "needs_send_data" event, which causes our core loop to wake up and flush client message queues. Notes ----- Threading: SAFE. May be called on any thread. N)rrRrrUrrs r8rqzRuntime._enqueued_some_messages8))++ 11*2K2OPPPPPr7rQc<|jtd|jS)zpReturn our AsyncObjects instance. If the Runtime hasn't been started, this will raise an error. NzRuntime hasn't started yet!)rgr]rys r8rzRuntime._get_async_objss%   #<== =r7c|jtjkri|jdkrN|j|tj dSdSdS)zlSet the runtime state to NO_SESSIONS_CONNECTED if the last active session was disconnected. rN) rkrJrMrrnum_active_sessionsrrTrrrLrys r8rz Runtime._on_session_disconnectedsy K<F F F!55771<<  " " 1 7 7 9 9 9 OOL> ? ? ? ? ? G F<rr>r[r<)rr<r[r)rr<rrr[r)rr<rrr[r)r[r)rrJr[r)rr!rrr[r)r[rQ)#r2r3r4rZrH classmethodr`rbrvpropertyrzr|r~rCrrrWrrrrrrrrrrrrrrrrqrrr6r7r8rYrYs#'I''''[)))[)/W/W/W/WbX###X#'''X'+++X+$$$X$X...X. # # # #!!!!:EEEE*????.2-1 ;;;;;B.2-1      ,((((2((((@11114;;;;>$$$X$****X    TTTTl9;9;9;9;v Q Q Q Q     @ @ @ @ @ @r7rY)R __future__rrrr dataclassesrrenumrtypingrrr r r r r typing_extensionsr streamlitrstreamlit.loggerrstreamlit.proto.BackMsg_pb2rstreamlit.proto.ForwardMsg_pb2rstreamlit.runtime.app_sessionrstreamlit.runtime.cachingrr:streamlit.runtime.caching.storage.local_disk_cache_storager#streamlit.runtime.forward_msg_cacherrr(streamlit.runtime.legacy_caching.cachingr$streamlit.runtime.media_file_managerr$streamlit.runtime.media_file_storager(streamlit.runtime.memory_session_storagerstreamlit.runtime.runtime_utilrstreamlit.runtime.script_datar+streamlit.runtime.scriptrunner.script_cacher !streamlit.runtime.session_managerr!r"r#r$r%streamlit.runtime.stater&r'streamlit.runtime.statsr('streamlit.runtime.uploaded_file_managerr)+streamlit.runtime.websocket_session_managerr*streamlit.watcherr+!streamlit.runtime.caching.storager,r.rHr2r/rr1r;rJrQrYr6r7r8rs#"""""" ((((((((TTTTTTTTTTTTTTTTTT######''''''//////555555444444 A@@@@@AAAAAAAAAAAAIIIIII;;;;;;444444CCCCCC100000GGGGGGOOOOOO111111FEEEEEE#%$$$$ 8$$$$$$FFFFF)FFF $>4""""":"""0a @a @a @a @a @a @a @a @a @a @r7