o Zh@s"UddlmZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z ddlmZddlmmZddlmZmZmZddlmZddlmZdd lmZmZdd lm Z dd l!m"Z"dd l#m$Z$m%Z%m&Z&m'Z'm(Z(dd l)m*Z*ddl+m,Z,ddl-m.Z.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5m6Z6m7Z7ddl8m9Z9ddl:m;Z;ddlm?Z?e rddl@mAZAddlBmCZCddlDmEZEddlFmGZGddlHmIZIddlJmKZKmLZLeeMZNdeOd<Gdd d eZPd7d#d$ZQGd%d&d&ZRd8d(d)ZSd9d-d.ZTd:d;d2d3ZUdZddAdBZdwddEdFZ      dddRdSZ      dddTdUZddVdWZddXdYZ  dddZd[Zdd^d_Zdd`daZdydbdcZ dwd~dddeZdydfdgZ dydhdiZ!dydjdkZ"ddmdnZ#ddqdrZ$ddudvZ%dS) 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_datar%uploaded_file_managerr) script_cacher&message_enqueued_callbackCallable[[], None] | None user_infodict[str, str | bool | None]session_id_override str | Noner6NonecCs|ptt|_t|_||_||_||_ t |j |j |_ t |_||_tj|_t|_d|_d|_d|_tddkrB|td|_d|_ddlm}||_||_ d|_!t"|_#t$%d|jdS)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)selfr=r>r?r@rBrDr(r4r4r5__init__Vs2-   zAppSession.__init__cC |dS)zGEnsure that we call shutdown() when an AppSession is garbage collected.N)shutdownrbr4r4r5__del__r;zAppSession.__del__cCsH|jdur t|j|_|j|jtj|jdd|_tj |j dS)a4Register 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 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) rUr#rPregister_file_change_callback_on_source_file_changedr on_config_parsedrVr file_change_listenerconnect_on_secrets_file_changedrfr4r4r5rYs  z!AppSession.register_file_watcherscCs\|jdur |j|jdur||jdur|tj|jd|_d|_d|_dS)zJDisconnect the file watcher handlers registered by register_file_watchers.N)rUcloserVrWr rl disconnectrnrfr4r4r5disconnect_file_watcherss     z#AppSession.disconnect_file_watcherslist[ForwardMsg]cCs |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. )rQflushrfr4r4r5flush_browser_queues zAppSession.flush_browser_queuecCsp|jtjkr6td|j|j|jt r(t }|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)rSr-r0r,rarHrMremove_session_filesr exists get_instancemedia_file_mgrclear_session_refsremove_orphaned_filesrequest_script_stoprq)rbrtr4r4r5res   zAppSession.shutdownmsgrcCs0|jr|j|_|j||jr|dSdS)zEnqueue a new ForwardMsg to our browser queue. This can be called on both the main thread and a ScriptRunner run thread. Parameters ---------- msg : ForwardMsg The message to enqueue N)r_debug_last_backmsg_idrQenqueuerRrbr}r4r4r5_enqueue_forward_msgs    zAppSession._enqueue_forward_msgr$c Cs zh|d}|dkr|jr|j|_||jWd S|dkr%|Wd S|dkr0|Wd S|dkr;|Wd S|dkrH||j Wd S|dkrS| Wd S|dkr`| |j Wd St d |Wd Sty}zt d ||WYd }~d Sd }~ww) zProcess a BackMsg.type rerun_script load_git_info clear_cache app_heartbeatset_run_on_save stop_scriptfile_urls_requestzNo handler for "%s"zError processing back messageN) WhichOneofr~r__handle_rerun_script_requestr_handle_git_information_request_handle_clear_cache_request_handle_app_heartbeat_request_handle_set_run_on_save_requestr_handle_stop_script_request_handle_file_urls_requestrr,warning Exception exceptionhandle_backmsg_exception)rbr}msg_typeexr4r4r5handle_backmsgs0  zAppSession.handle_backmsge BaseExceptioncsNjtjjjtjddjtjjfdddS)zGHandle an Exception raised while processing a BackMsg from the browser.)page_script_hashcsSN)r_create_exception_messager4rrbr4r5Qz5AppSession.handle_backmsg_exception..N)_on_scriptrunner_eventr[rSCRIPT_STOPPED_WITH_SUCCESSSCRIPT_STARTEDrKcall_soon_threadsafe)rbrr4rr5r7s z#AppSession.handle_backmsg_exception client_stateClientState | Nonec Cs|jtjkr tddS|rI|j}|r"|j|s"td|dS| dr/|j j |j t |j|j|j|j|r<|nd|jt|j|j d}nt }|jdurnttdrd|jsd|jd|_n |j|}|rndS||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 shutdownNznThe fragment with id %s does not exist anymore - it might have been removed during a preceding full-app rerun. context_info) query_string widget_statesr page_name fragment_id is_auto_reruncached_message_hashesrzrunner.fastReruns)rSr-r0r,rrr`containsinfoHasFieldrTrCopyFromrrrrrrsetrr[boolr rX request_stop request_rerun_create_scriptrunner)rbrr rerun_datasuccessr4r4r5rTsF        zAppSession.request_reruncCs|jdur |jdSdS)zgRequest that the scriptrunner stop execution. Does nothing if no scriptrunner exists. N)r[rrfr4r4r5r{s zAppSession.request_script_stopcCs|jdS)z%Clear the user info for this session.N)r^clearrfr4r4r5clear_user_infoszAppSession.clear_user_infoinitial_rerun_datarc CsLt|j|jj|j|j|j||j|j|j d |_ |j j |j |j dS)z;Create and run a new ScriptRunner with the given RerunData.) session_idrO session_stateuploaded_file_mgrr?rrBfragment_storage pages_managerN)rrHrLrOr]rMrNr^r`rPr[on_eventrmrstart)rbrr4r4r5rs zAppSession._create_scriptrunnerr(cCs|jSr)r]rfr4r4r5rszAppSession.session_statefilepathr7rcs@|jttfddd}|dur|jj}||kSdS)Ncs|dkS)N script_pathr4)krpagesr4r5rrz9AppSession._should_rerun_on_file_change..T)rP get_pagesnextfilterrTr)rbrchanged_page_script_hashcurrent_page_script_hashr4rr5_should_rerun_on_file_changes z'AppSession._should_rerun_on_file_changecCsH|j|dur||sdS|jr||jdS||dS)zfOne of our source files changed. Clear the cache and schedule a rerun if appropriate. N)rNrrrZrrTr_create_file_change_message)rbrr4r4r5rjs z"AppSession._on_source_file_changed_rcCrd)z:Called when `secrets.file_change_listener` emits a Signal.N)rj)rbrr4r4r5rns z#AppSession._on_secrets_file_changedfragment_ids_this_runlist[str] | NonecCs|jjd|ddS)NT)retain_lifecycle_msgsr)rQr)rbrr4r4r5 _clear_queues zAppSession._clear_queuesenderScriptRunner | Noneeventr forward_msgForwardMsg | NonerBaseException | Nonerrdict[PageHash, PageInfo] | Nonec s(jf dddS)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 sSr)(_handle_scriptrunner_event_on_event_loopr4 rrrrrrrrbrr4r5rsz3AppSession._on_scriptrunner_event..N)rKr) rbrrrrrrrrr4rr5rsz!AppSession._on_scriptrunner_eventc Cs:|jtkr td||jurtd|dS|j} |tj krN|jt j kr*t j |_|dur2td||j jkr<||j _||||||} || n|tjtjtjhvr|jt j krbt j|_|tjkrktj} n |tjkrttj} ntj} ||| d|_|tjtjhvr|jr|j|jnj|durtdt} t | j!j"||| nQ|tj#krt j|_||tj$|jr|jn6|tj%kr|durtd|jt j krt&'j()|j*||_ d|_n|tj+kr|durtd||| t j k} |jt j 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. 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. zrThis function must only be called on the eventloop thread the AppSession was created on. This should never happen.z0Ignoring event from non-current ScriptRunner: %sNzTpage_script_hash must be set for the SCRIPT_STARTED event. This should never happen.z`exception must be set for the SCRIPT_STOPPED_WITH_COMPILE_ERROR event. This should never happen.zJclient_state must be set for the SHUTDOWN event. This should never happen.zHnull forward_msg in ENQUEUE_FORWARD_MSG event. This should never happen.)-rKrIrJ RuntimeErrorr[r,rarSrrr-r0r/rTrr_create_new_session_messagerr!SCRIPT_STOPPED_WITH_COMPILE_ERRORFRAGMENT_STOPPED_WITH_SUCCESSr.rFINISHED_SUCCESSFULLY"FINISHED_FRAGMENT_RUN_SUCCESSFULLYFINISHED_WITH_COMPILE_ERROR_create_script_finished_messager_rUupdate_watched_modulesupdate_watched_pagesexception_utilsmarshall session_eventscript_compilation_exceptionSCRIPT_STOPPED_FOR_RERUNFINISHED_EARLY_FOR_RERUNSHUTDOWNr rwrxryrHENQUEUE_FORWARD_MSG&_create_session_status_changed_message)rbrrrrrrrr prev_stater}statusapp_was_runningapp_is_runningr4r4r5rs3                      z3AppSession._handle_scriptrunner_event_on_event_loopcCs$t}|j|j_|jtjk|j_|S)z6Create and return a session_status_changed ForwardMsg.)rrZsession_status_changed run_on_saverSr-r/script_is_runningrr4r4r5rs   z1AppSession._create_session_status_changed_messagecCst}d|j_|S)z8Create and return a 'script_changed_on_disk' ForwardMsg.T)rrscript_changed_on_diskrr4r4r5rsz&AppSession._create_file_change_messagecCst}t|j_|jj|j_|jj|j_|jj|j_||j_ |r'|jj || |j|p1|j t|jjt|jjt|jjjdtjjj|jj}t|jt|j_dtttj |j_!t"j#|j_$dt%j&vpsdt%j&v|j_'|j(|j)_*|j+t,j-k|j)_.|jj/|_/|j0|_1|S)z+Create and return a new_session ForwardMsg.ztheme..DISPLAYWAYLAND_DISPLAY)2rr: new_session script_run_idrLnamerPrOmain_script_hashrrextend_populate_app_pagesr_populate_config_msgr _populate_theme_msg custom_themesidebarCustomThemeCategoriesSIDEBARvalue initialize_populate_user_info_msgrBr"environment_infostreamlit_versionjoinmapr7sys version_infopython_versionr SYSTEM server_ososenviron has_displayrZsession_statusrrSr-r/ris_hellorHr)rbrrrr}imsgr4r4r5rs<           z&AppSession._create_new_session_messager)ForwardMsg.ScriptFinishedStatus.ValueTypecCst}||_|S)z/Create and return a script_finished ForwardMsg.)rscript_finished)rbrr}r4r4r5rsz*AppSession._create_script_finished_messagecCst}t|jjj||S)z*Create and return an Exception ForwardMsg.)rrrdelta new_elementr)rbrr}r4r4r5r sz$AppSession._create_exception_messagec Cst}zlddlm}||jj}|}|durWdS|\}}}|d}||j_||j_ ||j_ |j p5g|jj dd<|j p@g|jj dd<|j rQtjj|j_n|jrbt|jdkrbtjj|j_ntjj|j_||WdSty}ztjd|dWYd}~dSd}~ww)Nr)GitRepoz.gitz+Obtaining Git information produced an errorexc_info)rstreamlit.git_utilrrLrO get_repo_info removesuffixgit_info_changed repositorybranchmoduleuntracked_filesuncommitted_filesis_head_detachedr GitStates HEAD_DETACHEDstate ahead_commitslenAHEAD_OF_REMOTEDEFAULTrrr,ra) rbr}rrepo repo_inforepository_namer&r'rr4r4r5rs0     z*AppSession._handle_git_information_requestcCs||dS)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)rbrr4r4r5r3s z'AppSession._handle_rerun_script_requestcCrd)z1Tell the ScriptRunner to stop running its script.N)r{rfr4r4r5rAr;z&AppSession._handle_stop_script_requestcCs"tjtj|jdS)zjClear this app's cache. Because this cache is global, it will be cleared for all users. N)r cache_datarcache_resourcer]rfr4r4r5rEs  z&AppSession._handle_clear_cache_requestcCsdS)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 Nr4rfr4r4r5rOs z(AppSession._handle_app_heartbeat_request new_valuecCs||_||dS)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)rZrr)rbr7r4r4r5rZs z*AppSession._handle_set_run_on_save_requestrrcCsXt}|j|j_|j|j|j}|D]}|jj t |j |j |j dq||dS)z6Handle a file_urls_request BackMsg sent by the client.)file_id upload_url delete_urlN)r request_idfile_urls_response response_idrMget_upload_urlsrH file_names file_urlsappendrr8r9r:r)rbrr}upload_url_infosupload_url_infor4r4r5rhs z$AppSession._handle_file_urls_requestrdict[PageHash, PageInfo]cCsL|D]\}}|j}||_|ddd|_|d|_|d|_qdS)Nrr icon)items app_pagesaddrreplacer url_pathnamerF)rbr}rr page_info page_protor4r4r5r|s   zAppSession._populate_app_pagesr)r=r%r>r)r?r&r@rArBrCrDrEr6rF)r6rF)r6rr)r}rr6rF)r}r$r6rF)rrr6rF)rrr6rF)rrr6rF)r6r()rr7r6r)rrEr6rF)rrr6rF)rrr6rF)NNNNNN)rrrrrrrrrrrrErrrrr6rF)r6r)NN)rr7rrrrr6r)rrr6r)rrr6r)r7rr6rF)rrr6rF)r}rrrDr6rF)&r1r2r3__doc__rcrgrYrqrtrerrrrr{rrpropertyrrrjrnrrrrrrrrrrrrrrrrr4r4r4r5r<Jsj Z         T         "  +   5  %   r<Config.ToolbarMode.ValueTypecCs^d}t|}ttj|}|dur-dddtjD}td|d|d||S)Nzclient.toolbarModez, css|]}|VqdSr)lower).0rr4r4r5 z$_get_toolbar_mode..zConfig z. expects to have one of the following values: z. Current value: ) r rXgetattrr ToolbarModeupperr keys ValueError) config_key config_value enum_valueallowed_valuesr4r4r5_get_toolbar_modes  r^r}rrFcCsPtd|_td|_td|_td|_tddur"d|_t|_dS)Nzbrowser.gatherUsageStatszglobal.maxCachedMessageAgezserver.allowRunOnSavez ui.hideTopBarzclient.showSidebarNavigationFT) r rXgather_usage_statsmax_cached_message_ageallow_run_on_save hide_top_barhide_sidebar_navr^ toolbar_mode)r}r4r4r5rs     rthemersectionc Cszt|}tdd|DrdS|D]\}}|dvr*|dur*t|t||q|jj|jj d}| dd}|durS||vrNt d|t |n|||_| dd}|r^||_| dd}t|trzt|}Wnty} zt j d || d d}WYd} ~ nd} ~ ww|dur|D]7} zd | vrt| d | d <| d =|jt| tWqty} zt j d | | d WYd} ~ qd} ~ ww| dd} t| tr | } | dr| gd} n%zt| } Wnty } zt j d| | d d} WYd} ~ nd} ~ ww| durO| rt| dkr%tdt| | D]'} z |j| Wq'tyN} zt j d| | d WYd} ~ q'd} ~ ww| dd} t| trzt| } Wnty} zt j d| | d d} WYd} ~ nd} ~ wwt| t rt!ddD] }|j"| qnRt| t r| rt| dkrtdt| | dddgdt| }|D]'}z |j"|Wqty} zt j d|| d WYd} ~ qd} ~ ww| dd}t|trzt|}Wntj#y} zt j d|| d d}WYd} ~ nd} ~ ww|durI|D]'}z |j$|Wq!tyH} zt j d|| d WYd} ~ q!d} ~ ww| dd}t|tr{zt|}Wntj#yz} zt j d|| d d}WYd} ~ nd} ~ ww|durt|dkrt %d t||D])}z |j&|Wqty} zt j d|| d WYd} ~ qd} ~ wwdSdS)!Ncss|]}|duVqdSrr4)rRvalr4r4r5rSrTz&_populate_theme_msg..>basefont fontFacesheadingFontSizesheadingFontWeightschartSequentialColorschartCategoricalColors)lightdarkrhzb"%s" is an invalid value for theme.base. Allowed values include %s. Setting theme.base to "light".rirjzFFailed to parse the theme.fontFaces config option with json.loads: %s.rweight weight_rangez6Failed to parse the theme.fontFaces config option: %s.rk)pxremzMFailed to parse the theme.headingFontSizes config option with json.loads: %s.zUConfig theme.headingFontSizes should have 1-6 values corresponding to h1-h6, but got z=Failed to parse the theme.headingFontSizes config option: %s.rlzOFailed to parse the theme.headingFontWeights config option with json.loads: %s.zWConfig theme.headingFontWeights should have 1-6 values corresponding to h1-h6, but got iXz?Failed to parse the theme.headingFontWeights config option: %s.rnzCFailed to parse the theme.chartCategoricalColors config option: %s.rmzBFailed to parse the theme.chartSequentialColors config option: %s. zuConfig theme.chartSequentialColors should have 10 color values, but got %s. Defaulting to Streamlit's default colors.)'r get_options_for_sectionallvaluesrGsetattrr! BaseThemeLIGHTDARKgetr,rlistrXrh body_font isinstancer7jsonloadsr font_facesrArrstriprQendswithr/rYheading_font_sizesintrangeheading_font_weightsJSONDecodeErrorchart_categorical_colorserrorchart_sequential_colors)r}rf theme_opts option_name option_valbase_maprhrrr font_facersizerrheading_weightsrqrcolorrr4r4r5rsl                         rrcCs$t}|j|_|j|_|j|_dSr)rinstanceinstallation_idinstallation_id_v3installation_id_v4)r}instr4r4r5r s r )r6r7)r6rP)r}rr6rF)re)r}rrfr7r6rF)r}rr6rF)W __future__rrIrrrr8enumrtypingrrrrgoogle.protobuf.json_formatrstreamlit.elements.exceptionelementsrr streamlitr r r streamlit.loggerr streamlit.proto.ClientState_pb2r streamlit.proto.Common_pb2rrstreamlit.proto.ForwardMsg_pb2rstreamlit.proto.GitInfo_pb2rstreamlit.proto.NewSession_pb2rrrrrstreamlit.runtimer#streamlit.runtime.forward_msg_queuerstreamlit.runtime.fragmentrrstreamlit.runtime.metrics_utilrstreamlit.runtime.pages_managerrstreamlit.runtime.scriptrunnerrrrstreamlit.runtime.secretsr streamlit.string_utilr!streamlit.versionr"streamlit.watcherr#streamlit.proto.BackMsg_pb2r$streamlit.runtime.script_datar%+streamlit.runtime.scriptrunner.script_cacher&r\r('streamlit.runtime.uploaded_file_managerr)streamlit.source_utilr*r+r1r,__annotations__r-r:r<r^rrr r4r4r4r5sb                     I  [