U /et'@sUdZddlmZddlZeeZddlmZm Z m Z m Z m Z m Z ddlmZdZe e ee dfefZd d d d d d Zd dddddZdd ddddZGdddZGdddeZGdddeZGdddeZGdddeZGd d!d!eZGd"d#d#eZGd$d%d%eZGd&d'd'eZed(d)ed*d)ed+d)ed,d)ed-d)ed.d)ed/d)iZd0e d1<dd2dd3d4d5Z!dS)6z} The query module provides functions for searching collections of Bokeh models for instances that match specified criteria. ) annotationsN)AnyCallableDictIterableTypeUnion)Model) EQfindGEQGTINLEQLTNEQORr matchis_single_string_selector _OperatorzIterable[Model] SelectorType)objsselectorreturncsfdd|DS)a Query a collection of Bokeh models and yield any that match the a selector. Args: obj (Model) : object to test selector (JSON-like) : query selector Yields: Model : objects that match the query Queries are specified as selectors similar to MongoDB style query selectors, as described for :func:`~bokeh.core.query.match`. Examples: .. code-block:: python # find all objects with type Grid find(p.references(), {'type': Grid}) # find all objects with type Grid or Axis find(p.references(), {OR: [ {'type': Grid}, {'type': Axis} ]}) # same query, using IN operator find(p.references(), {'type': {IN: [Grid, Axis]}}) c3s|]}t|r|VqdSNr).0objr4/tmp/pip-unpacked-wheel-f5fndrjf/bokeh/core/query.py \s zfind..r )rrr rr!r >sr strbool)rfieldrcCs"t|dko ||ko t||tS)z Whether a selector is a simple single field, e.g. ``{name: "foo"}`` Args: selector (JSON-like) : query selector field (str) : field name to check for Returns bool )len isinstancer#)rr%r r r!r_s rr )rrrc s||D]l\}}t|tr.|dkrvt|trdt|tgkrdtfdd|tDstdSnt|stdSn|dkrt|tr|jkrdSnFzt |t j@sWdSWn&t k r|jkrYdSYnXnJt |sdSt |}t|trt ||s,dSn||krvdSq|tkrLt|svdSq|tkrnt||svdSqtdqdS)a Test whether a given Bokeh model matches a given selector. Args: obj (Model) : object to test selector (JSON-like) : query selector Returns: bool : True if the object matches, False otherwise In general, the selectors have the form: .. code-block:: python { attrname : predicate } Where a predicate is constructed from the operators ``EQ``, ``GT``, etc. and is used to compare against values of model attributes named ``attrname``. For example: .. code-block:: python >>> from bokeh.plotting import figure >>> p = figure(width=400) >>> match(p, {'width': {EQ: 400}}) True >>> match(p, {'width': {GT: 500}}) False There are two selector keys that are handled especially. The first is 'type', which will do an isinstance check: .. code-block:: python >>> from bokeh.plotting import figure >>> from bokeh.models import Axis >>> p = figure() >>> match(p.xaxis[0], {'type': Axis}) True >>> match(p.title, {'type': Axis}) False There is also a ``'tags'`` attribute that ``Model`` objects have, that is a list of user-supplied values. The ``'tags'`` selector key can be used to query against this list of tags. An object matches if any of the tags in the selector match any of the tags on the object: .. code-block:: python >>> from bokeh.plotting import figure >>> p = figure(tags = ["my plot", 10]) >>> match(p, {'tags': "my plot"}) True >>> match(p, {'tags': ["my plot", 10]}) True >>> match(p, {'tags': ["foo"]}) False typec3s|]}t|VqdSr)r()rxrr r!r"szmatch..Ftagszmalformed query selectorT)itemsr(r#dictlistkeysranyr,set TypeErrorhasattrgetattrrr_or _operators ValueError)rrkeyvalattrr r+r!rlsDD               rc@s eZdZdS)rN)__name__ __module__ __qualname__r r r r!rsc@seZdZdZdS)raY Form disjunctions from other query predicates. Construct an ``OR`` expression by making a dict with ``OR`` as the key, and a list of other query expressions as the value: .. code-block:: python # matches any Axis subclasses or models with .name == "mycircle" { OR: [dict(type=Axis), dict(name="mycircle")] } Nr<r=r>__doc__r r r r!rs rc@seZdZdZdS)raO Predicate to test if property values are in some collection. Construct and ``IN`` predicate as a dict with ``IN`` as the key, and a list of values to check against. .. code-block:: python # matches any models with .name in ['a', 'mycircle', 'myline'] dict(name={ IN: ['a', 'mycircle', 'myline'] }) Nr?r r r r!rs rc@seZdZdZdS)ra Predicate to test if property values are greater than some value. Construct and ``GT`` predicate as a dict with ``GT`` as the key, and a value to compare against. .. code-block:: python # matches any models with .size > 10 dict(size={ GT: 10 }) Nr?r r r r!rs rc@seZdZdZdS)ra Predicate to test if property values are less than some value. Construct and ``LT`` predicate as a dict with ``LT`` as the key, and a value to compare against. .. code-block:: python # matches any models with .size < 10 dict(size={ LT: 10 }) Nr?r r r r!rs rc@seZdZdZdS)r a Predicate to test if property values are equal to some value. Construct and ``EQ`` predicate as a dict with ``EQ`` as the key, and a value to compare against. .. code-block:: python # matches any models with .size == 10 dict(size={ EQ: 10 }) Nr?r r r r!r s r c@seZdZdZdS)r a. Predicate to test if property values are greater than or equal to some value. Construct and ``GEQ`` predicate as a dict with ``GEQ`` as the key, and a value to compare against. .. code-block:: python # matches any models with .size >= 10 dict(size={ GEQ: 10 }) Nr?r r r r!r -s r c@seZdZdZdS)ra+ Predicate to test if property values are less than or equal to some value. Construct and ``LEQ`` predicate as a dict with ``LEQ`` as the key, and a value to compare against. .. code-block:: python # matches any models with .size <= 10 dict(size={ LEQ: 10 }) Nr?r r r r!r<s rc@seZdZdZdS)ra Predicate to test if property values are unequal to some value. Construct and ``NEQ`` predicate as a dict with ``NEQ`` as the key, and a value to compare against. .. code-block:: python # matches any models with .size != 10 dict(size={ NEQ: 10 }) Nr?r r r r!rKs rcCs||kSrr r*yr r r!_rCcCs||kSrr rAr r r!rC`rDcCs||kSrr rAr r r!rCarDcCs||kSrr rAr r r!rCbrDcCs||kSrr rAr r r!rCcrDcCs||kSrr rAr r r!rCdrDcCs||kSrr rAr r r!rCerDz2Dict[Type['_Operator'], Callable[[Any, Any], Any]]r7zIterable[SelectorType])r selectorsrcstfdd|DS)Nc3s|]}t|VqdSrr)rrr+r r!r"jsz_or..)r1)rrEr r+r!r6isr6)"r@ __future__rlogging getLoggerr<logtypingrrrrrrmodelr __all__r#rr rrrrrrrr r rrr7__annotations__r6r r r r!sF   ! x