o :6fI@sddlZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z Gddde ZeZdd Zd Zed Zd d ZdddZd ddZd ddZd!ddZd"ddZd"ddZddZdS)#N)Path)validate_coerce_fig_to_dictvalidate_coerce_output_type) get_module)ImageUriValidatorc@s>eZdZdZddZeddZejddZeddZ d S) JsonConfig)jsonorjsonautocCs d|_dS)Nr _default_engineselfr/lib/python3.10/site-packages/plotly/io/_json.py__init__s zJsonConfig.__init__cCs|jSNr r rrrdefault_engineszJsonConfig.default_enginecCs8|tjvrtdjtj|d|dkr|||_dS)Nz9Supported JSON engines include {valid} Received {val})Zvalidvalr )r_valid_engines ValueErrorformatvalidate_orjsonr )rrrrrrs  cCstd}|dur tddS)Nr z-The orjson engine requires the orjson package)rr)clsr rrrr%szJsonConfig.validate_orjsonN) __name__ __module__ __qualname__rrpropertyrsetter classmethodrrrrrrs   rcCs|dvrdS|S)zM This is used to ultimately *encode* into strict JSON, see `encode` )ZInfinityz -InfinityZNaNNr)constrrrcoerce_to_strict/sr!))z\u003e)/z\u002f))u
z\u2028)u
z\u2029cCs*|}|D]\}}||vr|||}q|Sr)replace)json_strZ_swapoutZ unsafe_charZ safe_charrrr_safeFs   r(FcCsjtddd}|dur tj}|dkr|durd}n d}n |dvr%td|td d dtd d dtd d dtd d dd}|dkrai}|rId|d<nd|d<ddlm}ttj|fd|i|t S|dkrt |j |j B}|rv||jO}z|}Wn tyYnwzt|j||ddtWStyYnwt|dd|d}t|j||ddtSdS)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 TZ should_loadNr rr rInvalid json engine: %szsage.allFZnumpyZpandasz PIL.Image)sage_allnppdimageindent),:Z separatorsr)PlotlyJSONEncoderr)optionutf8) numpy_alloweddatetime_allowedmodules)rconfigrrZ_plotly_utils.utilsr4r(rdumps _swap_jsonrrZOPT_NON_STR_KEYSZOPT_SERIALIZE_NUMPYZ OPT_INDENT_2to_plotly_jsonAttributeErrordecode _swap_orjson TypeErrorclean_to_json_compatible)Z plotly_objectprettyenginer r9Zoptsr4Zcleanedrrrto_json_plotlyNs`              rETcCs:t||}|r|dgD]}|ddq t|||dS)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)rCrD)rgetpoprE)figvalidaterC remove_uidsrDfig_dictZtracerrrto_jsons 'rNcCst|||||d}t|trt|}n t|tr|}nd}|dur;z||WdSty2Ynwtdj|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 )rKrCrLrDNzX The 'file' argument '{file}' is not a string, pathlib.Path object, or file descriptor. )file) rN isinstancestrrwriter>rr write_text)rJrOrKrCrLrDr&pathrrr write_jsons,$       rUcCstddd}t|ttfstdjt||d|durtj}|dkr-|dur*d}n d}n |d vr7td ||dkrFt | |}|St|trP| d }t |}|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 rr*r+zutf-8)rrPrQbytesrrtyper:rrrloadsr?r)rWrDr Z value_dictrrrfrom_json_plotly,s0      r[FigurecCs$t||d}t|}|||d}|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 )rD) skip_invalid)r[r)rW output_typer]rDrMrrJrrr from_jsonms ( r_cCs\t|t}t|trt|}n t|tr|}nd}|dur"|}n|}t||||dS)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)r]r^rD)rPrQr read_textreadr_)rOr^r]rDZ file_is_strrTr&rrr read_jsons %    rbc sxt|tttfr |St|trfdd|DSt|ttfr,|r,fdd|DSdd}dd}di}|d }|d }|d }|d }|durd||j vr[t|S||j vrdt|S|dur||j j j urstd St||jr|r|jjdvr||S|jjdkr||S|jjdkr|S|jjdkr|}n t||jrt|S|dur'||jurdSt||j|jfr'|r|jjdvr||jS|jjdkr't||jr ttdt||j} Wdn 1swYn|} |s%t t!| D] } | | "| | <q| Sz|}Wn t#t$fy9Ynw|sOz|"WSt#t$fyNYn wt|t%j%rX|Sz|WSt$ygYnwt|t&j'rst|S|durt||j(rt)*|Sz|+}Wn t$yYnwt|trfdd|DSt|ttfr|rfdd|DS|S)Nc"i|] \}}|t|fiqSrrB.0kvkwargsrr "z,clean_to_json_compatible..cg|] }t|fiqSrrdrfrhrirr z,clean_to_json_compatible..r7Fr8r9r,r-r.r/nan)biufMUOignorecrcrrdrerirrrkRrlcrmrrdrnrirrroVrp),rPintfloatrQdictitemslisttuplerHZRRZZZZmacoreZmaskedZndarrayZdtypekindZascontiguousarrayZdatetime_as_stringtolistZ datetime64ZNaTZSeriesZ DatetimeIndexvalueswarningscatch_warnings simplefilter FutureWarningZarrayZdtZ to_pydatetimerangelenZ isoformatrAr>datetimedecimalZDecimalZImagerZpil_image_to_urir=) objrjr7r8r9r,r-r.r/Z dt_valuesrsrrirrBs                        rB)FN)TFTNr)r\FN)rrrrpathlibrZplotly.io._utilsrrZ_plotly_utils.optional_importsrZ_plotly_utils.basevalidatorsrobjectrr:r!r<r@r(rErNrUr[r_rbrBrrrrs*      ` 2 L A 4 ?