U /e2"@sdZddlmZddlZeeZddlZddlZddl m Z m Z m Z m Z mZmZddlmZddlmZddlmZe rd d lmZd ZGd d d ZdS)zR Encapulate the management of Document models with a DocumentModelManager class. ) annotationsN) TYPE_CHECKINGDict GeneratorIteratorListSet)ID)Model)MultiValuedDict)Document)DocumentModelManagerc@sJeZdZUdZded<ded<ded<ded <eZd ed <d d ddZddddZdddddZ ddddddZ dddddZ ddd d!Z ddd"d#Z ejd$dd%d&Zd'd(d)d*d+Zdd,dd-d.Zd'd,d)d/d0Zddd1d2Zddd3d4Zdddd5d6Zdd7d7dd8d9d:Zddd;d<Zddd=d>Zd?S)@rz 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] _documentint _freeze_countzDict[ID, Model]_modelszMultiValuedDict[str, Model]_models_by_namezSet[ID]_seen_model_idsr)documentcCs,t||_d|_i|_t|_t|_dS)z Args: document (Document): A Document to manage models for A weak reference to the Document will be retained rN) weakrefrefrrrr rsetr)selfrr9/tmp/pip-unpacked-wheel-f5fndrjf/bokeh/document/models.py__init__Gs  zDocumentModelManager.__init__)returncCs t|jSN)lenrrrrr__len__UszDocumentModelManager.__len__r r )idrcCs |j|Srrrr#rrr __getitem__Xsz DocumentModelManager.__getitem__None)r#modelrcCs||j|<dSrr$)rr#r(rrr __setitem__[sz DocumentModelManager.__setitem__boolcCs ||jkSrr$r%rrr __contains__^sz!DocumentModelManager.__contains__zIterator[Model]cCst|jSr)iterrvaluesr!rrr__iter__aszDocumentModelManager.__iter__cCs$|jD] }|q |`|`dS)z6 Clean up references to the Documents models N)rr-destroyr)rmrrrr/ds zDocumentModelManager.destroyzGenerator[(None, None, None)]ccs|dV|dS)a Defer expensive model recompuation until intermediate updates are complete. Making updates to the model graph might trigger events that cause more updates. This context manager can be used to prevent expensive model recompuation from happening until all events have finished and the Document state is quiescent. Example: .. code-block:: python with models.freeze(): # updates that might change the model graph, that might trigger # updates that change the model graph, etc. Recompuation will # happen once at the end. N) _push_freeze _pop_freezer!rrrfreezeqszDocumentModelManager.freezestrz List[Model])namercCs |j|S)z Find all the models for this Document with a given name. Args: name (str) : the name of a model to search for Returns A list of models )rget_allrr5rrrget_all_by_names z$DocumentModelManager.get_all_by_namez Model | NonecCs|j|dS)z Find the model for this Document with a given ID. Args: id (ID) : model ID to search for If no model with the given ID exists, returns None Return: a Model or None N)rgetr%rrr get_by_ids zDocumentModelManager.get_by_idcCs|j|d|dS)a Find a single model for this Document with a given name. If multiple models are found with the name, an error is raised. Args: name (str) : the name of a model to search for Returns A model with the given name, or None z!Found more than one model named '')rZget_oner7rrrget_one_by_names z$DocumentModelManager.get_one_by_namecCs|jdkr|dS)zf Recompute the set of all models, if not currently frozen Returns: None rNr recomputer!rrr invalidates zDocumentModelManager.invalidatec Cs|}|dkrdSt}|jD]}||O}q t|j}||}||}i}t}|D]&} | || j<| jdk r^| | j| q^|D]} |j | j| q|D]} | |q||_||_dS)a Recompute the set of all models based on references reachable from the Document's current roots. This computation can be expensive. Use ``freeze`` to wrap operations that update the model object graph to avoid over-recompuation .. note:: Any models that remove during recomputation will be noted as "previously seen" N)rrrootsZ referencesrr-r r#r5 add_valueraddZ_detach_documentZ_attach_documentr) rrZ new_modelsmrZ old_modelsZ to_detachZ to_attachZ recomputedZrecomputed_by_namemnZmdmarrrr>s,      zDocumentModelManager.recomputecCs ||jkS)z Report whether a model id has ever previously belonged to this Document. Args: id (ID) : the model id of a model to check Returns: bool )rr%rrrseens zDocumentModelManager.seenz str | None)r(old_namenew_namercCs0|dk r|j|||dk r,|j||dS)aD Update the name for a model. .. note:: This function and the internal name mapping exist to support optimizing the common case of name lookup for models. Keeping a dedicated name index is faster than using generic ``bokeh.query`` functions with a name selector Args: model (Model) : a model to update the name for old_name(str, None) : a previous name for the model, or None new_name(str, None) : a new name for the model, or None Returns: None N)rZ remove_valuerA)rr(rGrHrrr update_namesz DocumentModelManager.update_namecCs|jd7_dS)Nr )rr!rrrr1sz!DocumentModelManager._push_freezecCs$|jd8_|jdkr |dS)Nr rr=r!rrrr2s z DocumentModelManager._pop_freezeN)__name__ __module__ __qualname____doc____annotations__rrrr"r&r)r+r.r/ contextlibcontextmanagerr3r8r:r<r?r>rFrIr1r2rrrrr8s0     . r)rM __future__rlogging getLoggerrJlogrOrtypingrrrrrrZ core.typesr r(r Zutil.datatypesr rr__all__rrrrrs