o hx@sUddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl mZmZmZmZmZddlmZddlmZmZmZdd lmZdd lmZdd lmZdd lm Z dd l!m"Z"m#Z#ddl$m%Z%ddl&m'Z'm(Z(ddl)m*Z*ddl+m,Z,m-Z-ddl.m/Z/m0Z0m1Z1ddl2m3Z3m4Z4m5Z5ddl6m7Z7m8Z8m9Z9ddl:m;Z;erddlm?Z?ee@ZAdeBd<Gddde ZC d*ddZDGd d!d!ZEd+d$d%ZFd,d(d)ZGdS)-) annotationsN)contextmanager)Enum) default_timer) TYPE_CHECKINGCallableFinalLiteralcast)Signal)configruntimeutil)FragmentStorageKeyError) get_logger) ClientState) ForwardMsg)create_page_profile_messageto_microseconds) PagesManager)exec_func_with_error_handlingmodified_sys_path) ScriptCache)RerunException StopException) RerunDataScriptRequestsScriptRequestType)ScriptRunContextadd_script_run_ctxget_script_run_ctx)SCRIPT_RUN_WITHOUT_ERRORS_KEYSafeSessionState SessionState) page_sort_key)FragmentStorage)UploadedFileManagerr_LOGGERc@s(eZdZdZdZdZdZdZdZdZ dS) ScriptRunnerEventSCRIPT_STARTED!SCRIPT_STOPPED_WITH_COMPILE_ERRORSCRIPT_STOPPED_WITH_SUCCESSSCRIPT_STOPPED_FOR_RERUNFRAGMENT_STOPPED_WITH_SUCCESSSHUTDOWNENQUEUE_FORWARD_MSGN) __name__ __module__ __qualname__r)r*r+r,r-r.r/r3r3`/var/www/vscode/kcb/lib/python3.10/site-packages/streamlit/runtime/scriptrunner/script_runner.pyr(Hsr(main_script_pathstrc sddlm}ddlm}m}ddlm||}|jd d}t dd|Dt d }|d d }|gfd d|D}t d durKdnd}|td||dd} | j|jkrd| tdS)Nr)Path)PageType _navigation) StreamlitPagepagesz*.pycSsg|] }|jdr|qS)z.py)nameendswith.0pager3r3r4 sz_mpa_v1..)keyT)defaultcsg|] }|jqSr3)r<r> PAGES_FOLDERr:r3r4rAszclient.showSidebarNavigationFhiddensidebarzlist[PageType])positionexpanded)pathlibr7streamlit.commands.navigationr8r9streamlit.navigation.pager:resolveparentglobsortedr$r get_optionr _pagerunr) r5r7r8r9MAIN_SCRIPT_PATHr; main_page all_pagesrHr@r3rDr4_mpa_v1|s4       rWc@seZdZdddZd?ddZd>ddZd@d!d"Zd>d#d$Z dAd%d&Z dBd)d*Z d>d+d,Z e d-d.ZdCd/d0ZdDd5d6ZdEd9d:Zd;S)F ScriptRunner session_idr6r5 session_stater#uploaded_file_mgrr& script_cacherinitial_rerun_datar user_infodict[str, str | bool | None]fragment_storager% pages_managerrc Csj||_||_t||jd|_||_||_||_||_| |_ t |_ |j |t dd|_d|_d|_dS)a1Initialize the ScriptRunner. (The ScriptRunner won't start executing until start() is called.) Parameters ---------- session_id The AppSession's id. main_script_path Path to our main app script. session_state The AppSession's SessionState instance. uploaded_file_mgr The File manager to store the data uploaded by the file_uploader widget. script_cache A ScriptCache instance. initial_rerun_data RerunData to initialize this ScriptRunner with. user_info A dict that contains information about the current user. For now, it only contains the user's email address. { "email": "example@example.com" } Information about the current user is optionally provided when a websocket connection is initialized via the "X-Streamlit-User" header. fragment_storage The AppSession's FragmentStorage instance. )yield_callbackaEmitted when a ScriptRunnerEvent occurs. This signal is generally emitted on the ScriptRunner's script thread (which is *not* the same thread that the ScriptRunner was created on). Parameters ---------- sender: ScriptRunner The sender of the event (this ScriptRunner). event : ScriptRunnerEvent forward_msg : ForwardMsg | None The ForwardMsg to send to the frontend. Set only for the ENQUEUE_FORWARD_MSG event. exception : BaseException | None Our compile error. Set only for the SCRIPT_STOPPED_WITH_COMPILE_ERROR event. widget_states : streamlit.proto.WidgetStates_pb2.WidgetStates | None The ScriptRunner's final WidgetStates. Set only for the SHUTDOWN event. )docFN) _session_id_main_script_pathr"'_maybe_handle_execution_control_request_session_state_uploaded_file_mgr _script_cache _user_info_fragment_storage_pages_managerr _requests request_rerunr on_event_execing_script_thread) selfrYr5rZr[r\r]r^r`rar3r3r4__init__s"2  zScriptRunner.__init__returncCs t|SN)rrepr_rrr3r3r4__repr__s zScriptRunner.__repr__NonecCs|jdS)zRequest that the ScriptRunner stop running its script and shut down. The ScriptRunner will handle this request when it reaches an interrupt point. Safe to call from any thread. N)rm request_stoprwr3r3r4rz szScriptRunner.request_stop rerun_databoolcCs |j|S)aRequest that the ScriptRunner interrupt its currently-running script and restart it. If the ScriptRunner has been stopped, this request can't be honored: return False. Otherwise, record the request and return True. The ScriptRunner will handle the rerun request as soon as it reaches an interrupt point. Safe to call from any thread. )rmrn)rrr{r3r3r4rns zScriptRunner.request_reruncCs2|jdur tdtj|jdd|_|jdS)zeStart a new thread to process the ScriptEventQueue. This must be called only once. Nz ScriptRunner was already startedzScriptRunner.scriptThread)targetr<)rq Exception threadingThread_run_script_threadstartrwr3r3r4r"s zScriptRunner.startrcCs&|sJt}|durtd|S)atGet the ScriptRunContext for the current thread. Returns ------- ScriptRunContext The ScriptRunContext for the current thread. Raises ------ AssertionError If called outside of a ScriptRunner thread. RuntimeError If there is no ScriptRunContext for the current thread. NzOScriptRunner thread has a null ScriptRunContext. Something has gone very wrong!)_is_in_script_threadr RuntimeError)rrctxr3r3r4_get_script_run_ctx1s z ScriptRunner._get_script_run_ctxcCs|sJtdt|j|j|jd|j|j|j |j t t d|j|jdd }tt||j}|jtjkrL||j|j}|jtjks;|jtjksTJt}|j|_|j|_|jj|tj |ddS)aNThe entry point for the script thread. Processes the ScriptRequestQueue, which will at least contain the RERUN request that will trigger the first script-run. When the ScriptRequestQueue is empty, or when a SHUTDOWN request is dequeued, this function will exit and its thread will terminate. zBeginning script threadzbrowser.gatherUsageStatsN) rY_enqueuescript_requests query_stringrZr[r5r^gather_usage_statsr`ra context_info)event client_state)!rr'debugrrd_enqueue_forward_msgrmrgrhrerjr|r rQrkrlrrcurrent_threadon_scriptrunner_readytyperRERUN _run_scriptr{STOPrrpage_script_hashrosendr(r.)rrrrequestrr3r3r4rLs:        zScriptRunner._run_script_threadcCs|jtkS)z=True if the calling function is running in the script thread.)rqrrrwr3r3r4r~sz!ScriptRunner._is_in_script_threadmsgrcCs ||jj|tj|ddS)zEnqueue a ForwardMsg to our browser queue. This private function is called by ScriptRunContext only. It may be called from the script thread OR the main thread. )r forward_msgN)rfrorr(r/)rrrr3r3r4rs  z!ScriptRunner._enqueue_forward_msgcCsX|sdS|js dS|j}|durdS|jtjkr!t|j|jtj ks)Jt )asCheck our current ScriptRequestState to see if we have a pending STOP or RERUN request. This function is called every time the app script enqueues a ForwardMsg, which means that most `st.foo` commands - which generally involve sending a ForwardMsg to the frontend - act as implicit yield points in the script's execution. N) rrprmon_scriptrunner_yieldrrrrr{rr)rrrr3r3r4rfs    z4ScriptRunner._maybe_handle_execution_control_requestccs2|jrtdd|_z dVWd|_dSd|_w)zA context for setting the ScriptRunner._execing flag. Used by _maybe_handle_execution_control_request to ensure that we only handle requests while we're inside an exec() call zNested set_execing_flag callTNF)rprrwr3r3r4_set_execing_flagszScriptRunner._set_execing_flagc ssJ td|t}d}|jstjj |j |j j |j |j }j }|dur;|dn|d}}|j }||krgt} |jdura|jjduradd|jjD} j| t|j} |j|j|| |jdjjtj|| j d z"|dur|d } n|d } t} |j | j_ | | j!"| } Wn)t#y}ztj$d |d d jt%<jjtj&|dWYd}~dSd}~ww'd}|t(j)d<| |j*d<| |||ffdd }t|}t+||\}}}}}|jt%<|rtj,}n |jrtj-}ntj.}|j/rNz| t0|j1t2t|t2||r-t3|j4nddWnt#yM}z tjd|d WYd}~nd}~ww5|||t6t7|dura|}ndSq)zRun our script. Parameters ---------- rerun_data: RerunData The RerunData to use. TzRunning script %srNrcSsh|]}|jqSr3)id)r?wr3r3r4 sz+ScriptRunner._run_script..)rrfragment_ids_this_runr)rrrr; script_pathzScript compilation error)exc_infoF)r exception__main____file__c sHtjx|jdurj|j||jr\|jD]9}z j |}|Wq!t yC|j sAt d|dYq!ttfyR}z|d}~wtyZYq!wntjrdtjt||jjj|jdjWdn1swYWddSWddS1swYdS)NzCouldn't find fragment with id z. This can happen if the fragment does not exist anymore when this request is processed, for example because a full app rerun happened that did not register the fragment. Usually this doesn't happen or no action is required, so its mainly for debugging.)new_fragment_ids)rrer widget_statesrgon_script_will_rerunon_script_startfragment_id_queuerkgetr is_auto_rerunr'warningrrr~ruses_pages_directoryrWexec__dict__clearrmaybe_check_serializablerf)codemodulerr{ fragment_idwrapped_fragmenterwr3r4 code_to_execHsL        (    Pz.ScriptRunner._run_script..code_to_exec)commands exec_time prep_timeuncaught_exceptionzFailed to create page profile)8rr'rtimerrr get_instancemedia_file_mgrclear_session_refsrlset_script_intentr page_nameget_initial_active_script get_main_pagersetrwidgetsrgon_script_finishedlistresetrrrorr(r) get_pagesrpage_not_foundenqueueri get_bytecoder~rr!r* _new_modulesysmodulesrrr,r-r+rrtracked_commandsrrr0_on_script_finished _log_if_error_clean_problem_modules)rrr{ start_timer active_scriptmain_page_inforrprevious_page_script_hash widget_idsrrrrexrr_run_without_errorsrerun_exception_datapremature_stoprfinished_eventr3rwr4rs                A    zScriptRunner._run_scriptrrr(rcCsL|s |j|j|jj||dtjt dr$t ddSdS)zCalled when our script finishes executing, even if it finished early with an exception. We perform post-run cleanup here. )rzrunner.postScriptGCN) rgrwidget_ids_this_runrorr rrremove_orphaned_filesr rQgccollect)rrrrrr3r3r4rs z ScriptRunner._on_script_finishedr<types.ModuleTypecCs t|S)z(Create a new module with the given name.)types ModuleType)rrr<r3r3r4rs zScriptRunner._new_moduleN)rYr6r5r6rZr#r[r&r\rr]rr^r_r`r%rar)rtr6rtry)r{rrtr|)rtr)rtr|)rrrtry)r{rrtry)rrrr(rr|rtry)r<r6rtr)r0r1r2rsrxrzrnrrrrrrfrrrrrr3r3r3r4rXs"  c     2  "   zrXrtrycCstdtjvrz tjd}|jWn tyYnwdtjvr8z tjd}|dWdSty7YdSwdS)z;Some modules are stateful, so we have to clear their state.keraszmatplotlib.pyplotallN)rrbackend clear_sessionr~close)rpltr3r3r4rs      rfnCallable[[], None]c Cs>z|WdSty}z t|WYd}~dSd}~wwru)r~r'r)rrr3r3r4rs  r)r5r6r)rrrtry)H __future__rrrrr contextlibrenumrtimeitrrtypingrrrr r blinkerr streamlitr r rstreamlit.errorsrstreamlit.loggerrstreamlit.proto.ClientState_pb2rstreamlit.proto.ForwardMsg_pb2rstreamlit.runtime.metrics_utilrrstreamlit.runtime.pages_managerr(streamlit.runtime.scriptrunner.exec_coderr+streamlit.runtime.scriptrunner.script_cacher/streamlit.runtime.scriptrunner_utils.exceptionsrr4streamlit.runtime.scriptrunner_utils.script_requestsrrr7streamlit.runtime.scriptrunner_utils.script_run_contextrrr streamlit.runtime.stater!r"r#streamlit.source_utilr$streamlit.runtime.fragmentr%'streamlit.runtime.uploaded_file_managerr&r0r'__annotations__r(rWrXrrr3r3r3r4sN               ( :