&e4ddlZddlZddlZddlmZddlmZmZmZm Z m Z m Z ddl m cmZddlmZmZmZddlmZddlmZddlmZddlmZdd lmZmZdd lm Z dd l!m"Z"dd l#m$Z$m%Z%m&Z&m'Z'dd l(m)Z)ddl*m+Z+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4m5Z5m6Z6ddl7m8Z8ddl9m:Z:ddl;mZ>ddl?m@Z@eeAZBerddlCmDZDGddeZEdeFfdZGGddZHd'd ZId!e$ddfd"ZJd!e%ddfd#ZKd!e'ddfd$ZLd!e e&e)fd%eFddfd&ZMdS)(N)Enum) TYPE_CHECKINGCallableDictListOptionalUnion)configruntime source_util) to_snake_case) get_logger)BackMsg) ClientState)FileURLsFileURLsRequest) ForwardMsg)GitInfo)ConfigCustomThemeConfig NewSessionUserInfo) PagesChanged)cachinglegacy_caching)ForwardMsgQueue) Installation) ScriptData) RerunData ScriptRunnerScriptRunnerEvent) ScriptCache)secrets_singleton)UploadedFileManager)STREAMLIT_VERSION_STRING)LocalSourcesWatcher SessionStateceZdZdZdZdZdS)AppSessionStateAPP_NOT_RUNNINGAPP_IS_RUNNINGSHUTDOWN_REQUESTEDN)__name__ __module__ __qualname__r+r,r-=lib/python3.11/site-packages/streamlit/runtime/app_session.pyr*r*4s 'O%N-r2r*returncBttjS)z5Randomly generate a unique ID for a script execution.)struuiduuid4r1r2r3_generate_scriptrun_idr9:s tz||  r2ceZdZdZ d:dedededeegdfde de e ee fd ee d dfd Z d;d Z d;d Zd;dZd eefdZd;dZded dfdZded dfdZded dfdZdeed dfdZd;dZded dfdZed AppSessiona! Contains session data for a single "user" of an active app (that is, a connected browser tab). Each AppSession has its own ScriptData, root DeltaGenerator, ScriptRunner, and widget state. An AppSession is attached to each thread involved in running its script. N script_datauploaded_file_manager script_cachemessage_enqueued_callbacklocal_sources_watcher user_infosession_id_overrider4cB|pttj|_t j|_||_||_||_ t|_ ||_ tj|_t!|_||_d|_d|_|t-jd|_d|_ddlm}||_||_d|_t> d|jdS)aoInitialize the AppSession. Parameters ---------- script_data Object storing parameters related to running a script uploaded_file_manager Used to manage files uploaded by users via the Streamlit web client. script_cache The app's ScriptCache instance. Stores cached user scripts. ScriptRunner uses the ScriptCache to avoid having to reload user scripts from disk on each rerun. message_enqueued_callback After enqueuing a message, this callable notification will be invoked. local_sources_watcher The file watcher that lets the session know local files have changed. 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. session_id_override The ID to assign to this session. 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. Nzserver.runOnSaverr'zAppSession initialized (id=%s))!r6r7r8idasyncioget_running_loop _event_loop _script_data_uploaded_file_mgr _script_cacher_browser_queue_message_enqueued_callbackr*r+_stater _client_state_local_sources_watcher_stop_config_listener_stop_pages_listenerregister_file_watchersr get_option _run_on_save _scriptrunnerstreamlit.runtime.stater(_session_state _user_info_debug_last_backmsg_idLOGGERdebug) selfr<r=r>r?r@rArBr(s r3__init__zAppSession.__init__Ks`&:TZ\\):):"355'"7) .//*C'%5 )]] " #DH"BF! ##%%%"-.@AA59 988888*lnn#59# 5tw?????r2c.|dS)zGEnsure that we call shutdown() when an AppSession is garbage collected.N)shutdownr\s r3__del__zAppSession.__del__s r2cR|jt|jj|_|j|jt j|jd|_tj |j |_ tj|jdS)a8Register handlers to be called when various files are changed. Files that we watch include: * source files that already exist (for edits) * `.py` files in the the main script's `pages/` directory (for file additions and deletions) * project and user-level config.toml files * the project-level secrets.toml files This method is called automatically on AppSession construction, but it may be called again in the case when a session is disconnected and is being reconnect to. NT) force_connect)rOr&rHmain_script_pathregister_file_change_callback_on_source_file_changedr on_config_parsedrPr register_pages_changed_callback_on_pages_changedrQr#file_change_listenerconnect_on_secrets_file_changedr`s r3rRz!AppSession.register_file_watcherss  & .*=!2++D ' #AA  (   &,%<  (& & & "%0$O  "% % ! .66t7TUUUUUr2c$|j|j|j||j|tj|jd|_d|_d|_dS)zJDisconnect the file watcher handlers registered by register_file_watchers.N)rOcloserPrQr#rj disconnectrlr`s r3disconnect_file_watchersz#AppSession.disconnect_file_watcherss  & 2  ' - - / / /  % 1  & & ( ( (  $ 0  % % ' ' '.99$:WXXX&*#%)"$(!!!r2c4|jS)aiClear the forward message queue and return the messages it contained. The Server calls this periodically to deliver new messages to the browser connected to this app. Returns ------- list[ForwardMsg] The messages that were removed from the queue and should be delivered to the browser. )rKflushr`s r3flush_browser_queuezAppSession.flush_browser_queues"((***r2c|jtjkrtd|j|j|jtj rKtj }|j |j|j |tj|_|dSdS)ziShut down the AppSession. It's an error to use a AppSession after it's been shut down. zShutting down (id=%s)N)rMr*r-rZr[rDrIremove_session_filesr exists get_instancemedia_file_mgrclear_session_refsremove_orphaned_filesrequest_script_stoprp)r\rts r3r_zAppSession.shutdowns ;/< < < LL0$' : : :  # 8 8 A A A~ :)++!44TW===!77999  $ $ & & &))),L252KD/11#2BCCCCC_,,4466666]**0022222_,,2244444...44S5HIIIII]**0022222000..s/DEEEEE4h????? . . . LL     ) )" - - - - - - - - - .sBAD D(DD DD DD E!'/EE!ec(jtjjtjdjtjjfddS)zGHandle an Exception raised while processing a BackMsg from the browser.)page_script_hashcTSN)r_create_exception_message)rr\sr3z5AppSession.handle_backmsg_exception..Ms#D--d.L.LQ.O.OPPr2N)_on_scriptrunner_eventrUr!SCRIPT_STOPPED_WITH_SUCCESSSCRIPT_STARTEDrGcall_soon_threadsafe)r\rs``r3rz#AppSession.handle_backmsg_exception3s ##   1 M    ##    , $   ##   1 M    -- P P P P P     r2 client_statec|jtjkrtddS|r't |j|j|j|j }nt }|j `ttj dr!|j d|_ n|j |}|rdS||dS)aSignal that we're interested in running the script. If the script is not already running, it will be started immediately. Otherwise, a rerun will be requested. Parameters ---------- client_state : streamlit.proto.ClientState_pb2.ClientState | None The ClientState protobuf to run the script with, or None to use previous client state. z'Discarding rerun request after shutdownNzrunner.fastReruns)rMr*r-rZrr query_string widget_statesr page_namerUboolr rS request_stop request_rerun_create_scriptrunner)r\r rerun_datasuccesss r3rzAppSession.request_rerunPs ;/< < < NND E E E F  %")*-& JJ#J   )F%&9::;;  "//111%)"",:::FFF !!*-----r2cJ|j|jdSdS)zgRequest that the scriptrunner stop execution. Does nothing if no scriptrunner exists. N)rUrr`s r3r{zAppSession.request_script_stops0   )   + + - - - - - * )r2initial_rerun_datac t|j|jj|j|j|j||j|_|jj |j |j dS)z;Create and run a new ScriptRunner with the given RerunData.) session_idrd session_stateuploaded_file_mgrr>rrAN) r rDrHrdrWrIrJrXrUon_eventrkrstart)r\rs r3rzAppSession._create_scriptrunners{)w!.?-"5+1o    #++D,GHHH   """""r2r(c|jSr)rWr`s r3rzAppSession.session_states ""r2filepathc|jj}tj|t t fdd}||jj}||kSdS)Nc(|dkS)N script_pathr1)krpagess r3rz9AppSession._should_rerun_on_file_change..sU1Xm4@r2T)rHrdr get_pagesnextfilterrNr)r\rrdchanged_page_script_hashcurrent_page_script_hashrs ` @r3_should_rerun_on_file_changez'AppSession._should_rerun_on_file_changess,=%&677#' @@@@@% H H $ $  $ /'+'9'J $+/GG Gtr2c|j|||sdS|jr||jdS||dS)zUOne of our source files changed. Clear the cache and schedule a rerun if appropriate.N)rJclearrrTrrNr_create_file_change_message)r\rs r3rfz"AppSession._on_source_file_changeds   """  (I(I((S(S  F   J   t1 2 2 2 2 2  % %d&F&F&H&H I I I I Ir2c.|dS)z:Called when `secrets.file_change_listener` emits a Signal.N)rf)r\_s r3rlz#AppSession._on_secrets_file_changeds $$&&&&&r2ct}t|j|jj|||j|jdSdSr)r_populate_app_pages pages_changedrHrdrrOupdate_watched_pages)r\rr}s r3rizAppSession._on_pages_changedsdllC-t/@/QRRR !!#&&&  & 2  ' < < > > > > > 3 2r2c8|jdSr)rKrr`s r3 _clear_queuezAppSession._clear_queues !!#####r2senderevent forward_msg exceptionrc ZjfddS)aCalled when our ScriptRunner emits an event. This is generally called from the sender ScriptRunner's script thread. We forward the event on to _handle_scriptrunner_event_on_event_loop, which will be called on the main thread. c8Sr)(_handle_scriptrunner_event_on_event_loop)rrrrrr\rsr3rz3AppSession._on_scriptrunner_event..s%DAA{I|EUr2N)rGr)r\rrrrrrs```````r3rz!AppSession._on_scriptrunner_eventsb --               r2c|jtjks Jd||jurtd|dS|j}|tjkrp|jtj krtj |_| Jd| | ||n|tjks|tjkr|jtj krtj|_|tjk}||r t&jn t&j} | | d|_|r"|jr|jnV| Jdt'} t3j| jj|| | n|tjkrU|t&j} | | |jr|jn|tjkr`| Jd|jtj kr0tAj!j"#|j$||_%d|_n1|tj&kr!| Jd| ||tj k} |jtj k} | | kr)| |'dSdS)a_Handle a ScriptRunner event. This function must only be called on our eventloop thread. Parameters ---------- sender : ScriptRunner | None The ScriptRunner that emitted the event. (This may be set to None when called from `handle_backmsg_exception`, if no ScriptRunner was active when the backmsg exception was raised.) event : ScriptRunnerEvent The event type. forward_msg : ForwardMsg | None The ForwardMsg to send to the frontend. Set only for the ENQUEUE_FORWARD_MSG event. exception : BaseException | None An exception thrown during compilation. Set only for the SCRIPT_STOPPED_WITH_COMPILE_ERROR event. client_state : streamlit.proto.ClientState_pb2.ClientState | None The ScriptRunner's final ClientState. Set only for the SHUTDOWN event. page_script_hash : str | None A hash of the script path corresponding to the page currently being run. Set only for the SCRIPT_STARTED event. zXThis function must only be called on the eventloop thread the AppSession was created on.z0Ignoring event from non-current ScriptRunner: %sNz9page_script_hash must be set for the SCRIPT_STARTED eventzEexception must be set for the SCRIPT_STOPPED_WITH_COMPILE_ERROR eventz/client_state must be set for the SHUTDOWN eventz-null forward_msg in ENQUEUE_FORWARD_MSG event)(rGrErFrUrZr[rMr!rr*r-r,rr_create_new_session_messager!SCRIPT_STOPPED_WITH_COMPILE_ERRORr+_create_script_finished_messagerFINISHED_SUCCESSFULLYFINISHED_WITH_COMPILE_ERRORrYrOupdate_watched_modulesexception_utilsmarshall session_eventscript_compilation_exceptionSCRIPT_STOPPED_FOR_RERUNFINISHED_EARLY_FOR_RERUNSHUTDOWNr rwrxryrDrNENQUEUE_FORWARD_MSG&_create_session_status_changed_message) r\rrrrrr prev_statescript_succeededscript_finished_msgr}app_was_runningapp_is_runnings r3rz3AppSession._handle_scriptrunner_event_on_event_loopsdR   8 : : : : : e ; : : + + + LLKU S S S F[ %4 4 4{o@@@-< !,,J-,,        % %001ABB     &B B B)KKK{o@@@-= $(9(UU "&"F"F#< 00;##   % %&9 : : :*.D ' /.I/FFHHH ))Z*)) ll(%BI))#.... '@ @ @"&"F"F3##   % %&9 : : :* E+BBDDD '0 0 0((@)(({o@@@$&&5HHQQQ!-D !%D   '; ; ;''>(''  % %k 2 2 2%(FF(FF _ , ,  % %d&Q&Q&S&S T T T T T - ,r2ct}|j|j_|jt jk|j_|S)z6Create and return a session_status_changed ForwardMsg.)rrTsession_status_changed run_on_saverMr*r,script_is_runningrs r3rz1AppSession._create_session_status_changed_messagegs8ll151B". K?9 9 "4 r2c:t}d|j_|S)z8Create and return a 'script_changed_on_disk' ForwardMsg.T)rrscript_changed_on_diskrs r3rz&AppSession._create_file_change_messagepsll370 r2ct}t|j_|jj|j_|jj|j_||j_t|j|jjt|jj t|jj |jj }t|jt |j_dt)t*t,j|j_|j|j_|jt:jk|j_|jj |_ |j!|_"|S)z+Create and return a new_session ForwardMsg..)#rr9 new_session script_run_idrHnamerdrr_populate_config_msgr _populate_theme_msg custom_theme initialize_populate_user_info_msgrAr%environment_infostreamlit_versionjoinmapr6sys version_infopython_versionrTsession_statusrrMr*r,ris_hellorDr)r\rr}imsgs r3rz&AppSession._create_new_session_messagevsll(>(@(@%#05+/+<+M(+;(COT->-OPPPS_3444CO8999 )///2J//2xxCAQ8R8R/S/S,*.*;' K?9 9 -)2 ' r2statusz)ForwardMsg.ScriptFinishedStatus.ValueTypec0t}||_|S)z/Create and return a script_finished ForwardMsg.)rscript_finished)r\rr}s r3rz*AppSession._create_script_finished_messagesll$ r2cjt}tj|jjj||S)z*Create and return an Exception ForwardMsg.)rrrdelta new_elementr)r\rr}s r3rz$AppSession._create_exception_messages+ll !6!@!DDD r2ct} ddlm}||jj}|}|dS|\}}}|dr |dd}||j_||j_ ||j_ |j |jj dd<|j |jj dd<|j rtjj|j_nOt%|jdkrtjj|j_ntjj|j_||dS#t.$r'}t0d|Yd}~dSd}~wwxYw)Nr)GitRepoz.gitz+Obtaining Git information produced an error)exc_info)rstreamlit.git_utilrrHrd get_repo_infoendswithgit_info_changed repositorybranchmoduleuntracked_filesuncommitted_filesis_head_detachedr GitStates HEAD_DETACHEDstatelen ahead_commitsAHEAD_OF_REMOTEDEFAULTrrrZr[) r\r}rrepo repo_inforepository_namerrrs r3rz*AppSession._handle_git_information_requestsll! U 2 2 2 2 2 274,=>>D**,,I .7 +OVV''// 7"1#2#"6.=C +*0C '*0C '6:6JC 0 38<8NC 2111 5$ G-4->-L$**T'((1,,-4->-N$**-4->-F$*  % %c * * * * * U U U LLFQSL T T T T T T T T T Us1ECt}|j|j_|j|j|j}|D]@}|jj t|j |j |j A||dS)z6Handle a file_urls_request BackMsg sent by the client.)file_id upload_url delete_urlN)r request_idfile_urls_response response_idrIget_upload_urlsrD file_names file_urlsappendrr/r0r1r)r\rr}upload_url_infosupload_url_infos r3rz$AppSession._handle_file_urls_requestsll->-I*2BB G&1   0  O  " , 3 3+3.9.9     !!#&&&&&r2r)r4N)r4r()NNNN)7r.r/r0__doc__rr$r"rrr&rr6r]rarRrprrrsr_rrr BaseExceptionrrrr{rrpropertyrrrrfrlrirr r!rrrrrrrrrrrrrrrr1r2r3r;r;?s=  &.2W@W@W@ 3W@" W@ $,HRX,>#? W@ 3 W@Xc]*+W@&c]W@ W@W@W@W@rVVVV< ) ) ) ) +T*%5 + + + +,,,,8. .t....,.'.d....: - D    :-.(;*?-.D-.-.-.-.^.... #y #T # # # ####X# S T     J J J J J J J''''????$$$$-1-1.2*.  & ! j)  M*  {+  #3-      2-1-1.2*.GUGU&GU!GUj) GU M* GU {+ GU#3-GU GUGUGUGUR Z C J    DA =Z $U$U$U$UN59 ) )$[1 )  ) ) ) )#### $ $ $ $      Q Q$ Q Q Q Q'?'t''''''r2r;Config.ToolbarMode.ValueTypec4d}tj|}ttj|}|SddtjD}td|d|d||S)Nzclient.toolbarModez, c3>K|]}|VdSr)lower).0rs r3 z$_get_toolbar_mode..#s*"P"P17799"P"P"P"P"P"Pr2zConfig z. expects to have one of the following values: z. Current value: ) r rSgetattrr ToolbarModeupperrkeys ValueError) config_key config_value enum_valueallowed_valuess r3_get_toolbar_moderMs%J$Z00L;BL..00<<J"P"Pf6H6M6M6O6O"P"P"PPP -j - -%3 - -* - -   r2r}cdtjd|_tjd|_tjd|_tjd|_tjd|_tjddkrd|_t|_dS) Nzbrowser.gatherUsageStatszglobal.maxCachedMessageAgezserver.allowRunOnSavez ui.hideTopBarzui.hideSidebarNavzclient.showSidebarNavigationFT) r rSgather_usage_statsmax_cached_message_ageallow_run_on_save hide_top_barhide_sidebar_navrM toolbar_moder}s r3rr,s#./IJJC!'!23O!P!PC"-.EFFC(99C!,-@AAC 788EAA#(**Cr2c ddh}tjd}t|sdS|D])\}}||vr |t |t ||*|jj|jj d}|d}|R||vrAt d|dt| dn |||_|jj|jj|jjd}|d}|U||vrBt d|d t| d dS|||_dSdS) Nbasefonttheme)lightdark"z=" is an invalid value for theme.base. Allowed values include z . Setting theme.base to "light".)z sans serifserif monospacez=" is an invalid value for theme.font. Allowed values include z%. Setting theme.font to "sans serif".)r get_options_for_sectionanyvaluesitemssetattrr BaseThemeLIGHTDARKrZrlistrGrW FontFamily SANS_SERIFSERIF MONOSPACErX) r}enum_encoded_options theme_opts option_name option_valbase_maprWfont_maprXs r3rr8s"F+/88J z  "" # ##-#3#3#5#5AA Z 2 2 2z7M C{33Z @ @ @$ "H f D  x   NN2D22+/ +@+@222      ~CHn/%^-H f D  x   NN7D77+/ +@+@777       ~CHHHr2cztjj|_tjj|_dSr)rinstanceinstallation_idinstallation_id_v3rUs r3rrgs0&/11AC)244GCr2rdctj|D]?\}}|j}||_|d|_|d|_@dS)Nricon)r rrb app_pagesaddrrrw)r}rdr page_info page_protos r3rrlsq(3'<=M'N'N'T'T'V'V,,#)]&&(( &6 #(5 #F+  ,,r2)r4r>)NrErr7enumrtypingrrrrrr streamlit.elements.exceptionelementsrr streamlitr r r streamlit.case_convertersr streamlit.loggerrstreamlit.proto.BackMsg_pb2rstreamlit.proto.ClientState_pb2rstreamlit.proto.Common_pb2rrstreamlit.proto.ForwardMsg_pb2rstreamlit.proto.GitInfo_pb2rstreamlit.proto.NewSession_pb2rrrr streamlit.proto.PagesChanged_pb2rstreamlit.runtimerr#streamlit.runtime.forward_msg_queuerstreamlit.runtime.metrics_utilrstreamlit.runtime.script_datarstreamlit.runtime.scriptrunnerrr r!+streamlit.runtime.scriptrunner.script_cacher"streamlit.runtime.secretsr#'streamlit.runtime.uploaded_file_managerr$streamlit.versionr%streamlit.watcherr&r.rZrVr(r*r6r9r;rMrrrrr1r2r3rs GGGGGGGGGGGGGGGG6666666662222222222333333''''''//////777777@@@@@@@@555555////// :9999955555555??????777777444444UUUUUUUUUUCCCCCC777777GGGGGG666666111111 H  5444444.....d...  U 'U 'U 'U 'U 'U 'U 'U 'z     +f + + + + +,&.,&4,&,&,&,&^HHdHHHH , z<' (,