U /e;@sdZddlmZddlZeeZddlZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZddlmZmZdd lmZdd lmZmZmZmZdd lmZdd l m!Z!d dlm"Z"m#Z#m$Z$m%Z%d dl&m'Z'e r,ddl(m)Z)ddl*m+Z+ddlm,Z,ddl-m.Z.d dl/m0Z0d dlm1Z1m2Z2m3Z3dZ4egdfZ5edefZ6eegdfZ7eegdfZ8GdddZ9ddddddZ:d d!dd"d#d$Z;dd%d%dd&d'ZZd?dd d@dAZdBddCdDdEZdFdGddHdIdJZd2dKddLdMdNZdOdd dPdQZdBdd dRdSZd2dKddTdUdVZd"ddWdXdYZd2d1ddZd[d\Zd]dd^d_d`ZdaddbdcddZdddedfZdS)ir%z Manage and provide access to all of the models that belong to a Bokeh Document. The set of "all models" means specifically all the models reachable from references form a Document's roots. zweakref.ReferenceType[Document] _documentz!Dict[Any, DocumentChangeCallback]_change_callbackszDict[str, List[EventCallback]]_event_callbacksz Dict[str, List[MessageCallback]]_message_callbackszSet[SessionDestroyedCallback]_session_destroyed_callbackszSet[SessionCallback]_session_callbacksz,Dict[str, Set[weakref.ReferenceType[Model]]]_subscribed_modelsNzHoldPolicyType | None_holdList[DocumentChangedEvent] _held_eventsr!)documentcCs^t||_i|_tt|_tt|_t|_ t|_ tt|_ d|_ g|_ |d|jdS)z Args: document (Document): A Document to manage models for A weak reference to the Document will be retained NZ bokeh_event)weakrefrefr'r(rlistr)r*setr+r,r-r.r0 on_messagetrigger_json_event)selfr1r9r9r9r:session_destroyed_callbackssz3DocumentCallbackManager.session_destroyed_callbacksNone) callbacksr=cCs ||_dSNr@)r8rCr9r9r:rAsr Callbackbool) callback_objcallbackone_shotr=csp}|dkrtd|r@tddfdd }|}n}t||_jt|S)a Internal implementation for adding session callbacks. Args: callback_obj (SessionCallback) : A session callback object that wraps a callable and is passed to ``trigger_on_change``. callback (callable) : A callable to execute when session events happen. one_shot (bool) : Whether the callback should immediately auto-remove itself after one execution. Returns: SessionCallback : passed in as ``callback_obj``. Raises: ValueError, if the callback has been previously added Nz@Attempting to add session callback to already-destroyed DocumentrBr<csjkrSrD)r,remove_session_callbackr9rHrGr8r9r:remove_then_invokes  zHDocumentCallbackManager.add_session_callback..remove_then_invoke) r' RuntimeErrorr_wrap_with_curdocZ _callbackr,addtrigger_on_changer)r8rGrHrIdocrLZactual_callbackr9rKr:add_session_callbacks  z,DocumentCallbackManager.add_session_callbackcCs.|j|`|j|`|j|`dS)z6 Clean up references to the Documents models N)r(clearr)r*r>r9r9r:destroys    zDocumentCallbackManager.destroycombiner)policyr=cCsR|jdk r2|j|kr2td|jd|ddS|tkrHtd|||_dS)Nzhold already active with 'z ', ignoring ''zUnknown hold policy )r.logwarningr ValueError)r8rVr9r9r:holds zDocumentCallbackManager.holdcCs|jSrD)r.r>r9r9r: hold_valuesz"DocumentCallbackManager.hold_valuerstrrzDocumentPatchedEvent | Nonez Setter | NonezInvoker | None)modelattroldnewhintsettercallback_invokerr=c Csp|}|dkrdS|dkr,|j||||dkrF|||} nd} t|||||| ||| } || dS)z* Called by Model when it changes Nname)r'modelsZ update_namelookupZserializable_valuerrP) r8r^r_r`rarbrcrdrQZserializable_neweventr9r9r: notify_changesz%DocumentCallbackManager.notify_changerr$)r^rhrdr=cCs"|}|dkrdSt||dSz N)r'r&)r8r^rhrdrQr9r9r: notify_eventsz$DocumentCallbackManager.notify_eventr"cGs.|D]$}||jkrqt|d||j|<qdS)zm Provide callbacks to invoke if the document or any Model reachable from its roots changes. rhN)r(rr8rCrHr9r9r: on_changes   z!DocumentCallbackManager.on_changer)receiverr=cs |jkrfdd|j<dS)Ncs |SrD)dispatchrlror9r:z?DocumentCallbackManager.on_change_dispatch_to..r()r8ror9rqr:on_change_dispatch_tos z-DocumentCallbackManager.on_change_dispatch_tozstr | Type[Event] EventCallback)rhrCr=cGsrt|tst|tr|j}|tkr0td|tt|tsFtd|D]}t|dddqJ|j | |dS)zD Provide callbacks to invoke if a bokeh event is received. zUnknown event zDocument.on_event may only be used to subscribe to events of type DocumentEvent. To subscribe to a ModelEvent use the Model.on_event method.rlzEvent callback)whatN) isinstancer] issubclassr event_namerrZrrr)extend)r8rhrCrHr9r9r:on_event sz DocumentCallbackManager.on_eventMessageCallback)msg_typerCr=cGs|j||dSrD)r*r{)r8r~rCr9r9r:r6sz"DocumentCallbackManager.on_messagercGs$|D]}t|d|j|qdS)za Provide callbacks to invoke when the session serving the Document is destroyed )Zsession_contextN)rr+rOrmr9r9r:on_session_destroyed s z,DocumentCallbackManager.on_session_destroyedcGs|D] }|j|=qdS)z Remove a callback added earlier with ``on_change``. Raises: KeyError, if the callback was never added Nrtrmr9r9r:remove_on_change)sz(DocumentCallbackManager.remove_on_change)r~rHr=cCs,|j|d}|dk r(||kr(||dSrj)r*getremove)r8r~rHZmessage_callbacksr9r9r:remove_on_message3sz)DocumentCallbackManager.remove_on_message)rGr=cCsfz|g}|j|Wntk r2tdYnX|}|dkrHdS|D]}|t||qLdS)z Remove a callback added earlier with ``add_periodic_callback``, ``add_timeout_callback``, or ``add_next_tick_callback``. Returns: None Raises: KeyError, if the callback was never added zDcallback already ran or was already removed, cannot be removed againN)r,rKeyErrorrZr'rPr)r8rGZ callback_objsrQr9r9r:rJ;s z/DocumentCallbackManager.remove_session_callback)keyr^r=cCs|j|t|dSrD)r-rOr2r3)r8rr^r9r9r: subscribeTsz!DocumentCallbackManager.subscriber)jsonr=cCszt|}Wn"tk r0td|YnXt|trj|j|j }|D]}|}|rP| |qP|j |jgD] }||qzdS)NzCould not decode event json: %s) rZ decode_jsonrZrXrYrxrr-rzcopyZ_trigger_eventr)r)r8rrhZ subscribedZ model_refr^cbr9r9r:r7Ws  z*DocumentCallbackManager.trigger_json_eventr#)rhr=cs}|dkrdSjdkr.jdSjdkrHtjdSjdk r^t|jddfdd }t||dS)NZcollectrUrBr<csjD] }|q dSrD)r(values)rrhr8r9r:invoke_callbackswszCDocumentCallbackManager.trigger_on_change..invoke_callbacks)r'r.r0append_combine_document_eventsrdr&)r8rhrQrr9rr:rPhs      z)DocumentCallbackManager.trigger_on_changecCs<|jdkrdSd|_t|j}g|_|D]}||q(dS)zo Turn off any active document hold and apply any collected events. Returns: None N)r.r4r0rP)r8eventsrhr9r9r:unhold|s  zDocumentCallbackManager.unhold)rU)NNN)__name__ __module__ __qualname____doc____annotations__r.r;propertyr?rArcrRrTr[r\rirkrnrur|r6rrrrJrr7rPrr9r9r9r:r%UsN  ,     r%r!zCallable[[], None]rB)rQfr=c CsJddlm}t|ddr t|n|}|||W5QRSQRXdS)Nr ) patch_curdocZnolockF)Zio.docrgetattrr)rQrrZcurdocr9r9r:r&s  r&r#r/) new_event old_eventsr=cCs,t|D]}||rdSq||dS)a Attempt to combine a new event with a list of previous events. The ``old_event`` will be scanned in reverse, and ``.combine(new_event)`` will be called on each. If a combination can be made, the function will return immediately. Otherwise, ``new_event`` will be appended to ``old_events``. Args: new_event (DocumentChangedEvent) : The new event to attempt to combine old_events (list[DocumentChangedEvent]) A list of previous events to attempt to combine new_event with **This is an "out" parameter**. The values it contains will be modified in-place. Returns: None N)reversedrUr)rrrhr9r9r:rs  rzCallable[..., Any]cs$tddddfdd }|S)NrrB)argskwargsr=cs(tddfdd }t|S)Nrr<cs SrDr9r9)rrrr9r:invokesz2_wrap_with_curdoc..wrapper..invoke)rr&)rrrrQr)rrr:wrappersz"_wrap_with_curdoc..wrapperr)rQrrr9rr:rNsrN)=r __future__rlogging getLoggerrrXr2 collectionsr functoolsrtypingrrrr r r r Z core.enumsrrZ core.typesrrrrrrr^rZutil.callback_managerrrrrrZlockingrZapplication.applicationrZcore.has_propsrrZserver.callbacksr r1r!r"r#r$__all__rEZ Originatorr}rvr%r&rrNr9r9r9r:s@    $            ?