o i@sddlmZmZmZddlmZddlZddlmZddl m Z ddl m Z m Z mZmZGdd d eZ dd edededededef ddZd eeefdefddZde defddZ dde d edededededdfddZdedefddZdS) )UnionIOAny)StringION)YAML)RepresenterError) force_pathFilePath YAMLInput YAMLOutputc@s eZdZdddZd ddZdS) CustomYamlsafeTcCs&tj|||dd|_d|_d|_dS)N)typpureFTzutf-8)r__init__default_flow_style allow_unicodeencoding)selfrrrC/home/ubuntu/.local/lib/python3.10/site-packages/srsly/_yaml_api.pyr s zCustomYaml.__init__NcKs<d}|dur d}t}tj|||fi||r|SdS)NFT)rrdumpgetvalue)rdatastreamkw inefficientrrrrszCustomYaml.dump)rT)N)__name__ __module__ __qualname__rrrrrrr s r Frindent_mappingindent_sequence indent_offset sort_keysreturncCs&t}||_|j|||d||S)aSerialize an object to a YAML string. See the ruamel.yaml docs on indentation for more details on the expected format. https://yaml.readthedocs.io/en/latest/detail.html?highlight=indentation#indentation-of-block-sequences data: The YAML-serializable data. indent_mapping (int): Mapping indentation. indent_sequence (int): Sequence indentation. indent_offset (int): Indentation offset. sort_keys (bool): Sort dictionary keys. RETURNS (str): The serialized string. )mappingsequenceoffset)r sort_base_mapping_type_on_outputindentr)rr#r$r%r&yamlrrr yaml_dumpss r.c Cs:t}z||WSty}ztd|d}~ww)zDeserialize unicode or a file object a Python object. data (str / file): The data to deserialize. RETURNS: The deserialized Python object. zInvalid YAML: N)r load Exception ValueError)rr-errr yaml_loads4s r3pathcCs\|dkr tj}t|St|}|jddd }t|WdS1s'wYdS)zLoad YAML from file or standard input. location (FilePath): The file path. "-" for reading from stdin. RETURNS (YAMLOutput): The loaded content. -rutf8rN)sysstdinreadr3r open)r4r file_pathfrrr read_yamlAs $r?c Cspt|||||d}|dkrt|dSt|dd}|jddd}||WddS1s1wYdS) aCreate a .json file and dump contents or write to standard output. location (FilePath): The file path. "-" for writing to stdout. data (YAMLInput): The JSON-serializable data to output. indent_mapping (int): Mapping indentation. indent_sequence (int): Sequence indentation. indent_offset (int): Indentation offset. sort_keys (bool): Sort dictionary keys. )r#r$r%r&r5F)require_existswr7r8N)r.printr r<write) r4rr#r$r%r& yaml_datar=r>rrr write_yamlOs   "rEobjcCs$zt|WdStyYdSw)zCheck if a Python object is YAML-serializable (strict). obj: The object to check. RETURNS (bool): Whether the object is YAML-serializable. TF)r.r)rFrrris_yaml_serializableps  rG)r!r"r!F)typingrrriorr9 ruamel_yamlrruamel_yaml.representerrutilr r r r r intboolstrr.r3r?rErGrrrrs\      !