o hBu@sUddlmZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z ddlmZddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZdd lmZmZm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,m-Z-m.Z.m/Z/m0Z0ddl1m2Z2m3Z3ddl4m5Z5ddl6m7Z7e rddl8m9Z9ddl:m;Z;ddlm?Z?ddl@mAZAddlBmCZCdZDdeEd<eeFZGdeEd <Gd!d"d"eHZIed#d$Gd%d&d&ZJGd'd(d(e ZKGd)d*d*e ZLGd+d,d,ZMdS)-) annotationsN) dataclassfield)Enum) TYPE_CHECKINGFinal NamedTuple)config)LocalComponentRegistry) get_logger) ForwardMsg) AppSession)get_data_cache_stats_provider!get_resource_cache_stats_provider)LocalDiskCacheStorageManager)ForwardMsgCachecreate_reference_msgpopulate_hash_if_needed)MediaFileManager)MemorySessionStorage)is_cacheable_msg) ScriptData) ScriptCache)ActiveSessionInfo SessionClientSessionClientDisconnectedErrorSessionManagerSessionStorage)SCRIPT_RUN_WITHOUT_ERRORS_KEYSessionStateStatProvider) StatsManager)WebsocketSessionManager) Awaitable)BaseComponentRegistry)BackMsg)CacheStorageManager)MediaFileStorage)UploadedFileManager<rSCRIPT_RUN_CHECK_TIMEOUT_LOGGERc@seZdZdZdS)RuntimeStoppedErrorz;Raised by operations on a Runtime instance that is stopped.N)__name__ __module__ __qualname____doc__r0r0M/var/www/vscode/kcb/lib/python3.10/site-packages/streamlit/runtime/runtime.pyr+Ksr+T)frozenc@seZdZUdZded<ded<ded<ded <eed Zd ed <eed Z d ed<e Z ded<ee d Z ded<dZded<dS) RuntimeConfigz$Config options for StreamlitRuntime.str script_path str | None command_liner&media_file_storager'uploaded_file_manager)default_factoryr%cache_storage_managerr#component_registryztype[SessionManager]session_manager_classrsession_storageFboolis_helloN)r,r-r.r/__annotations__rrr;r r<r!r=rr>r@r0r0r0r1r3Os  r3c@s eZdZdZdZdZdZdZdS) RuntimeStateINITIALNO_SESSIONS_CONNECTEDONE_OR_MORE_SESSIONS_CONNECTEDSTOPPINGSTOPPEDN)r,r-r.rCrDrErFrGr0r0r0r1rBzs 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)r,r-r.r/rAr0r0r0r1rHs  rHc@s~eZdZUdZded<edaddZedbdd Zdcd d Ze ddddZ e deddZ e dfddZ e dgddZ e dhddZe diddZe djd!d"Ze dkd$d%Zdld)d*Zdmd,d-Zdnd.d/Zdnd0d1Zdod2d3Z  dpdqd;d<Z  dpdqd=d>Zdmd?d@ZdmdAdBZdrdEdFZdsdIdJZe dtdLdMZdtdNdOZdudQdRZdndSdTZ dvdXdYZ!dndZd[Z"dwd]d^Z#dnd_d`Z$dS)xRuntimeNzRuntime | None _instancereturncC|jdur td|jS)znReturn the singleton Runtime instance. Raise an Error if the Runtime hasn't been created yet. NzRuntime hasn't been created!)rP RuntimeErrorclsr0r0r1instances zRuntime.instancer?cCs |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)rPrTr0r0r1existss zRuntime.existsr r3cCstjdur td|t_d|_d|_|j|_|j|_t j |_ |j |_ t|_|j|_t|jd|_|j|_t|_|j|j|j|j|jd|_t|_|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!)storage)r>r9 script_cachemessage_enqueued_callback)#rOrPrS _async_objs_loop_coroutine_taskr5_main_script_pathr@ _is_hellorBrC_stater<_component_registryr_message_cacher9_uploaded_file_mgrrr8_media_file_mgrr;_cache_storage_managerr _script_cacher=r>_enqueued_some_message _session_mgrr _stats_mgrregister_providerrrr)selfr r0r0r1__init__s4 zRuntime.__init__rBcC|jSN)r_rjr0r0r1statez Runtime.stater#cCrlrm)r`rnr0r0r1r<rpzRuntime.component_registryrcCrlrm)rarnr0r0r1 message_cacherpzRuntime.message_cacher'cCrlrm)rbrnr0r0r1uploaded_file_mgrrpzRuntime.uploaded_file_mgrr%cCrlrm)rdrnr0r0r1r;rpzRuntime.cache_storage_managerrcCrlrm)rcrnr0r0r1media_file_mgrrpzRuntime.media_file_mgrr cCrlrm)rhrnr0r0r1 stats_mgrrpzRuntime.stats_mgrAwaitable[None]cCs |jS)z?A Future that completes when the Runtime's run loop has exited.)_get_async_objsrNrnr0r0r1rNs zRuntime.stopped session_idr4SessionClient | NonecCs|j|}|dur 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)rgget_active_session_infoclientrjrw session_infor0r0r1 get_client s zRuntime.get_clientNonecCs&|j|}|dur|jdSdS)zClear the user_info for the given session_id. Notes ----- Threading: SAFE. May be called on any thread. N)rgget_session_infosessionclear_user_infor{r0r0r1clear_user_info_for_sessions z#Runtime.clear_user_info_for_sessioncsXtttttttd}||_tj|dd|_|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. )rIrJrKrLrMrNzRuntime.loop_coroutine)nameN) rHasyncioget_running_loopEventFuturer[ create_task_loop_coroutiner\rMrj async_objsr0r0r1start!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:jtjtjfvr dStdtjjdS)NzRuntime stopping...) r_rBrFrGr*debug _set_staterJsetr0rrjr0r1stop_on_eventloopIs   z'Runtime.stop..stop_on_eventloopN)rvrIcall_soon_threadsafe)rjrr0rr1stop>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. )rgis_active_session)rjrwr0r0r1rSs zRuntime.is_active_sessionrzr user_infodict[str, str | bool | None]existing_session_idr6session_id_overridecCsr|r|rJd|jtjtjfvrtd|jd|jj|t|j|j |||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" } 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=))rz script_datarrr)r_rBrFrGr+rgconnect_sessionrr]r^rrErvrKr)rjrzrrrrwr0r0r1r\s *  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.)rzrrr)r*warningr)rjrzrrrr0r0r1create_sessions zRuntime.create_sessioncC6|j|}|r|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)rgrraremove_refs_for_sessionr close_session_on_session_disconnectedr{r0r0r1rs   zRuntime.close_sessioncCr)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)rgryrarrdisconnect_sessionrr{r0r0r1rs   zRuntime.disconnect_sessionmsgr$cCX|jtjtjfvrtd|jd|j|}|dur$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)) r_rBrFrGr+rgryr*rrhandle_backmsg)rjrwrr|r0r0r1rs zRuntime.handle_backmsgexc BaseExceptioncCr)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+rgryr*rrhandle_backmsg_exception)rjrwrr|r0r0r1(handle_backmsg_deserialization_exceptions  z0Runtime.handle_backmsg_deserialization_exceptiontuple[bool, str]cs |jtjtjtjfvrdSdS)N)Tok)F unavailable)r_rBrCrFrGrnr0r0r1is_ready_for_browser_connection!sz'Runtime.is_ready_for_browser_connectioncstt|j|j|j|j|jddid}zK|dt }t |j vr@t |t kr@t dIdHt |j vr@t |t ks+t |j vrLW|dS|j t }|rUdnd}||fW|S|w) 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. emailztest@example.com)rr9rYrZrNg?)Ftimeoutrerror)r rr]r^rbrerf request_reruntime perf_counterr session_stater)rsleepshutdown)rjrnowrrr0r0r1does_script_run_without_error,s.       z%Runtime.does_script_run_without_error new_statecCstd|j|||_dS)NzRuntime state: %s -> %s)r*rr_)rjrr0r0r1rWs zRuntime._set_statec s|}z|jtjkr|tjn|jtjkrntd|j|j d|j s|jtjkrXt j t |j t |j ft jdIdH\}}|D]}|qPnI|jtjkr|j|jD].}|j}|D]$}z|||Wnty|j|jjYnwt dIdHqqqht dIdHnn)t j t |j t |j ft jdIdH\}}|D]}|q|j r-|jD] }|j|jjq|tj|j dWdSt!y } z|j "| t#$t%&dWYd} ~ dSd} ~ ww)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. )'rvr_rBrCrrDrErSrM set_resultrJis_setrwaitrrKFIRST_COMPLETEDcancelrLclearrglist_active_sessionsrflush_browser_queue _send_messagerridr list_sessionsrrGrN Exception set_exception traceback print_excr*info) rjr_ pending_taskstaskactive_session_infomsg_listrr|er0r0r1r[sr              7  zRuntime._loop_coroutiner|rr cCst||j_|}|jjr6t||j||j|jr%t d|j t |}t d|j |j ||j|j| ddkrg|jtjksNtdrg|jtjkrgt dtd|jd7_|j|j|j|j|d S) 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_finishedz4global.includeFragmentRunsInForwardMessageCacheCountzYScript run finished successfully; removing expired entries from MessageCache (max_age=%s)zglobal.maxCachedMessageAgeN)rmetadata cacheablerrahas_message_referencerscript_run_countr*rhashr add_message WhichOneofrr FINISHED_SUCCESSFULLYr get_option"FINISHED_FRAGMENT_RUN_SUCCESSFULLY"remove_expired_entries_for_sessionrzwrite_forward_msg)rjr|r msg_to_sendr0r0r1rs:     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)rvrIrrLrrr0r0r1rfs zRuntime._enqueued_some_messagerHcCrR)zpReturn our AsyncObjects instance. If the Runtime hasn't been started, this will raise an error. NzRuntime hasn't started yet!)r[rSrnr0r0r1rvs zRuntime._get_async_objscCs@|jtjkr|jdkr|j|tj dSdSdS)zlSet the runtime state to NO_SESSIONS_CONNECTED if the last active session was disconnected. rN) r_rBrErgnum_active_sessionsrvrKrrrDrnr0r0r1rs z Runtime._on_session_disconnected)rQrO)rQr?)r r3)rQrB)rQr#)rQr)rQr')rQr%)rQr)rQr )rQru)rwr4rQrx)rwr4rQr~)rQr~)rwr4rQr?)NN) rzrrrrr6rr6rQr4)rwr4rr$rQr~)rwr4rrrQr~)rQr)rrBrQr~)r|rrr rQr~)rQrH)%r,r-r.rPrA classmethodrVrWrkpropertyror<rqrrr;rsrtrNr}rrrrrrrrrrrrrrrrfrvrr0r0r0r1rOs^    1            A     +  ` @ rO)N __future__rrrr dataclassesrrenumrtypingrrr streamlitr 1streamlit.components.lib.local_component_registryr streamlit.loggerr streamlit.proto.ForwardMsg_pb2r streamlit.runtime.app_sessionr streamlit.runtime.cachingrr:streamlit.runtime.caching.storage.local_disk_cache_storager#streamlit.runtime.forward_msg_cacherrr$streamlit.runtime.media_file_managerr(streamlit.runtime.memory_session_storagerstreamlit.runtime.runtime_utilrstreamlit.runtime.script_datar+streamlit.runtime.scriptrunner.script_cacher!streamlit.runtime.session_managerrrrrrstreamlit.runtime.staterrstreamlit.runtime.statsr +streamlit.runtime.websocket_session_managerr!collections.abcr"2streamlit.components.types.base_component_registryr#streamlit.proto.BackMsg_pb2r$!streamlit.runtime.caching.storager%$streamlit.runtime.media_file_storager&'streamlit.runtime.uploaded_file_managerr'r)rAr,r*rr+r3rBrHrOr0r0r0r1sN                     *