3 d\@sddZGdddZdS)cCst||||||d}|dS)a Add/Remove notifiers on objects following the description on an ObserverGraph. All nodes in ``ObserverGraph`` are required to be instances of ``IObserver``. The interface of ``IObserver`` supports this function. Parameters ---------- object : object An object to be observed. graph : ObserverGraph A graph describing what and how extended traits are being observed. All nodes must be ``IObserver``. handler : callable(event) User-defined callable to handle change events. ``event`` is an object representing the change. Its type and content depends on the change. target : Any An object for defining the context of the user's handler notifier. This is typically an instance of HasTraits seen by the user as the "owner" of the observer. dispatcher : callable(callable, event) Callable for dispatching the user-defined handler, e.g. dispatching callback on a different thread. remove : boolean If true, notifiers are being removed. Raises ------ NotiferNotFound Raised when notifier cannot be found for removal. )objectgraphhandlertarget dispatcherremoveN)_AddOrRemoveNotifier)rrrrrrZ callable_r=/tmp/pip-build-7vycvbft/traits/traits/observation/_observe.pyadd_or_remove_notifiers s"r c@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)rzo Callable for adding or removing notifiers. See ``add_or_remove_notifiers`` for the input parameters. cCs.||_||_||_||_||_||_g|_dS)N)rrrrrr _processed)selfrrrrrrrrr __init__?sz_AddOrRemoveNotifier.__init__c Cs|j|j|j|jg}|jr(|ddd}yx|D] }|q0WWnLtk rx4|jr|jj\}}|jrx|j|qR|j |qRWYn X|jj dS)z6 Main function for adding/removing notifiers. N) _add_or_remove_notifiers_add_or_remove_maintainers!_add_or_remove_children_notifiers_add_or_remove_extra_graphsr Exceptionr popadd_to remove_fromclear)r Zstepsstepnotifier observablerrr __call__Js"  z_AddOrRemoveNotifier.__call__c Cs<x6|jjj|jD]"}t|j||j|j|j|jdqWdS)z~ Add or remove additional ObserverGraph contributed by the root observer. e.g. for handing trait_added event. )rrrrrrN) rnodeZiter_extra_graphsr rrrrr)r Z extra_graphrrr rksz0_AddOrRemoveNotifier._add_or_remove_extra_graphsc CsLxF|jjD]:}x4|jjj|jD] }t|||j|j|j|j dq Wq WdS)zM Recursively add or remove notifiers for the children ObserverGraph. )rrrrrrN) rchildrenrZ iter_objectsrr rrrr)r child_graphZ next_objectrrr rysz6_AddOrRemoveNotifier._add_or_remove_children_notifierscCsxxr|jjj|jD]^}xX|jjD]L}|jjj||j|j|jd}|j rR|j |n |j ||j j ||fq WqWdS)z Add or remove notifiers for maintaining children notifiers when the objects being observed by the root observer change. )rrrrN)rriter_observablesrrget_maintainerrrrrrrr append)r rrZchange_notifierrrr rs   z/_AddOrRemoveNotifier._add_or_remove_maintainerscCsr|jjjsdSx^|jjj|jD]J}|jjj|j|j|jd}|j rP|j |n |j ||j j ||fq WdS)z] Add or remove user notifiers for the objects observed by the root observer. N)rrr)rrnotifyr rZ get_notifierrrrrrrr r")r rrrrr rs    z-_AddOrRemoveNotifier._add_or_remove_notifiersN) __name__ __module__ __qualname____doc__r rrrrrrrrr r9s !rN)r rrrrr  s-