U ARdpf@sUddlmZddlZddlZddlZddlZddlmZmZddl m Z ddl m Z m Z mZmZmZmZmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lm Z m!Z!dd l"m#Z#ddl$m%Z%m&Z&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5m6Z6m7Z7m8Z8m9Z9ddl:m;Z;mZ>ddl?m@Z@ddlAmBZBddlCmDZDe rddlEmFZFdZGdeHd<eeIZJdeHd<Gd d!d!eKZLed"d#Gd$d%d%ZMGd&d'd'e ZNGd(d)d)eZOGd*d+d+ZPdS),) annotationsN) dataclassfield)Enum) TYPE_CHECKING AwaitableDict NamedTupleOptionalTupleType)Finalconfig) 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)ActiveSessionInfo SessionClientSessionClientDisconnectedErrorSessionManagerSessionStorage)SCRIPT_RUN_WITHOUT_ERRORS_KEYSessionStateStatProvider) StatsManager)UploadedFileManager)WebsocketSessionManager)LocalSourcesWatcher)CacheStorageManager<r SCRIPT_RUN_CHECK_TIMEOUTLOGGERc@seZdZdZdS)RuntimeStoppedErrorz;Raised by operations on a Runtime instance that is stopped.N)__name__ __module__ __qualname____doc__r4r4=/tmp/pip-unpacked-wheel-b9et7o5g/streamlit/runtime/runtime.pyr/Ksr/T)frozenc@sZeZdZUdZded<ded<ded<eedZd ed <eZ d ed <ee dZ d ed<dS) RuntimeConfigz$Config options for StreamlitRuntime.str script_path Optional[str] command_linermedia_file_storage)default_factoryr+cache_storage_managerzType[SessionManager]session_manager_classr$session_storageN) r0r1r2r3__annotations__rrr>r)r?rr@r4r4r4r5r7Os  r7c@s eZdZdZdZdZdZdZdS) RuntimeStateINITIALNO_SESSIONS_CONNECTEDONE_OR_MORE_SESSIONS_CONNECTEDSTOPPINGSTOPPEDN)r0r1r2rCrDrErFrGr4r4r4r5rBis rBc@sBeZdZUdZded<ded<ded<ded<ded <ded <d S) AsyncObjectszContainer for all asyncio objects that Runtime manages. These cannot be initialized until the Runtime's eventloop is assigned. zasyncio.AbstractEventLoop eventloopz asyncio.Event must_stophas_connectionneed_send_datazasyncio.Future[None]startedstoppedN)r0r1r2r3rAr4r4r4r5rHqs rHc@seZdZUdZded<eddddZedddd Zd d d d Ze ddddZ e ddddZ e ddddZ e ddddZ e ddddZe ddddZe d dd!d"Zd#d$d%d&d'Zd#d(d%d)d*Zd(dd+d,Zd(dd-d.Zd#dd%d/d0Zd[d1d2d3d#d4d5d6Zd\d1d2d3d#d4d7d8Zd#d(d%d9d:Zd#d(d%d;d<Zd#d=d(d>d?d@Zd#dAd(dBdCdDZe dEddFdGZdEddHdIZdd(dJdKdLZd(ddMdNZdOdPd(dQdRdSZ d(ddTdUZ!dVddWdXZ"d(ddYdZZ#dS)]RuntimeNzOptional[Runtime] _instance)returncCs|jdkrtd|jS)znReturn the singleton Runtime instance. Raise an Error if the Runtime hasn't been created yet. NzRuntime hasn't been created!)rP RuntimeErrorclsr4r4r5instances zRuntime.instanceboolcCs |jdk S)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)rPrSr4r4r5existss zRuntime.existsr7rcCstjdk rtd|t_d|_d|_|j|_|jp4d|_t j |_ t |_ t|_|jj|jt|jd|_|j|_|j|j|j|jd|_t|_|jt|jt |jt!|j|j |j|j|jt"|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!)Zstorage)r@uploaded_file_managermessage_enqueued_callback)#rOrPrR _async_objs_loop_coroutine_taskr9_main_script_pathr; _command_linerBrC_stater_message_cacher(_uploaded_file_mgrZon_files_updatedconnect_on_files_updatedrr<_media_file_mgrr>_cache_storage_managerr?r@_enqueued_some_message _session_mgrr' _stats_mgrZregister_providerrrrr&)selfrr4r4r5__init__s2   zRuntime.__init__rBcCs|jSN)r_rir4r4r5statesz Runtime.statercCs|jSrk)r`rlr4r4r5 message_cacheszRuntime.message_cacher(cCs|jSrk)rarlr4r4r5uploaded_file_mgrszRuntime.uploaded_file_mgrr+cCs|jSrk)rerlr4r4r5r>szRuntime.cache_storage_managerrcCs|jSrk)rdrlr4r4r5media_file_mgrszRuntime.media_file_mgrr'cCs|jSrk)rhrlr4r4r5 stats_mgrszRuntime.stats_mgrzAwaitable[None]cCs |jS)z?A Future that completes when the Runtime's run loop has exited.)_get_async_objsrNrlr4r4r5rNszRuntime.stoppedr8zOptional[SessionClient]) session_idrQcCs|j|}|dkrdS|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)rgget_active_session_infoclient)rirs session_infor4r4r5 get_clients zRuntime.get_clientNonecCs|j|s|j|dS)zEvent handler for UploadedFileManager.on_file_added. Ensures that uploaded files from stale sessions get deleted. Notes ----- Threading: SAFE. May be called on any thread. N)rgis_active_sessionraZremove_session_filesrirsr4r4r5rcs zRuntime._on_files_updatedcsrtttttttd}||_tjdkrRtj| dd|_ nt| |_ |j IdHdS)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. )rIrJrKrLrMrN)rzRuntime.loop_coroutine)nameN) rHasyncioZget_running_loopEventZFuturer[sys version_info create_task_loop_coroutiner\rMri async_objsr4r4r5start s   z Runtime.startcs&fdd}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. cs:jtjtjfkrdStdtjjdS)NzRuntime stopping...) r_rBrFrGr.debug _set_staterJsetr4rrir4r5stop_on_eventloop:s   z'Runtime.stop..stop_on_eventloopN)rrrIcall_soon_threadsafe)rirr4rr5stop/s z Runtime.stopcCs |j|S)zTrue if the session_id belongs to an active session. Notes ----- Threading: SAFE. May be called on any thread. )rgryrzr4r4r5ryDszRuntime.is_active_sessionr!zDict[str, Optional[str]]r:)ru user_infoexisting_session_idrQcCsd|jtjtjfkr$td|jd|jj|t|j|j p:d||d}| 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" } Returns ------- str The session's unique string ID. Notes ----- Threading: UNSAFE. Must be called on the eventloop thread. zCan't connect_session (state=)rX)ru script_datarr)r_rBrFrGr/rgconnect_sessionrr]r^rrErrrKr)rirurrrsr4r4r5rMs zRuntime.connect_sessioncCstd|j|||dS)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.)rurr)r.warningr)rirurrr4r4r5create_sessionys zRuntime.create_sessioncCs|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)rg close_session_on_session_disconnectedrzr4r4r5rs zRuntime.close_sessioncCs|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)rgdisconnect_sessionrrzr4r4r5rs zRuntime.disconnect_sessionr)rsmsgrQcCsX|jtjtjfkr$td|jd|j|}|dkrHtd|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)) r_rBrFrGr/rgrtr.rsessionhandle_backmsg)rirsrrvr4r4r5rs zRuntime.handle_backmsg BaseException)rsexcrQcCsX|jtjtjfkr$td|jd|j|}|dkrHtd|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)) r_rBrFrGr/rgrtr.rrZhandle_backmsg_exception)rirsrrvr4r4r5(handle_backmsg_deserialization_exceptions  z0Runtime.handle_backmsg_deserialization_exceptionzTuple[bool, str]cs|jtjtjtjfkrdSdS)N)Tok)F unavailable)r_rBrCrFrGrlr4r4r5is_ready_for_browser_connectionsz'Runtime.is_ready_for_browser_connectioncstt|j|j|j|jt|jddid}zr|dt }t |j krjt |t krjtdIdHq>t |j kr|W&dS|j t }|rdnd}||fWS|XdS) 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)rrYrZZlocal_sources_watcherrNg?)Ftimeoutrerror)rrr]r^rarfr*shutdownZ request_reruntime perf_counterr%Z session_stater-r~sleep)rirnowrrr4r4r5does_script_run_without_errors*      z%Runtime.does_script_run_without_error) new_staterQcCstd|j|||_dS)NzRuntime state: %s -> %s)r.rr_)rirr4r4r5r&szRuntime._set_statec s|}z|jtjkr&|tjn|jtjkr4ntd|j|j d|j sj|jtjkrt j t |j t |j ft jdIdHn|jtjkrj|j|jD]`}|j}|D]L}z|||Wn&tk r |j|jjYnXt dIdHqqt dIdHnqjt j t |j t |j ft jdIdHqP|jD]}|j|jjqt|tj|j dWnBt k r}z"|j!|t"#t$%dW5d}~XYnXdS)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)Z return_whenrg{Gz?zJ Please report this bug at https://github.com/streamlit/streamlit/issues. )&rrr_rBrCrrDrErRrMZ set_resultrJis_setr~waitrrKZFIRST_COMPLETEDrLclearrgZlist_active_sessionsrZflush_browser_queue _send_messager"ridrZ list_sessionsrrGrN ExceptionZ set_exception traceback print_excr.info)rirZactive_session_infoZmsg_listrrver4r4r5r*s\            zRuntime._loop_coroutiner r)rvrrQcCst||j_|}|jjrlt||j||j|jrJt d|j t |}t d|j |j ||j|j| ddkr|jtjkrt dtd|jd7_|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 cacheablerr`Zhas_message_referencerZscript_run_countr.rhashrZ add_messageZ WhichOneofrrZFINISHED_SUCCESSFULLYr get_optionZremove_expired_session_entriesruZwrite_forward_msg)rirvrZ msg_to_sendr4r4r5r~s>   zRuntime._send_messagecCs|}|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)rrrIrrLrrr4r4r5rfs zRuntime._enqueued_some_messagerHcCs|jdkrtd|jS)zpReturn our AsyncObjects instance. If the Runtime hasn't been started, this will raise an error. NzRuntime hasn't started yet!)r[rRrlr4r4r5rrs zRuntime._get_async_objscCs8|jtjkr4|jdkr4|j|tj dS)zlSet the runtime state to NO_SESSIONS_CONNECTED if the last active session was disconnected. rN) r_rBrErgZnum_active_sessionsrrrKrrrDrlr4r4r5rs   z Runtime._on_session_disconnected)N)N)$r0r1r2rPrA classmethodrUrWrjpropertyrmrnror>rprqrNrwrcrrryrrrrrrrrrrrrfrrrr4r4r4r5rOsT   0   " 0 +T; rO)Q __future__rr~rrrZ dataclassesrrenumrtypingrrrr r r r Ztyping_extensionsr Z streamlitrZstreamlit.loggerrZstreamlit.proto.BackMsg_pb2rZstreamlit.proto.ForwardMsg_pb2rZstreamlit.runtime.app_sessionrZstreamlit.runtime.cachingrrZ:streamlit.runtime.caching.storage.local_disk_cache_storagerZ#streamlit.runtime.forward_msg_cacherrrZ(streamlit.runtime.legacy_caching.cachingrZ$streamlit.runtime.media_file_managerrZ$streamlit.runtime.media_file_storagerZ(streamlit.runtime.memory_session_storagerZstreamlit.runtime.runtime_utilrZstreamlit.runtime.script_datarZ!streamlit.runtime.session_managerr r!r"r#r$Zstreamlit.runtime.stater%r&Zstreamlit.runtime.statsr'Z'streamlit.runtime.uploaded_file_managerr(Z+streamlit.runtime.websocket_session_managerr)Zstreamlit.watcherr*Z!streamlit.runtime.caching.storager+r-rAr0r.rr/r7rBrHrOr4r4r4r5sN  $