o Zhn@sUddlmZddlZddlZddlZddlZddlmZmZddl m Z ddl m Z m Z mZddlmZddlmZddlmZdd lmZdd lmZmZdd lmZdd lmZdd lmZddl m!Z!ddl"m#Z#ddl$m%Z%m&Z&m'Z'm(Z(m)Z)ddl*m+Z+m,Z,ddl-m.Z.ddl/m0Z0e rddl1m2Z2ddl3m4Z4ddl5m6Z6ddl7m8Z8ddl9m:Z:ddl;md<ej?dkrdndZ@de>d<eeAZBde>d <Gd!d"d"eCZDed#d$Gd%d&d&ZEGd'd(d(e ZFGd)d*d*eZGGd+d,d,ZHdS)-) annotationsN) dataclassfield)Enum) TYPE_CHECKINGFinal NamedTuple)LocalComponentRegistry) get_logger) ForwardMsg) AppSession)get_data_cache_stats_provider!get_resource_cache_stats_provider)LocalDiskCacheStorageManager)MediaFileManager)MemorySessionStorage) ScriptData) ScriptCache)ActiveSessionInfo SessionClientSessionClientDisconnectedErrorSessionManagerSessionStorage)SCRIPT_RUN_WITHOUT_ERRORS_KEYSessionStateStatProvider) StatsManager)WebsocketSessionManager) Awaitable)BaseComponentRegistry)BackMsg)CacheStorageManager)MediaFileStorage)UploadedFileManager<rSCRIPT_RUN_CHECK_TIMEOUTwin32g?_SIGNAL_CHECK_INTERVAL_LOGGERc@seZdZdZdS)RuntimeStoppedErrorz;Raised by operations on a Runtime instance that is stopped.N)__name__ __module__ __qualname____doc__r-r-Z/var/www/html/Persson_Maskin/env/lib/python3.10/site-packages/streamlit/runtime/runtime.pyr(Isr(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_managerrcomponent_registryztype[SessionManager]session_manager_classrsession_storageFboolis_helloN)r)r*r+r,__annotations__rrr8r r9rr:rr;r=r-r-r-r.r0Ms  r0c@s eZdZdZdZdZdZdZdS) RuntimeStateINITIALNO_SESSIONS_CONNECTEDONE_OR_MORE_SESSIONS_CONNECTEDSTOPPINGSTOPPEDN)r)r*r+r@rArBrCrDr-r-r-r.r?xs r?c@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,r>r-r-r-r.rEs  rEc@speZdZUdZded<ed^ddZed_dd Zd`d dZe daddZ e dbddZ e dcddZ e ddddZ e deddZe dfdd Ze dgd"d#Zdhd'd(Zdid)d*Zdjd+d,Zdjd-d.Zdkd/d0Z  dldmd8d9Z  dldmd:d;Zdidd?ZdndBdCZdodFdGZe dpdIdJZdpdKdLZdqdNdOZdjdPdQZdrdUdVZ djdWdXZ!dsdZd[Z"djd\d]Z#dS)tRuntimeNzRuntime | 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!)rM RuntimeErrorclsr-r-r.instances 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)rMrQr-r-r.existss zRuntime.existsconfigr0NonecCstjdur td|t_d|_d|_|j|_|j|_t j |_ |j |_ |j|_t|jd|_|j|_t|_|j|j|j|j|jd|_t|_|jt|jt|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;r6 script_cachemessage_enqueued_callback)!rLrMrP _async_objs_loop_coroutine_taskr2_main_script_pathr= _is_hellor?r@_stater9_component_registryr6_uploaded_file_mgrrr5_media_file_mgrr8_cache_storage_managerr _script_cacher:r;_enqueued_some_message _session_mgrr _stats_mgrregister_providerr rr)selfrUr-r-r.__init__s0 zRuntime.__init__r?cC|jSN)r^rhr-r-r.statez Runtime.statercCrjrk)r_rlr-r-r.r9rnzRuntime.component_registryr"cCrjrk)r`rlr-r-r.uploaded_file_mgrrnzRuntime.uploaded_file_mgrr cCrjrk)rbrlr-r-r.r8rnzRuntime.cache_storage_managerrcCrjrk)rarlr-r-r.media_file_mgrrnzRuntime.media_file_mgrrcCrjrk)rfrlr-r-r. stats_mgrrnzRuntime.stats_mgrAwaitable[None]cCs |jS)z?A Future that completes when the Runtime's run loop has exited.)_get_async_objsrKrlr-r-r.rKs zRuntime.stopped session_idr1SessionClient | 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)reget_active_session_infoclientrhrt session_infor-r-r. get_clients zRuntime.get_clientcCs&|j|}|dur|jdSdS)zClear the user_info for the given session_id. Notes ----- Threading: SAFE. May be called on any thread. N)reget_session_infosessionclear_user_inforxr-r-r.clear_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. )rFrGrHrIrJrKzRuntime.loop_coroutine)nameN) rEasyncioget_running_loopEventFuturerZ create_task_loop_coroutiner[rJrh async_objsr-r-r.starts z Runtime.startcs(dfdd }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. rNrVcs:jtjtjfvr dStdtjjdS)NzRuntime stopping...) r^r?rCrDr'debug _set_staterGsetr-rrhr-r.stop_on_eventloopAs   z'Runtime.stop..stop_on_eventloopNrNrV)rsrFcall_soon_threadsafe)rhrr-rr.stop6s z Runtime.stopcCs |j|S)zTrue if the session_id belongs to an active session. Notes ----- Threading: SAFE. May be called on any thread. )reis_active_session)rhrtr-r-r.rKs zRuntime.is_active_sessionrwr user_infodict[str, str | bool | None]existing_session_idr3session_id_overridecCsr|r|rtd|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. z`Only one of existing_session_id and session_id_override should be set. This should never happen.zCan't connect_session (state=))rw script_datarrr)rPr^r?rCrDr(reconnect_sessionrr\r]rrBrsrHr)rhrwrrrrtr-r-r.rTs *  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.)rwrrr)r'warningr)rhrwrrrr-r-r.create_sessions zRuntime.create_sessioncC(|j|}|r|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)rer{ close_session_on_session_disconnectedrxr-r-r.r   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)rervdisconnect_sessionrrxr-r-r.rrzRuntime.disconnect_sessionmsgrcCX|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^r?rCrDr(rervr'rr|handle_backmsg)rhrtrryr-r-r.rs 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^r?rCrDr(rervr'rr|handle_backmsg_exception)rhrtrryr-r-r.(handle_backmsg_deserialization_exceptions  z0Runtime.handle_backmsg_deserialization_exceptiontuple[bool, str]cs |jtjtjtjfvrdSdS)N)Tok)F unavailable)r^r?r@rCrDrlr-r-r.is_ready_for_browser_connectionsz'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)rr6rXrYrNg?)Ftimeoutrerror)r rr\r]r`rcrd request_reruntime perf_counterr session_stater$rsleepshutdown)rhr|nowrrr-r-r.does_script_run_without_errors.       z%Runtime.does_script_run_without_error new_statecCstd|j|||_dS)NzRuntime state: %s -> %s)r'rr^)rhrr-r-r.rHs zRuntime._set_statec s8|}z|jtjkr|tjn|jtjkrntd|j|j d|j s|jtjkr`t j t |j t |j ft jtdIdH\}}|D]}|qQ|s_tdur_q(nI|jtjkr|j|jD].}|j}|D]$}z|||Wnty|j|jjYnwt dIdHqyqpt dIdHnn1t j t |j t |j ft jtdIdH\}}|D]}|q|stdurq(|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_whenrrg{Gz?zJ Please report this bug at https://github.com/streamlit/streamlit/issues. )(rsr^r?r@rrArBrPrJ set_resultrGis_setrwaitrrHFIRST_COMPLETEDr&cancelrIclearrelist_active_sessionsr|flush_browser_queue _send_messagerridr list_sessionsrrDrK Exception set_exception traceback print_excr'info) rhr done_tasks pending_taskstaskactive_session_infomsg_listrryer-r-r.rLs~                C  zRuntime._loop_coroutineryrr cCs8|ddkr|jtjkr|jd7_|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. typescript_finishedN) WhichOneofrr FINISHED_SUCCESSFULLYscript_run_countrwwrite_forward_msg)rhryrr-r-r.rs 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)rsrFrrIrrr-r-r.rds zRuntime._enqueued_some_messagerEcCrO)zpReturn our AsyncObjects instance. If the Runtime hasn't been started, this will raise an error. NzRuntime hasn't started yet!)rZrPrlr-r-r.rss 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^r?rBrenum_active_sessionsrsrHrrrArlr-r-r.rs z Runtime._on_session_disconnected)rNrL)rNr<)rUr0rNrV)rNr?)rNr)rNr")rNr )rNr)rNr)rNrr)rtr1rNru)rtr1rNrVr)rtr1rNr<)NN) rwrrrrr3rr3rNr1)rtr1rrrNrV)rtr1rrrNrV)rNr)rr?rNrV)ryrrr rNrV)rNrE)$r)r*r+rMr> classmethodrSrTripropertyrmr9ror8rprqrKrzr~rrrrrrrrrrrrrrrdrsrr-r-r-r.rLsZ    /           C      +  l  rL)I __future__rrsysrr dataclassesrrenumrtypingrrr1streamlit.components.lib.local_component_registryr streamlit.loggerr streamlit.proto.ForwardMsg_pb2r streamlit.runtime.app_sessionr streamlit.runtime.cachingr r:streamlit.runtime.caching.storage.local_disk_cache_storager$streamlit.runtime.media_file_managerr(streamlit.runtime.memory_session_storagerstreamlit.runtime.script_datar+streamlit.runtime.scriptrunner.script_cacher!streamlit.runtime.session_managerrrrrrstreamlit.runtime.staterrstreamlit.runtime.statsr+streamlit.runtime.websocket_session_managerrcollections.abcr2streamlit.components.types.base_component_registryrstreamlit.proto.BackMsg_pb2r!streamlit.runtime.caching.storager $streamlit.runtime.media_file_storager!'streamlit.runtime.uploaded_file_managerr"r$r>platformr&r)r'rr(r0r?rErLr-r-r-r.sL                   *