U /e`$@sdZddlmZddlZeeZddlZddlZ ddl Z ddl Z ddl m Z mZddlZe rtddlmZddlZnddlmZedZedZdd lmZdd lmZmZmZmZmZmZd Z dd d dd ddddZ!Gddde j"Z#dS)a^ Provide a functions and classes to implement a custom JSON encoder for serializing objects for BokehJS. The primary interface is provided by the |serialize_json| function, which uses the custom |BokehJSONEncoder| to produce JSON output. In general, functions in this module convert values in the following way: * Datetime values (Python, Pandas, NumPy) are converted to floating point milliseconds since epoch. * TimeDelta values are converted to absolute floating point milliseconds. * RelativeDelta values are converted to dictionaries. * Decimal values are converted to floating point. * Sequences (Pandas Series, NumPy arrays, python sequences) that are passed though this interface are converted to lists. Note, however, that arrays in data sources inside Bokeh Documents are converted elsewhere, and by default use a binary encoded format. * Bokeh ``Model`` instances are usually serialized elsewhere in the context of an entire Bokeh Document. Models passed trough this interface are converted to references. * ``HasProps`` (that are not Bokeh models) are converted to key/value dicts or all their properties and values. * ``Color`` instances are converted to CSS color values. .. |serialize_json| replace:: :class:`~bokeh.core.json_encoder.serialize_json` .. |BokehJSONEncoder| replace:: :class:`~bokeh.core.json_encoder.BokehJSONEncoder` ) annotationsN) TYPE_CHECKINGAny)import_optionalzdateutil.relativedeltapandas)settings)convert_datetime_typeconvert_timedelta_typeis_datetime_typeis_timedelta_typetransform_arraytransform_series)BokehJSONEncoderserialize_jsonrz bool | Nonez int | Nonestr)objprettyindentkwargsreturncKshdD]}||krtd|dqt|}|r6d}nd}|rJ|dkrJd}tj|ftd||d d |S) a Return a serialized JSON representation of objects, suitable to send to BokehJS. This function is typically used to serialize single python objects in the manner expected by BokehJS. In particular, many datetime values are automatically normalized to an expected format. Some Bokeh objects can also be passed, but note that Bokeh models are typically properly serialized in the context of an entire Bokeh document. The resulting JSON always has sorted keys. By default. the output is as compact as possible unless pretty output or indentation is requested. Args: obj (obj) : the object to serialize to JSON format pretty (bool, optional) : Whether to generate prettified output. If ``True``, spaces are added after added after separators, and indentation and newlines are applied. (default: False) Pretty output can also be enabled with the environment variable ``BOKEH_PRETTY``, which overrides this argument, if set. indent (int or None, optional) : Amount of indentation to use in generated JSON output. If ``None`` then no indentation is used, unless pretty output is enabled, in which case two spaces are used. (default: None) Any additional keyword arguments are passed to ``json.dumps``, except for some that are computed internally, and cannot be overridden: * allow_nan * indent * separators * sort_keys Examples: .. code-block:: python >>> data = dict(b=np.datetime64('2017-01-01'), a = np.arange(3)) >>>print(serialize_json(data)) {"a":[0,1,2],"b":1483228800000.0} >>> print(serialize_json(data, pretty=True)) { "a": [ 0, 1, 2 ], "b": 1483228800000.0 } ) allow_nan separators sort_keysz The value of z7 is computed internally, overriding is not permissible.),z: )r:NrFT)clsrrrr) ValueErrorrrjsondumpsr)rrrrnamerr!;/tmp/pip-unpacked-wheel-f5fndrjf/bokeh/core/json_encoder.pyras<  rcs8eZdZdZdddfdd ZdddddZZS)rzo A custom ``json.JSONEncoder`` subclass for encoding objects in accordance with the BokehJS protocol. r)rrc st|rt|St|r t|St|tjr4|St|trRt |j |j |j dSt t|t jrlt|St t|t jrt|St t|t jrt|St|tjrt|Strt|tjrt |j|j|j|j|j|j|j dSt!"|SdS)a* Handle special scalars such as (Python, NumPy, or Pandas) datetimes, or Decimal values. Args: obj (obj) : The object to encode. Anything not specifically handled in this method is passed on to the default system JSON encoder. )startstopstep)yearsmonthsdayshoursminutesseconds microsecondsN)#r r r r isinstancedtdate isoformatslicedictr#r$r%npZ issubdtypetypeZfloatingfloatintegerintZbool_booldecimalDecimalrd relativedeltar&r'r(r)r*r+r,superdefault)selfr __class__r!r"transform_python_typess4     z'BokehJSONEncoder.transform_python_typescsddlm}ddlm}ddlm}trFt|tjtj frFt |ddSt|t j r^t |ddSt|tjr|fdd |DSt||r|jSt||r|jd d St||r|S|Sd S) a  The required ``default`` method for ``JSONEncoder`` subclasses. Args: obj (obj) : The object to encode. Anything not specifically handled in this method is passed on to the default system JSON encoder. r)Color)Model)HasPropsT)Z force_listcsg|]}|qSr!)r>).0itemr?r!r" sz,BokehJSONEncoder.default..F)Zinclude_defaultsN)colorsrCmodelrDZ has_propsrFpdr-ZSeriesZIndexrr3Zndarrayr collectionsdequerefZproperties_with_valuesZto_cssrB)r?rrCrDrFr!rIr"r>s           zBokehJSONEncoder.default)__name__ __module__ __qualname____doc__rBr> __classcell__r!r!r@r"rs4r)NN)$rT __future__rlogging getLoggerrQlogrNdatetimer.r9rtypingrrZnumpyr3Zdateutil.relativedeltar<r;rrMZutil.dependenciesrrZutil.serializationr r r r r r__all__r JSONEncoderrr!r!r!r"s('        Q