o ,&]ip@sUddlmZddlZddlZddlZddlZddlmZmZddl m Z ddl m Z m Z mZddlmZddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZdd lmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'm(Z(m)Z)m*Z*m+Z+ddl,m-Z-m.Z.ddl/m0Z0ddl1m2Z2e rddl3m4Z4ddl5m6Z6ddl7m8Z8ddl9m:Z:ddl;mZ>dZ?de@d<ejAdkrdndZBde@d <eeCZDde@d!<Gd"d#d#eEZFed$d%Gd&d'd'ZGGd(d)d)e ZHGd*d+d+eZIGd,d-d-ZJdS).) annotationsN) dataclassfield)Enum) TYPE_CHECKINGFinal NamedTuple)LocalComponentRegistry)BidiComponentManager) 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.W/var/www/html/IGF-ODF-V3/venv/lib/python3.10/site-packages/streamlit/runtime/runtime.pyr)Jsr)T)frozenc@seZdZUdZded<ded<ded<ded <eed Zd ed <eed Z d ed<ee d Z ded<e Z ded<eed Zded<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_registryr bidi_component_registryztype[SessionManager]session_manager_classrsession_storageFboolis_helloN)r*r+r,r-__annotations__rrr9r r:r r;rr<rr=r?r.r.r.r/r1Ns$  r1c@s eZdZdZdZdZdZdZdS) RuntimeStateINITIALNO_SESSIONS_CONNECTEDONE_OR_MORE_SESSIONS_CONNECTEDSTOPPINGSTOPPEDN)r*r+r,rBrCrDrErFr.r.r.r/rA~s rAc@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/rGs  rGc@s~eZdZUdZded<edaddZedbdd Zdcd dZe ddddZ e deddZ e dfddZ e dgddZ e dhddZe didd Ze 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!)rO 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)rOrSr.r.r/existss zRuntime.existsconfigr1NonecCstjdur td|t_d|_d|_|j|_|j|_t j |_ |j |_ |j|_|j|_t|jd|_|j|_t|_|j|j|j|j|j|jd|_t|_|j t!|j t"|j |j|j t#|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=r7 script_cachemessage_enqueued_callback)$rNrOrR _async_objs_loop_coroutine_taskr3_main_script_pathr? _is_hellorArB_stater:_component_registryr;_bidi_component_registryr7_uploaded_file_mgrrr6_media_file_mgrr9_cache_storage_managerr _script_cache discover_and_register_componentsr<r=_enqueued_some_message _session_mgrr _stats_mgrregister_providerrrr)selfrWr.r.r/__init__s4  zRuntime.__init__rAcC|jSN)r`rlr.r.r/statez Runtime.statercCrnro)rarpr.r.r/r:rrzRuntime.component_registryr cCrnro)rbrpr.r.r/r;rrzRuntime.bidi_component_registryr#cCrnro)rcrpr.r.r/uploaded_file_mgrrrzRuntime.uploaded_file_mgrr!cCrnro)rerpr.r.r/r9rrzRuntime.cache_storage_managerrcCrnro)rdrpr.r.r/media_file_mgrrrzRuntime.media_file_mgrrcCrnro)rjrpr.r.r/ stats_mgrrrzRuntime.stats_mgrAwaitable[None]cCs |jS)z?A Future that completes when the Runtime's run loop has exited.)_get_async_objsrMrpr.r.r/rMs zRuntime.stopped session_idr2SessionClient | 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)riget_active_session_infoclientrlrx 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)riget_session_infosessionclear_user_infor|r.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. )rHrIrJrKrLrMzRuntime.loop_coroutine)nameN) rGasyncioget_running_loopEventFuturer\ create_task_loop_coroutiner]rLrl async_objsr.r.r/start's 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. rPrXcs:jtjtjfvr dStdtjjdS)NzRuntime stopping...) r`rArErFr(debug _set_staterIsetr.rrlr.r/stop_on_eventloopOs   z'Runtime.stop..stop_on_eventloopNrPrX)rwrHcall_soon_threadsafe)rlrr.rr/stopDs z Runtime.stopcCs |j|S)zTrue if the session_id belongs to an active session. Notes ----- Threading: SAFE. May be called on any thread. )riis_active_session)rlrxr.r.r/rYs zRuntime.is_active_sessionr{r user_infodict[str, str | bool | None]existing_session_idr4session_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=))r{ script_datarrr)rRr`rArErFr)riconnect_sessionrr^r_rrDrwrJr)rlr{rrrrxr.r.r/rbs *  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.)r{rrr)r(warningr)rlr{rrrr.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)rir close_session_on_session_disconnectedr|r.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)rirzdisconnect_sessionrr|r.r.r/rrzRuntime.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`rArErFr)rirzr(rrhandle_backmsg)rlrxrr}r.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`rArErFr)rirzr(rrhandle_backmsg_exception)rlrxrr}r.r.r/(handle_backmsg_deserialization_exceptions  z0Runtime.handle_backmsg_deserialization_exceptiontuple[bool, str]cs |jtjtjtjfvrdSdS)N)Tok)F unavailable)r`rArBrErFrpr.r.r/is_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)rr7rZr[rNg?)Ftimeoutrerror)r rr^r_rcrfrh request_reruntime perf_counterr session_stater%rsleepshutdown)rlrnowrrr.r.r/does_script_run_without_error+s.       z%Runtime.does_script_run_without_error new_statecCstd|j|||_dS)NzRuntime state: %s -> %s)r(rr`)rlrr.r.r/rVs 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. )(rwr`rArBrrCrDrRrL set_resultrIis_setrwaitrrJFIRST_COMPLETEDr'cancelrKclearrilist_active_sessionsrflush_browser_queue _send_messagerridr list_sessionsrrFrM Exception set_exception traceback print_excr(info) rlr done_tasks pending_taskstaskactive_session_infomsg_listrr}er.r.r/rZs~                C  zRuntime._loop_coroutiner}rr 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_countr{write_forward_msg)rlr}rr.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)rwrHrrKrrr.r.r/rhs zRuntime._enqueued_some_messagerGcCrQ)zpReturn our AsyncObjects instance. If the Runtime hasn't been started, this will raise an error. NzRuntime hasn't started yet!)r\rRrpr.r.r/rws 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`rArDrinum_active_sessionsrwrJrrrCrpr.r.r/rs z Runtime._on_session_disconnected)rPrN)rPr>)rWr1rPrX)rPrA)rPr)rPr )rPr#)rPr!)rPr)rPr)rPrv)rxr2rPry)rxr2rPrXr)rxr2rPr>)NN) r{rrrrr4rr4rPr2)rxr2rr rPrX)rxr2rrrPrX)rPr)rrArPrX)r}rrr rPrX)rPrG)%r*r+r,rOr@ classmethodrUrVrmpropertyrqr:r;rsr9rtrurMr~rrrrrrrrrrrrrrrrhrwrr.r.r.r/rNs^    3            C      +  l  rN)K __future__rrsysrr dataclassesrrenumrtypingrrr1streamlit.components.lib.local_component_registryr )streamlit.components.v2.component_managerr streamlit.loggerr streamlit.proto.ForwardMsg_pb2r streamlit.runtime.app_sessionr streamlit.runtime.cachingrr: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)r1rArGrNr.r.r.r/sN                    /