cFddlmZddlZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z GddeZeZd Zdd Zdd ZddZddZddZddZdZdS))absolute_importN)Path)validate_coerce_fig_to_dictvalidate_coerce_output_type) get_module)ImageUriValidatorcdeZdZdZdZedZejdZedZ dS) JsonConfig)jsonorjsonautocd|_dS)Nr _default_engineselfs /lib/python3.11/site-packages/plotly/io/_json.py__init__zJsonConfig.__init__s%c|jSNrrs rdefault_enginezJsonConfig.default_engines ##rc|tjvr.tdtj||dkr|||_dS)Nz9Supported JSON engines include {valid} Received {val})validvalr )r _valid_engines ValueErrorformatvalidate_orjsonr)rrs rrzJsonConfig.default_enginesl j/ / %%+V*2KQTV%U%U  (? #  " " ""rcFtd}|tddS)Nr z-The orjson engine requires the orjson package)rr)clsr s rrzJsonConfig.validate_orjson&s3H%%  NLMM M N NrN) __name__ __module__ __qualname__rrpropertyrsetter classmethodrrrr r s/N&&&$$X$ # # #NN[NNNrr c|dvrdS|S)zM This is used to ultimately *encode* into strict JSON, see `encode` )Infinityz -InfinityNaNNr()consts rcoerce_to_strictr-0s  00t rFc tdd}| tj}|dkr|d}nd}n|dvrtd|ztd d td d td d td d d}|dkr)i}|rd|d<nd|d<ddlm}t j|fd|i|S|dkrt |j |j z}|r ||j z} | }n#t$rYnwxYw |||dS#t $rYnwxYwt#|dd|}|||dSdS)a Convert a plotly/Dash object to a JSON string representation Parameters ---------- plotly_object: A plotly/Dash object represented as a dict, graph_object, or Dash component pretty: bool (default False) True if JSON representation should be pretty-printed, False if representation should be as compact as possible. engine: str (default None) The JSON encoding engine to use. One of: - "json" for an engine based on the built-in Python json module - "orjson" for a faster engine that requires the orjson package - "auto" for the "orjson" engine if available, otherwise "json" If not specified, the default engine is set to the current value of plotly.io.json.config.default_engine. Returns ------- str Representation of input object as a JSON string See Also -------- to_json : Convert a plotly Figure to JSON with validation r T should_loadNr r r r Invalid json engine: %szsage.allFnumpypandasz PIL.Image)sage_allnppdimageindent),: separatorsr)PlotlyJSONEncoderr!)optionutf8) numpy_alloweddatetime_allowedmodules)rconfigrr_plotly_utils.utilsr>r dumpsr rOPT_NON_STR_KEYSOPT_SERIALIZE_NUMPY OPT_INDENT_2to_plotly_jsonAttributeErrordecode TypeErrorclean_to_json_compatible) plotly_objectprettyenginer rCoptsr>cleaneds rto_json_plotlyrT<s9<d 3 3 3F'& =  FFFF ) )=2V;<<<zu===e444u555KU;;; G$A  ,DNN",D 999999z-GG->G$GGG 8 A""$$$&)CC  ( F' 'D )88::MM    D  << d<;;BB6JJ J    D + !     ||GD|1188@@@3AAs$1D DD)E E ETct||}|r/|dgD]}|ddt|||S)a Convert a figure to a JSON string representation Parameters ---------- fig: Figure object or dict representing a figure validate: bool (default True) True if the figure should be validated before being converted to JSON, False otherwise. pretty: bool (default False) True if JSON representation should be pretty-printed, False if representation should be as compact as possible. remove_uids: bool (default True) True if trace UIDs should be omitted from the JSON representation engine: str (default None) The JSON encoding engine to use. One of: - "json" for an engine based on the built-in Python json module - "orjson" for a faster engine that requires the orjson package - "auto" for the "orjson" engine if available, otherwise "json" If not specified, the default engine is set to the current value of plotly.io.json.config.default_engine. Returns ------- str Representation of figure as a JSON string See Also -------- to_json_plotly : Convert an arbitrary plotly graph_object or Dash component to JSON datauidN)rPrQ)rgetpoprT)figvalidaterP remove_uidsrQfig_dicttraces rto_jsonr_sgN+399H#\\&"-- # #E IIeT " " " " (6& A A AArcpt|||||}t|trt|}nt|tr|}nd}|K ||dS#t $rYnwxYwt d|||dS)a+ Convert a figure to JSON and write it to a file or writeable object Parameters ---------- fig: Figure object or dict representing a figure file: str or writeable A string representing a local file path or a writeable object (e.g. a pathlib.Path object or an open file descriptor) pretty: bool (default False) True if JSON representation should be pretty-printed, False if representation should be as compact as possible. remove_uids: bool (default True) True if trace UIDs should be omitted from the JSON representation engine: str (default None) The JSON encoding engine to use. One of: - "json" for an engine based on the built-in Python json module - "orjson" for a faster engine that requires the orjson package - "auto" for the "orjson" engine if available, otherwise "json" If not specified, the default engine is set to the current value of plotly.io.json.config.default_engine. Returns ------- None )r[rPr\rQNzX The 'file' argument '{file}' is not a string, pathlib.Path object, or file descriptor. )file) r_ isinstancestrrwriterKrr write_text)rZrar[rPr\rQjson_strpaths r write_jsonrhsH hv;vH $ Dzz D$   "  JJx F    D   F     !!!!!sA.. A;:A;c tdd}t|ttfs1t dt ||| tj}|dkr|d}nd}n|d vrt d |z|dkr/t | |}n>t|tr| d }tj |}|S) a Parse JSON string using the specified JSON engine Parameters ---------- value: str or bytes A JSON string or bytes object engine: str (default None) The JSON decoding engine to use. One of: - if "json", parse JSON using built in json module - if "orjson", parse using the faster orjson module, requires the orjson package - if "auto" use orjson module if available, otherwise use the json module If not specified, the default engine is set to the current value of plotly.io.json.config.default_engine. Returns ------- dict See Also -------- from_json_plotly : Parse JSON with plotly conventions into a dict r Tr/zr from_json_plotly requires a string or bytes argument but received value of type {typ} Received value: {value})typvalueNr r r1r2zutf-8)rrbrcbytesrrtyperDrr rloadsrLr )rkrQr value_dicts rfrom_json_plotlyrps'6d 3 3 3F ec5\ * *  %vKKu &     '& =  FFFF ) )=2V;<<< '""$$$\\%((  eU # # *LL))EZ&& rFigurec`t||}t|}|||}|S)a Construct a figure from a JSON string Parameters ---------- value: str or bytes String or bytes object containing the JSON representation of a figure output_type: type or str (default 'Figure') The output figure type or type name. One of: graph_objs.Figure, 'Figure', graph_objs.FigureWidget, 'FigureWidget' skip_invalid: bool (default False) False if invalid figure properties should result in an exception. True if invalid figure properties should be silently ignored. engine: str (default None) The JSON decoding engine to use. One of: - if "json", parse JSON using built in json module - if "orjson", parse using the faster orjson module, requires the orjson package - if "auto" use orjson module if available, otherwise use the json module If not specified, the default engine is set to the current value of plotly.io.json.config.default_engine. Raises ------ ValueError if value is not a string, or if skip_invalid=False and value contains invalid figure properties Returns ------- Figure or FigureWidget )rQ) skip_invalid)rpr)rk output_typersrQr]r!rZs r from_jsonruWsAP f555H &k 2 2C #h\ 2 2 2C Jrc&t|t}t|trt|}nt|tr|}nd}||}n|}t ||||S)a Construct a figure from the JSON contents of a local file or readable Python object Parameters ---------- file: str or readable A string containing the path to a local file or a read-able Python object (e.g. a pathlib.Path object or an open file descriptor) output_type: type or str (default 'Figure') The output figure type or type name. One of: graph_objs.Figure, 'Figure', graph_objs.FigureWidget, 'FigureWidget' skip_invalid: bool (default False) False if invalid figure properties should result in an exception. True if invalid figure properties should be silently ignored. engine: str (default None) The JSON decoding engine to use. One of: - if "json", parse JSON using built in json module - if "orjson", parse using the faster orjson module, requires the orjson package - if "auto" use orjson module if available, otherwise use the json module If not specified, the default engine is set to the current value of plotly.io.json.config.default_engine. Returns ------- Figure or FigureWidget N)rsrtrQ)rbrcr read_textreadru)rartrsrQ file_is_strrgrfs r read_jsonrzsJT3''K$ Dzz D$   >>##99;; |V   rc D t|tttfr|St|tr fd|DSt|t tfr|rfd|DSdd}dd}di}|d}|d}|d }|d }|0||j vrt|S||j vrt|S|||j j j urtd St||jr|r#|jjd vr||S|jjd kr'||S|jjdkr|S|jjdkr|}n$t||jrt|S|||jurdSt||j|jfr|r(|jjd vr||jS|jjd krt||jr,|j} n&|} |sf$rYnwxYw|s, |S#t<t>f$rYn wxYwt|t@j r|S |S#t>$rYnwxYwt|tBj"rt|S|)t||j#rtIj%|S |&}n#t>$rYnwxYwt|tr fd|DSt|t tfr|rfd|DS|S)Nc2i|]\}}|t|fiSr(rN.0kvkwargss r z,clean_to_json_compatible..0QQQTQ+A8888QQQrc*g|]}t|fiSr(r}rrrs r z,clean_to_json_compatible..*GGGa,Q99&99GGGrrAFrBrCr5r6r7r8nan)biufMUOc2i|]\}}|t|fiSr(r}r~s rrz,clean_to_json_compatible..8rrc*g|]}t|fiSr(r}rs rrz,clean_to_json_compatible..<rr)'rbintfloatrcdictitemslisttuplerXRRZZmacoremaskedndarraydtypekindascontiguousarraydatetime_as_stringtolist datetime64NaTSeries DatetimeIndexvaluesdt to_pydatetimerangelen isoformatrMrKdatetimedecimalDecimalImagerpil_image_to_urirJ) objrrArBrCr5r6r7r8 dt_valuesrs ` rrNrNs #UC()) #tHQQQQSYY[[QQQQ C$ ' 'H  HGGGG3GGG GJJ66Mzz"4e<<jjB''Gz"H B B G E (+  ::  HK  s88O  "%*# # <<  RZ ( (  #3G!G #++C0003& #,,S1188:::3& #zz||#3& #jjll R] + + s88O ! "&= !4 bi)9: ; ; ! !3G!G !++CJ7773& !c29--= # 4 4 6 6 = = ? ?II # 1 1 3 3 : : < >22@@'0|'='='?'? !    !! ~ &       ==?? ">*    D  C* + +  zz||      #w''Szz 7ZU[997 1#666   ""      #tHQQQQSYY[[QQQQ C$ ' 'H  HGGGG3GGG G JsH*L??MMM--NN!N55 OOP// P<;P<)FN)TFTNr)rqFN) __future__rr rrpathlibrplotly.io._utilsrr_plotly_utils.optional_importsr_plotly_utils.basevalidatorsrobjectr rDr-rTr_rhrprurzrNr(rrrsz&&&&&& UUUUUUUU555555:::::: NNNNNNNN<    YAYAYAYAx/B/B/B/BdI"I"I"I"X>>>>B1111h<<<<~tttttr