a @DfA@sdZddlmZddlZeeZddlZddlm Z ddl m Z ddl m Z mZmZddlmZmZdd lmZmZmZmZdd lmZdd lmZdd lmZd dlmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%d dl&m'Z'e rddl(m)Z)ddl*m+Z+ddl,m-Z-d dl.m/Z/d dlm0Z0m1Z1m2Z2dZ3egdfZedefZ4eegdfZ5eegdfZ6GdddZ7ddddddZ8dd dd!d"d#Z9dd$d$dd%d&Z:dS)'zX Encapulate the management of Document callbacks with a DocumentCallbackManager class. ) annotationsN) defaultdictwraps) TYPE_CHECKINGAnyCallable) HoldPolicyHoldPolicyType)_CONCRETE_EVENT_CLASSES DocumentEventEvent ModelEvent)Model)Callback)_check_callback)DocumentPatchedEventModelChangedEventRootAddedEventRootRemovedEventSessionCallbackAddedSessionCallbackRemovedTitleChangedEvent)UnlockedDocumentProxy)SessionDestroyedCallback)Setter)SessionCallback)Document)DocumentChangeCallbackDocumentChangedEventInvoker) DocumentCallbackManagerrinvoke_with_curdocrrrrrr.c @s.eZdZUdZded<ded<ded<ded <d ed <d ed <ded<ded<dZded<ded<ddddZeddddZed ddd Z e j d d!d"d#d Z d$d%d&d$d'd(d)Z d!dd*d+Z dtd-d!d.d/d0Z eddd1d2Zdud3d4d5d5d6d7d8d!d9d:d;Zd3dd?d@ZdAd!d"dBdCZd5d!dDdEdFZdGdHd!dIdJdKZdGdLd!dIdMdNZdGdOd!dIdPdQZd4dRd!dSdTdUZdVd!d"dWdXZd5d!d"dYdZZd4dRd!d[d\d]Zd$d!d^d_d`Zd4d3d!dadbdcZd4dddedfdgZdhddidjZdkd!dldmdnZdod!dldpdqZd!ddrdsZ dS)vr#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[JSEventCallback]]_js_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)documentcCsht||_i|_tt|_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(r)setr*r+r,r-r/ on_message trigger_event)selfr0r8e/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/bokeh/document/callbacks.py__init__ms     z DocumentCallbackManager.__init__zlist[SessionCallback]returncCs t|jS)zA A list of all the session callbacks for this document. )r3r+r7r8r8r9session_callbackssz)DocumentCallbackManager.session_callbackscCs|jS)zN A list of all the on_session_destroyed callbacks for this document. r*r=r8r8r9session_destroyed_callbackssz3DocumentCallbackManager.session_destroyed_callbacksNone) callbacksr<cCs ||_dSNr?)r7rBr8r8r9r@srrbool) callback_objcallbackone_shotr<csp}|durtd|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 DocumentrAr;csjvrSrC)r+remove_session_callbackr8rFrEr7r8r9remove_then_invokes  zHDocumentCallbackManager.add_session_callback..remove_then_invoke) r% RuntimeErrorr_wrap_with_curdocZ _callbackr+addtrigger_on_changer)r7rErFrGdocrJZactual_callbackr8rIr9add_session_callbacks  z,DocumentCallbackManager.add_session_callbackcCs<|j|`|j|`|j|`|j|`dS)z6 Clean up references to the Documents models N)r&clearr'r(r)r=r8r8r9destroys    zDocumentCallbackManager.destroycombiner )policyr<cCsR|jdur2|j|kr2td|jd|ddS|tvrHtd|||_dS)Nzhold already active with 'z ', ignoring ''zUnknown hold policy )r-logwarningr ValueError)r7rTr8r8r9holds zDocumentCallbackManager.holdcCs|jSrC)r-r=r8r8r9 hold_valuesz"DocumentCallbackManager.hold_valuerstrrzDocumentPatchedEvent | Nonez Setter | NonezInvoker | None)modelattroldnewhintsettercallback_invokerr<c Cs|}|durdS|dkr,|j||||durX|||}t||||||} n(|jdusfJ||_|jdur|||_|} || dS)z* Called by Model when it changes Nname) r%modelsZ update_namelookup get_valuerrbrarN) r7r\r]r^r_r`rarbrOeventr8r8r9 notify_changes z%DocumentCallbackManager.notify_changerr")r\rgrbr<cCs"|}|durdSt||dSz N)r%r$)r7r\rgrbrOr8r8r9 notify_eventsz$DocumentCallbackManager.notify_eventr cGs.|D]$}||jvrqt|d||j|<qdS)zm Provide callbacks to invoke if the document or any Model reachable from its roots changes. rgN)r&rr7rBrFr8r8r9 on_changes   z!DocumentCallbackManager.on_change)receiverr<cs |jvrfdd|j<dS)Ncs |SrC)dispatchrkrnr8r9z?DocumentCallbackManager.on_change_dispatch_to..r&)r7rnr8rpr9on_change_dispatch_tos z-DocumentCallbackManager.on_change_dispatch_tozstr | type[Event] EventCallback)rgrBr<cGs|j|g|RdS)zD Provide callbacks to invoke if a bokeh event is received. N _on_eventr7rgrBr8r8r9on_eventsz DocumentCallbackManager.on_eventJSEventCallbackcGs|j|g|RdS)zG Provide JS callbacks to invoke if a bokeh event is received. Nrvrxr8r8r9 js_on_eventsz#DocumentCallbackManager.js_on_eventzEventCallback | JSEventCallbackcGst|tst|tr|j}|tvr0td|tt|tsFtd|D]>}t|trj|j | |qJt |ddd|j | |qJdS)NzUnknown 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.rkzEvent callback)what) isinstancer[ issubclassr event_namer rXr rzr(appendrr')r7rgrBrFr8r8r9rw"s z!DocumentCallbackManager._on_eventMessageCallback)msg_typerBr<cGs|j||dSrC)r)extend)r7rrBr8r8r9r55sz"DocumentCallbackManager.on_messagercGs$|D]}t|d|j|qdS)za Provide callbacks to invoke when the session serving the Document is destroyed )Zsession_contextN)rr*rMrlr8r8r9on_session_destroyed8s 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 Nrsrlr8r8r9remove_on_changeAsz(DocumentCallbackManager.remove_on_change)rrFr<cCs,|j|d}|dur(||vr(||dSri)r)getremove)r7rrFZmessage_callbacksr8r8r9remove_on_messageKsz)DocumentCallbackManager.remove_on_message)rEr<cCsdz|g}|j|Wnty0tdYn0|}|durFdS|D]}|t||qJdS)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+rKeyErrorrXr%rNr)r7rEZ callback_objsrOr8r8r9rHSs  z/DocumentCallbackManager.remove_session_callback)keyr\r<cCs|j|t|dSrC)r,rMr1r2)r7rr\r8r8r9 subscribelsz!DocumentCallbackManager.subscribeztuple[EventCallback, ...])rr<cCst|j|gS)z Return a tuple containing all current event callbacks for the given event name. Args: event_name (str) : the event name to look up callbacks for )tupler'r)r7rr8r8r9event_callbacks_for_event_nameosz6DocumentCallbackManager.event_callbacks_for_event_namez"tuple[DocumentChangeCallback, ...]cCst|jS)zB Return a tuple containing all current change callbacks. )rr&valuesr=r8r8r9change_callbacksysz(DocumentCallbackManager.change_callbacksr)rgr<cCsVt|tr8|j|j}|D]}|}|r||q||jD] }||qDdSrC)r}rr,rcopyZ_trigger_eventr)r7rgZ subscribedZ model_refr\cbr8r8r9r6s  z%DocumentCallbackManager.trigger_eventr!cs}|durdSjdkr.jdSjdkrHtjdSjdur^t|jddfdd }t||dS)NZcollectrSrAr;csD] }|qdSrC)r)rrgr7r8r9invoke_callbackss zCDocumentCallbackManager.trigger_on_change..invoke_callbacks)r%r-r/r_combine_document_eventsrbr$)r7rgrOrr8rr9rNs      z)DocumentCallbackManager.trigger_on_changecCs<|jdurdSd|_t|j}g|_|D]}||q(dS)zo Turn off any active document hold and apply any collected events. Returns: None N)r-r3r/rN)r7eventsrgr8r8r9unholds  zDocumentCallbackManager.unhold)rS)NNN)!__name__ __module__ __qualname____doc____annotations__r-r:propertyr>r@rarPrRrYrZrhrjrmrtryr{rwr5rrrrHrrrr6rNrr8r8r8r9r#VsT  ,      r#rzCallable[[], None]rA)rOfr<cCsVddlm}t|ddr t|n|}|||WdS1sH0YdS)Nr ) patch_curdocZnolockF)Zio.docrgetattrr)rOrrZcurdocr8r8r9r$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)reversedrSr)rrrgr8r8r9rs  rzCallable[..., Any]cs$tddddfdd }|S)NrrA)argskwargsr<cs(tddfdd }t|S)Nrr;csiSrCr8r8)rrrr8r9invokesz2_wrap_with_curdoc..wrapper..invoke)rr$)rrrrOr)rrr9wrappersz"_wrap_with_curdoc..wrapperr)rOrrr8rr9rLsrL);r __future__rlogging getLoggerrrVr1 collectionsr functoolsrtypingrrrZ core.enumsr r rr r rrr\rZmodels.callbacksrrzZutil.callback_managerrrrrrrrrZlockingrZapplication.applicationrZcore.has_propsrZserver.callbacksrr0rr r!r"__all__Z Originatorrrur#r$rrLr8r8r8r9s>       $       a