gUddlmZddlZddlZddlZddlmZddlmZm Z m Z ddl m cm ZddlmZmZddlmZddlmZddlmZmZdd lmZdd lmZdd lmZmZm Z m!Z!dd l"m#Z#dd l$m%Z%ddl&m'Z'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.m/Z/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6ddl7m8Z8er,ddl9m:Z:ddl;mZ>ddl?m@Z@ddlAmBZBddlCmDZDddlEmFZFmGZGeeHZIdeJd<Gdd eZKd(d!ZLGd"d#ZMd)d$ZNd*d%ZOd+d&ZPd,d'ZQy)-) annotationsN)Enum) TYPE_CHECKINGCallableFinal)configruntime) get_logger) ClientState)FileURLsFileURLsRequest) ForwardMsg)GitInfo)ConfigCustomThemeConfig NewSessionUserInfo)caching)ForwardMsgQueue)FragmentStorageMemoryFragmentStorage) Installation) PagesManager) RerunData ScriptRunnerScriptRunnerEvent)secrets_singleton) to_snake_case)STREAMLIT_VERSION_STRING)LocalSourcesWatcher)BackMsg) PagesChanged) ScriptData) ScriptCache SessionState)UploadedFileManager)PageHashPageInfor_LOGGERceZdZdZdZdZy)AppSessionStateAPP_NOT_RUNNINGAPP_IS_RUNNINGSHUTDOWN_REQUESTEDN)__name__ __module__ __qualname__r-r.r/R/var/www/openai/venv/lib/python3.12/site-packages/streamlit/runtime/app_session.pyr,r,;s'O%N-r4r,c<ttjS)z5Randomly generate a unique ID for a script execution.)struuiduuid4r3r4r5_generate_scriptrun_idr:As tzz| r4ceZdZdZ d% d&dZd'dZd'dZd'dZd(dZd'dZ d)d Z d*d Z d+d Z d,d Z d'd Zd'dZd-dZed.dZd/dZd%d0dZd'dZd'dZd%d1dZ d2 d3dZ d2 d3dZd4dZd4dZ d5 d6dZ d7dZd8dZd'dZ d% d,dZ d'dZ!d'd Z"d'd!Z#d9d"Z$d:d#Z% d;d$Z&y)< 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. Nc|xsttj|_t j |_||_||_||_ t|j|j|_ t|_||_t j"|_t'|_d|_d|_d|_t1j2ddk7r|j5t1j2d|_d|_ddlm}||_||_ d|_!tE|_#tHjKd|jy)aInitialize 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. 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.fileWatcherTypenonezserver.runOnSaverr%zAppSession initialized (id=%s))&r7r8r9idasyncioget_running_loop _event_loop _script_data_uploaded_file_mgr _script_cachermain_script_path_pages_managerr_browser_queue_message_enqueued_callbackr,r-_stater _client_state_local_sources_watcher_stop_config_listener_stop_pages_listenerr get_optionregister_file_watchers _run_on_save _scriptrunnerstreamlit.runtime.stater&_session_state _user_info_debug_last_backmsg_idr_fragment_storager*debug)self script_datauploaded_file_manager script_cachemessage_enqueued_callback user_infosession_id_overrider&s r5__init__zAppSession.__init__Rs!Z&:TZZ\):"335'"7)*  ( ($*<*< ./*C'%55 )]BF#@D"?C!   5 6& @  ' ' )"--.@A26 9*n#26#2G2I 6@r4c$|jy)zGEnsure that we call shutdown() when an AppSession is garbage collected.N)shutdownrYs r5__del__zAppSession.__del__s  r4c|jt|j|_|jj|jt j |jd|_|jj|j|_ tjj|jy)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)rLr rGregister_file_change_callback_on_source_file_changedron_config_parsedrMregister_pages_changed_callback_on_pages_changedrNrfile_change_listenerconnect_on_secrets_file_changedrcs r5rPz!AppSession.register_file_watcherss  & & .*=d>Q>Q*RD ' ##AA  ( ( &,%<%<  ( (& "%)$7$7$W$W  " "% ! ..66t7T7TUr4c<|j|jj|j|j|j|jtj j |jd|_d|_d|_y)zJDisconnect the file watcher handlers registered by register_file_watchers.N)rLcloserMrNrrl disconnectrnrcs r5disconnect_file_watchersz#AppSession.disconnect_file_watcherss  & & 2  ' ' - - /  % % 1  & & (  $ $ 0  % % '..99$:W:WX&*#%)"$(!r4c6|jjS)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. )rHflushrcs r5flush_browser_queuezAppSession.flush_browser_queues""((**r4c|jtjk7rtj d|j |j j|j tjrStj}|jj|j |jj|jtj|_|jyy)ziShut down the AppSession. It's an error to use a AppSession after it's been shut down. zShutting down (id=%s)N)rJr,r/r*rXr?rDremove_session_filesr exists get_instancemedia_file_mgrclear_session_refsremove_orphaned_filesrequest_script_stoprr)rYrts r5rbzAppSession.shutdowns ;;/<< < MM1477 ;  # # 8 8 A~~))+!!44TWW=!!779  $ $ &)<),,252K2KD/11#2B2BC_,446]*002_,224..44S5H5HI]*00200..s/D/DE 5x@ .   = >  ) )" - - .sBADD'D=D D4D D+D D; &D66D;cFjjtjjjtjdjjtjj j fdy)zGHandle an Exception raised while processing a BackMsg from the browser.)page_script_hashcDjjSN)r_create_exception_message)erYsr5z5AppSession.handle_backmsg_exception..PsD--d.L.LQ.OPr4N)_on_scriptrunner_eventrRrSCRIPT_STOPPED_WITH_SUCCESSSCRIPT_STARTEDrBcall_soon_threadsafe)rYrs``r5rz#AppSession.handle_backmsg_exception6s ##    1 M M  ##     , , $ ##    1 M M  -- P r4c|jtjk(rtj dy|r|j }|r5|j j|stjd|dyt|j|j|j|j|r|nd|j}n t}|jjt!t#j$dr.|j s"|jj'd|_n|jj)|}|ry|j+|y)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 shutdownNzThe fragment with id zW does not exist anymore - it might have been removed during a preceding full-app rerun.) fragment_id is_auto_rerunzrunner.fastReruns)rJr,r/r*rrrWcontainsinfor query_string widget_statesr page_namerrRboolrrO request_stop request_rerun_create_scriptrunner)rY client_stater rerun_datasuccesss r5rzAppSession.request_rerunSs* ;;/<< < OOE F  &22K"4#9#9#B#B;#O +K=9TT"))**--&&+6KD*88 J#J    )V&&':;<".. ""//1%)" ,,:::F !!*-r4cR|j|jjyy)zgRequest that the scriptrunner stop execution. Does nothing if no scriptrunner exists. N)rRrrcs r5r}zAppSession.request_script_stops'    )    + + - *r4c8|jjy)z%Clear the user info for this session.N)rUclearrcs r5clear_user_infozAppSession.clear_user_infos r4c |t|j|jj|j|j |j ||j|j|j |_ |jjj|j|jjy)z;Create and run a new ScriptRunner with the given RerunData.) session_idrF session_stateuploaded_file_mgrr\initial_rerun_datar^fragment_storage pages_managerN)rr?rCrFrTrDrErUrWrGrRon_eventrmrstart)rYrs r5rzAppSession._create_scriptrunners)ww!..??--"55++1oo!33--   ##++D,G,GH   "r4c|jSr)rTrcs r5rzAppSession.session_states"""r4c|jjttfdd}||jj }||k(Sy)Nc|dk(S)N script_pathr3)kfilepathpagess r5rz9AppSession._should_rerun_on_file_change..sU1Xm4@r4T)rG get_pagesnextfilterrKr)rYrchanged_page_script_hashcurrent_page_script_hashrs ` @r5_should_rerun_on_file_changez'AppSession._should_rerun_on_file_changesY##--/#' @% H $  $ /'+'9'9'J'J $+/GG Gr4c|jj||j|sy|jr|j |j y|j |jy)zfOne of our source files changed. Clear the cache and schedule a rerun if appropriate. N)rErrrQrrKr_create_file_change_message)rYrs r5rhz"AppSession._on_source_file_changeds`   "  (I(I((S       t11 2  % %d&F&F&H Ir4c$|jy)z:Called when `secrets.file_change_listener` emits a Signal.N)rh)rY_s r5rnz#AppSession._on_secrets_file_changeds $$&r4ct}|j|j|jj |j ||j |j jyyr)r_populate_app_pages pages_changedrGrrrLupdate_watched_pages)rYrrs r5rkzAppSession._on_pages_changeds_l   !2!2D4G4G4Q4Q4ST !!#&  & & 2  ' ' < < > 3r4c>|jjd|y)NT)retain_lifecycle_msgsfragment_ids_this_run)rHr)rYrs r5 _clear_queuezAppSession._clear_queues! !!"&>S " r4c bjjf dy)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. c 4 jSr)(_handle_scriptrunner_event_on_event_loop) reventr forward_msgrrrrYsenders r5rz3AppSession._on_scriptrunner_event..s(DAA % r4N)rBr) rYrrrrrrrrs `````````r5rz!AppSession._on_scriptrunner_events$" --   r4c T|jtjk(sJd||jurtj d|y|j } |tjk(r|j tjk7rtj|_|Jd||jjk7r||j_ |j||j|j!|||n|tj"k(s'|tj$k(s|tj&k(rj|j tjk7rtj(|_|tj"k(rt*j,} n4|tj&k(rt*j.} nt*j0} |j|j3| d|_|tj"k(s|tj&k(rC|j6r|j6j9|j6j;ny|Jdt+} t=j>| j@jB||j| n)|tjDk(rjtj(|_|j|j3t*jF|j6r|j6j9n|tjHk(rl|Jd|j tjk(r7tKjLjNjQ|jR||_ d|_n-|tjTk(r|Jd|j|| tjk(} |j tjk(} | | k7r |j|jWyy)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. fragment_ids_this_run : list[str] | None The fragment IDs of the fragments being executed in this script run. Only set for the SCRIPT_STARTED event. If this value is falsy, this script run must be for the full script. clear_forward_msg_queue : bool If set (the default), clears the queue of forward messages to be sent to the browser. 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),rBr@rArRr*rXrJrrr,r/r.rKrrr_create_new_session_messager!SCRIPT_STOPPED_WITH_COMPILE_ERRORFRAGMENT_STOPPED_WITH_SUCCESSr-rFINISHED_SUCCESSFULLY"FINISHED_FRAGMENT_RUN_SUCCESSFULLYFINISHED_WITH_COMPILE_ERROR_create_script_finished_messagerVrLupdate_watched_modulesrexception_utilsmarshall session_eventscript_compilation_exceptionSCRIPT_STOPPED_FOR_RERUNFINISHED_EARLY_FOR_RERUNSHUTDOWNr ryrzr{r?ENQUEUE_FORWARD_MSG&_create_session_status_changed_message)rYrrrrrrrr prev_statestatusrapp_was_runningapp_is_runnings r5rz3AppSession._handle_scriptrunner_event_on_event_loopsf7#;#;#== f = ++ + MMLe T [[ %44 4{{o@@@-<< #/ K / 4#5#5#F#FF6F""3   3 4  % %00$&;U  &BB B)KKK)GGG{{o@@@-== )EEE#99+III#FF#??  % %d&J&J6&R S*.D '*FFF-KKK ..//FFH//DDF!,[,!l((%%BBI))#. '@@ @)99DK  % %4477  **++BBD '00 0+ A +{{o@@@$$&55HHQ!-D !%D  ';; ;* ? *  % %k 2%(F(FF(F(FF _ ,  % %d&Q&Q&S T -r4ct}|j|j_|jt j k(|j_|S)z6Create and return a session_status_changed ForwardMsg.)rrQsession_status_changed run_on_saverJr,r.script_is_runningrs r5rz1AppSession._create_session_status_changed_messagesDl151B1B"". KK?99 9 ""4 r4c<t}d|j_|S)z8Create and return a 'script_changed_on_disk' ForwardMsg.T)rrscript_changed_on_diskrs r5rz&AppSession._create_file_change_messagesl370 r4cdt}t|j_|jj |j_|j j|j_|j j|j_||j_ |r%|jjj||j|j|xs|j jt|jjt!|jj"|jj$}t'|j(t*|j,_dj1t3t4t6j8|j,_|j<|j>_ |jBtDjFk(|j>_$|jjJ|_%|jL|_'|S)z+Create and return a new_session ForwardMsg..)(rr: new_session script_run_idrCnamerGrFmain_script_hashrrextendrr_populate_config_msgr_populate_theme_msg custom_theme initialize_populate_user_info_msgr^renvironment_infostreamlit_versionjoinmapr7sys version_infopython_versionrQsession_statusrrJr,r.ris_hellor?r)rYrrrrimsgs r5rz&AppSession._create_new_session_messagesl(>(@%#0055+/+>+>+O+O(+/+>+>+O+O(+;( OO 1 1 8 89N O   OOUEd&9&9&C&C&E  S__334COO889 ))/2J//2xxCAQAQ8R/S,*.*;*;' KK?99 9 -))22 '' r4c(t}||_|S)z/Create and return a script_finished ForwardMsg.)rscript_finished)rYrrs r5rz*AppSession._create_script_finished_messagesl$ r4ct}tj|jjj ||S)z*Create and return an Exception ForwardMsg.)rrrdelta new_elementr)rYrrs r5rz$AppSession._create_exception_messages.l  !6!6!@!@!D r4clt} ddlm}||jj}|j }|y|\}}}|j dr|dd}||j_||j_ ||j_ |j|jjdd|j|jjdd|jr*tjj |j_nkt%|j&dkDr*tjj(|j_n)tjj*|j_|j-|y#t.$r!}t0j3d|Yd}~yd}~wwxYw)Nr)GitRepoz.gitz+Obtaining Git information produced an error)exc_info)rstreamlit.git_utilrrCrF get_repo_infoendswithgit_info_changed repositorybranchmoduleuntracked_filesuncommitted_filesis_head_detachedr GitStates HEAD_DETACHEDstatelen ahead_commitsAHEAD_OF_REMOTEDEFAULTrrr*rX) rYrrrepo repo_inforepository_namer$r%rs r5rz*AppSession._handle_git_information_requestshl! V 24,,==>D**,I .7 +OVV''/"1#2"6.=C +*0C '*0C '6:6J6JC 0 0 38<8N8NC 2 21 5$$-4->->-L-L$$*T''(1,-4->->-N-N$$*-4->->-F-F$$*  % %c * V MMGRTM U U Vs4F EF F3F..F3c&|j|y)aTell the ScriptRunner to re-run its script. Parameters ---------- client_state : streamlit.proto.ClientState_pb2.ClientState | None The ClientState protobuf to run the script with, or None to use previous client state. N)r)rYrs r5rz'AppSession._handle_rerun_script_request&s <(r4c$|jy)z1Tell the ScriptRunner to stop running its script.N)r}rcs r5rz&AppSession._handle_stop_script_request4s   "r4ctjjtjj|jjy)zjClear this app's cache. Because this cache is global, it will be cleared for all users. N)r cache_datarcache_resourcerTrcs r5rz&AppSession._handle_clear_cache_request8s:   "$$& !!#r4cy)zHandle an incoming app heartbeat. The heartbeat indicates the frontend is active and keeps the websocket from going idle and disconnecting. The actual handler here is a noop Nr3rcs r5rz(AppSession._handle_app_heartbeat_requestBs r4cP||_|j|jy)zChange our run_on_save flag to the given value. The browser will be notified of the change. Parameters ---------- new_value : bool New run_on_save value N)rQrr)rY new_values r5rz*AppSession._handle_set_run_on_save_requestMs#& !!$"M"M"OPr4c|t}|j|j_|jj |j |j}|D]Q}|jjjt|j|j|jS|j|y)z6Handle a file_urls_request BackMsg sent by the client.)file_id upload_url delete_urlN)r request_idfile_urls_response response_idrDget_upload_urlsr? file_names file_urlsappendr r<r=r>r)rYrrupload_url_infosupload_url_infos r5rz$AppSession._handle_file_urls_request[sl->-I-I*22BB GG&11  0O  " " , , 3 3+33.99.99  0 !!#&r4c|jD]T\}}|jj}||_|dj dd|_|d|_|d|_Vy)Nrr icon)items app_pagesaddrreplacer url_pathnamerJ)rYrrr page_info page_protos r5rzAppSession._populate_app_pagesosg,1;;= ' i**,J*:J '#,[#9#A#A#s#KJ &/ &z$_get_toolbar_mode..s"P6O17796OszConfig z. expects to have one of the following values: z. Current value: ) rrOgetattrr ToolbarModeupperrkeys ValueError) config_key config_value enum_valueallowed_valuess r5_get_toolbar_moderis%J$$Z0L6=L..07J"Pf6H6H6M6M6O"PPj^$%%3$45*^ -  r4c.tjd|_tjd|_tjd|_tjd|_tjddurd|_t|_y)Nzbrowser.gatherUsageStatszglobal.maxCachedMessageAgezserver.allowRunOnSavez ui.hideTopBarzclient.showSidebarNavigationFT) rrOgather_usage_statsmax_cached_message_ageallow_run_on_save hide_top_barhide_sidebar_navri toolbar_moders r5rrsx#../IJC!'!2!23O!PC"--.EFC((9C 78EA#(*Cr4c ddh}tjd}t|jsy|j D]#\}}||vs |t |t ||%|jj|jjd}|d}|B||vr4tjd|dt|jdn |||_|jj |jj"|jj$d}|d}|C||vr4tjd|d t|jd y|||_yy) 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".)rget_options_for_sectionanyvaluesrKsetattrr BaseThemeLIGHTDARKr*rlistrcrs FontFamily SANS_SERIFSERIF MONOSPACErt) renum_encoded_options theme_opts option_name option_valbase_maprsfont_maprts r5rrsv"F+//8J z  " ##-#3#3#5 Z 2 2z7M C{3Z @$6$$ ""H f D  x  OOD6++/ +@*AB22   ~CHnn//%%^^--H f D  x  OOD6++/ +@*AB77   ~CHr4ctjj|_tjj|_yr)rinstanceinstallation_idinstallation_id_v3rqs r5r r s0&//1AAC)224GGCr4)rSr7)rSzConfig.ToolbarMode.ValueType)rrrSrT)rrrSrT)rrrSrT)R __future__rr@rr8enumrtypingrrrstreamlit.elements.exceptionelementsrr streamlitrr streamlit.loggerr streamlit.proto.ClientState_pb2r streamlit.proto.Common_pb2r r streamlit.proto.ForwardMsg_pb2rstreamlit.proto.GitInfo_pb2rstreamlit.proto.NewSession_pb2rrrrstreamlit.runtimer#streamlit.runtime.forward_msg_queuerstreamlit.runtime.fragmentrrstreamlit.runtime.metrics_utilrstreamlit.runtime.pages_managerrstreamlit.runtime.scriptrunnerrrrstreamlit.runtime.secretsrstreamlit.string_utilrstreamlit.versionrstreamlit.watcherr streamlit.proto.BackMsg_pb2r! streamlit.proto.PagesChanged_pb2r"streamlit.runtime.script_datar#+streamlit.runtime.scriptrunner.script_cacher$rSr&'streamlit.runtime.uploaded_file_managerr'streamlit.source_utilr(r)r0r*__annotations__r,r:r<rirrr r3r4r5rs# 1166%'7@5/ &?M78UU7/613=8G4K8H%%.d.  r 0r 0t  +,&^Hr4