U TZfI@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_engineselfr3/tmp/pip-unpacked-wheel-5ksk5baj/plotly/io/_json.py__init__szJsonConfig.__init__cCs|jSNr r rrrdefault_engineszJsonConfig.default_enginecCs8|tjkrtdjtj|d|dkr.|||_dS)Nz9Supported JSON engines include {valid} Received {val})Zvalidvalr )r_valid_engines ValueErrorformatvalidate_orjsonr )rrrrrrs cCstd}|dkrtddS)Nr z-The orjson engine requires the orjson package)rr)clsr rrrr%szJsonConfig.validate_orjsonN) __name__ __module__ __qualname__rrpropertyrsetter classmethodrrrrrrs  rcCs|dkr dS|SdS)zM This is used to ultimately *encode* into strict JSON, see `encode` )Infinityz -InfinityNaNNr)constrrrcoerce_to_strict/sr#))z\u003e)/z\u002f))u
z\u2028)u
z\u2029cCs*|}|D]\}}||kr|||}q|Sr)replace)json_strZ_swapoutZ unsafe_charZ safe_charrrr_safeFs  r*FcCsttddd}|dkrtj}|dkr6|dk r0d}qJd}n|dkrJtd|td d dtd d dtd d dtd d dd}|dkri}|rd|d<nd|d<ddlm}ttj|fd|i|t S|dkrpt |j |j B}|r||jO}z |}Wntk rYnXzt|j||ddtWStk rDYnXt|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),: separatorsr)PlotlyJSONEncoderr)optionutf8) numpy_alloweddatetime_allowedmodules)rconfigrrZ_plotly_utils.utilsr7r*rdumps _swap_jsonrrZOPT_NON_STR_KEYSZOPT_SERIALIZE_NUMPYZ OPT_INDENT_2to_plotly_jsonAttributeErrordecode _swap_orjson TypeErrorclean_to_json_compatible)Z plotly_objectprettyenginer r<optsr7cleanedrrrto_json_plotlyNs^             rJTcCs:t||}|r,|dgD]}|ddqt|||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)rFrG)rgetpoprJ)figvalidaterF remove_uidsrGfig_dicttracerrrto_jsons ' rTcCst|||||d}t|tr&t|}nt|tr6|}nd}|dkrzz||WdStk rfYnXtdj|dn ||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 )rPrFrQrGNzX The 'file' argument '{file}' is not a string, pathlib.Path object, or file descriptor. )file) rT isinstancestrrwriterArr write_text)rOrUrPrFrQrGr(pathrrr write_jsons0$     r[cCstddd}t|ttfs0tdjt||d|dkr>tj}|dkrZ|dk rTd}qnd}n|d krntd ||dkrt | |}nt|tr| 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)rrVrWbytesrrtyper=rrrloadsrBr)r]rGr Z value_dictrrrfrom_json_plotly,s.       raFigurecCs$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 )rG) skip_invalid)rar)r] output_typercrGrRrrOrrr from_jsonms(  recCs\t|t}t|trt|}nt|tr.|}nd}|dk rD|}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)rcrdrG)rVrWr read_textreadre)rUrdrcrGZ file_is_strrZr(rrr read_jsons%     rhc st|tttfr|St|tr4fdd|DSt|ttfrX|rXfdd|DSdd}dd}di}|d }|d }|d }|d }|dk r||j krt|S||j krt|S|dk rt||j j j krtd St||jr^|r|jjdkr||S|jjdkr0||S|jjdkrF|S|jjdkrt|}nt||jrtt|S|dk rN||jkrdSt||j|jfrN|r|jjdkr||jS|jjdkrNt||jrt&tdt||j} W5QRXn |} |sJt t!| D]} | | "| | <q2| Sz |}Wnt#t$fk rtYnX|sz |"WSt#t$fk rYnXnt|t%j%r|Sz |WSt$k rYnXt|t&j'rt|S|dk rt||j(rt)*|Sz |+}Wnt$k r2YnXt|trVfdd|DSt|ttfr~|r~fdd|DS|S)Ncsi|]\}}|t|fqSrrE.0kvkwargsrr sz,clean_to_json_compatible..csg|]}t|fqSrrirkrmrnrr sz,clean_to_json_compatible..r:Fr;r<r.r/r0r1nan)biufMUOignorecsi|]\}}|t|fqSrrirjrnrrrpRscsg|]}t|fqSrrirqrnrrrrVs),rVintfloatrWdictitemslisttuplerMZRRZZZmacoreZmaskedZndarrayZdtypekindZascontiguousarrayZdatetime_as_stringtolistZ datetime64ZNaTZSeriesZ DatetimeIndexvalueswarningscatch_warnings simplefilter FutureWarningarraydtZ to_pydatetimerangelen isoformatrDrAdatetimedecimalDecimalZImagerZpil_image_to_urir@) objror:r;r<r.r/r0r1Z dt_valuesrurrnrrEs                      rE)FN)TFTN)TFTN)N)rbFN)rbFN)rrrrpathlibrZplotly.io._utilsrrZ_plotly_utils.optional_importsrZ_plotly_utils.basevalidatorsrobjectrr=r#r?rCr*rJrTr[rarerhrErrrrs(     ` 2 L A 4 ?