gqUddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl mZmZmZddlmZddlmZmZmZdd lmZdd lmZdd lmZdd lmZdd lm Z m!Z!ddl"m#Z#m$Z$ddl%m&Z&ddl'm(Z(m)Z)ddl*m+Z+m,Z,m-Z-ddl.m/Z/m0Z0m1Z1ddl2m3Z3m4Z4m5Z5erddl6m7Z7ddl8m9Z9ddl%m&Z&ddl:m;Z;ee<Z=de>d<Gdde Z? GddZ@ddZAd dZBy)!) annotationsN)contextmanager)Enum) default_timer) TYPE_CHECKINGCallableFinal)Signal)configruntimeutil)FragmentStorageKeyError) get_logger) ClientState) ForwardMsg)create_page_profile_messageto_microseconds)exec_func_with_error_handlingmodified_sys_path) ScriptCache)RerunException StopException) RerunDataScriptRequestsScriptRequestType)ScriptRunContextadd_script_run_ctxget_script_run_ctx)SCRIPT_RUN_WITHOUT_ERRORS_KEYSafeSessionState SessionState)FragmentStorage) PagesManager)UploadedFileManagerr _LOGGERc(eZdZdZdZdZdZdZdZdZ y) 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.a/var/www/openai/venv/lib/python3.12/site-packages/streamlit/runtime/scriptrunner/script_runner.pyr'r'Gs;&N)L%#@ :%D!H 0r3r'ceZdZ ddZddZddZddZddZddZddZ ddZ dd Z dd Z e d Zdd Z dd ZddZy) ScriptRunnerc 0||_||_t||j|_||_||_||_||_| |_ t|_ |jj|td|_d|_d|_y)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) self session_idmain_script_path session_stateuploaded_file_mgr script_cacheinitial_rerun_data user_infofragment_storage pages_managers r4__init__zScriptRunner.__init__xsd&!1. $*V*V #4)#!1+') $$%78  < 8<r3c,tj|SN)r repr_rHs r4__repr__zScriptRunner.__repr__szz$r3c8|jjy)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)rC request_stoprVs r4rYzScriptRunner.request_stops ##%r3c8|jj|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. )rCrD)rH rerun_datas r4rDzScriptRunner.request_reruns~~++J77r3c|j tdtj|jd|_|jj y)zeStart a new thread to process the ScriptEventQueue. This must be called only once. Nz ScriptRunner was already startedzScriptRunner.scriptThread)targetname)rG Exception threadingThread_run_script_threadstartrVs r4rczScriptRunner.startsO    *>? ?'..**,  !!#r3cX|jsJt}| td|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. zOScriptRunner thread has a null ScriptRunContext. Something has gone very wrong!)_is_in_script_threadr RuntimeError)rHctxs r4_get_script_run_ctxz ScriptRunner._get_script_run_ctxs< ((*** " ;1  r3c|jsJtjdt|j|j |j d|j|j|j|jttjd|j|j }t!t#j$||j j'}|j(t*j,k(rS|j/|j0|j j'}|j(t*j,k(rS|j(t*j2k(sJt5}|j6|_|j8|_|j:j=|t>j@|y)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.gatherUsageStats) rI_enqueuescript_requests query_stringrKrLrJrOgather_usage_statsrPrQ)event client_stateN)!rer%debugrr:_enqueue_forward_msgrCr=r>r;r@boolr get_optionrArBrr`current_threadon_scriptrunner_readytyperRERUN _run_scriptr[STOPrrmpage_script_hashrEsendr'r-)rHrgrequestrps r4rbzScriptRunner._run_script_threadso((*** /0''.. NN--"55!33oo#F$5$56P$QR!33--   9335s;..668ll/555   W// 0nn::&>&@@@r3c||j|jj|tj|y)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. )ro forward_msgN)r<rEr|r'r.)rHmsgs r4rrz!ScriptRunner._enqueue_forward_msgTs7 446  )==3  r3c.|jsy|jsy|jj}|y|jt j k(rt|j|jt jk(sJt)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) rerFrCon_scriptrunner_yieldrwrrxrr[rzr)rHr}s r4r<z4ScriptRunner._maybe_handle_execution_control_requestds((* }} ..668 ?  <<,22 2 !3!34 4||055555or3c#xK|jr tdd|_ dd|_y#d|_wxYww)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)rFrfrVs r4_set_execing_flagzScriptRunner._set_execing_flags8 ===> >  " !DMEDMs:.: 7:c jsJ tjd|t}d}|js,t j jjjj|j|jjj|j|j}jj}||dn|d}j}|j}||k7rtt!} |j"C|j"j$-|j"j$D chc]} | j&} } j(j+| t-|j} |j/|j0|| j2j5t6j8|| jj; ||d} n;|d} t=} |j| j>_ |jA| jBjE| }jMd }|tNjPd <| |jRd <||||ffd }t|z }tU||\}}}}}|j(tH<|rt6jV}n-|jrt6jX}nt6jZ}|j\r] |jAt_|j`tct|z tc||rte|jfndji|||tktl||}nycc} w#tF$r`}tjd| d j(tH<j2j5t6jJ| Yd}~yd}~wwxYw#tF$r!}tjd| Yd}~d}~wwxYw)zRun our script. Parameters ---------- rerun_data: RerunData The RerunData to use. zRunning script %srNr{)rmr{fragment_ids_this_run)ror{rpages script_pathzFatal script error)exc_infoF)ro exception__main____file__c tj5j5|j%jj |j|j |jr5|jD]%} jj|}|'n.code_to_execsl%d&<&<=**,"//;++@@&44'')!33+5+G+GK$%373I3I3M3M$/4" 0!1 2 ,HPT6??3..44-0-A-A5''@@B@@Bu->=&$;&(2'?'?$+OO*I+WR)R%&%3M#B('(#,% !% %W-,>=s`FAE4"D"(A)E4F".E1 E4E1 E" " E1 .E40E1 1E44E= 9FF )commands exec_time prep_timeuncaught_exceptionzFailed to create page profile)7rer%rqtimerrr get_instancemedia_file_mgrclear_session_refsrBset_script_intentr{ page_nameget_initial_active_script get_main_pagerhsetrwidgetsidr=on_script_finishedlistresetrmrEr|r'r( get_pagesrpage_not_foundenqueuer? get_bytecoder_rr) _new_modulesysmodulesrrr+r,r*rnrtracked_commandsrrwr/_on_script_finished _log_if_error_clean_problem_modules)rHr[ start_timer active_scriptmain_page_infor{rgprevious_page_script_hash widget_idswrrrrexrr_run_without_errorsrerun_exception_datapremature_stoprfinished_events` r4ryzScriptRunner._run_scripts((*** MM-z : %J I// $$&55HHJ    1 1++Z-A-A !//II++Z-A-AM"00>>@N!,01#$67  **,C),(<(< %(,<<(+u ,,8"0088D0:0H0H0P0P!Q0P1!$$0PJ!Q##66zB$()E)E$F ! II'44!1&;   MM  '66!1&;))335    ,"/ ">K#1"?K%,C3=3G3GC&&0KK$))66{C&%%j1F'-CKK #+6FOOJ '"&v3:= C~*,I.lC@  "$"BTD   = >#"3!K!K--!2!P!P!2!N!N%%PKK3%(%9%9&5eg 6J&K&5i&@$6!%%7 8 A A%)    $ $S.. I 0 1#/1 OX"RT  2R @EJ##$AB ""+MM #  H!PMM"ABMOOPs9N$AN)AP) P2AP  P P?P::P?c:|s%|jj|j|jj ||t j jjtjdrtjdyy)zCalled when our script finishes executing, even if it finished early with an exception. We perform post-run cleanup here. )rozrunner.postScriptGCN) r=rwidget_ids_this_runrEr|r rrremove_orphaned_filesr rtgccollect)rHrgrors r4rz ScriptRunner._on_script_finishedsx    2 233J3J K 4u- --CCE   2 3 JJqM 4r3c,tj|S)z(Create a new module with the given name.)types ModuleType)rHr^s r4rzScriptRunner._new_modules%%r3N)rIstrrJrrKr!rLr$rMrrNrrOzdict[str, str | bool | None]rPr"rQr#)returnrrNone)r[rrrs)rr)rrs)rrrr)r[rrr)rgrror'rrsrr)r^rrztypes.ModuleType)r/r0r1rRrWrYrDrcrhrbrerrr<rrryrrr2r3r4r6r6wsa<a<a<$ a< / a< " a<&a<0a<*a<$ars#  %)11++4'75D   :<GKH%%00>"o&o&d,r3