o ߥi@sjddlmZmZddlmZddlmZddlmZm Z ee e dZ ddd Z dd d Z d d Z dS))BytesIOStringIO)Path)File) JsonHandler YamlHandler)jsonyamlymlNcKst|tr t|}|durt|tr|dd}|tvr$td|t|}t|trv|jrStt |}|j |fi|}Wd|S1sLwY|St t |}|j |fi|}Wd|S1sowY|St |dr|j |fi|}|Std)aLoad data from json/yaml/pickle files. This method provides a unified api for loading data from serialized files. Args: file (str or :obj:`Path` or file-like object): Filename or a file-like object. file_format (str, optional): If not specified, the file format will be inferred from the file extension, otherwise use the specified one. Currently supported formats include "json", "yaml/yml". Examples: >>> load('/path/of/your/file') # file is stored in disk >>> load('https://path/of/your/file') # file is stored on internet >>> load('oss://path/of/your/file') # file is stored in petrel Returns: The content from the file. N.Unsupported format: readz."file" must be a filepath str or a file-object) isinstancerstrsplitformat_handlers TypeError text_moderr read_textloadrrhasattr)file file_formatkwargshandlerfobjrH/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/fileio/io.pyrs2       rcKsVt|tr t|}|dur"t|tr|dd}n|dur"td|tvr-td|t|}|dur>|j|fi|St|tr|jrnt }|j ||fi|t | |WddS1sgwYdSt}|j ||fi|t | |WddS1swYdSt|dr|j ||fi|dStd)aDump data to json/yaml strings or files. This method provides a unified api for dumping data as strings or to files. Args: obj (any): The python object to be dumped. file (str or :obj:`Path` or file-like object, optional): If not specified, then the object is dumped to a str, otherwise to a file specified by the filename or file-like object. file_format (str, optional): Same as :func:`load`. Examples: >>> dump('hello world', '/path/of/your/file') # disk >>> dump('hello world', 'oss://path/of/your/file') # oss Returns: bool: True for success, False otherwise. Nr r z0file_format must be specified since file is Nonerwritez."file" must be a filename str or a file-object)rrrr ValueErrorrr dump_to_strrrdumpr write_textgetvaluerr!r)rrrrrrrrr r$;s6   "" r$cKs0|tvr td|t|}|j|fi|S)aDump data to json/yaml strings or files. This method provides a unified api for dumping data as strings or to files. Args: obj (any): The python object to be dumped. format (str, optional): Same as file_format :func:`load`. Examples: >>> dumps('hello world', 'json') # json >>> dumps('hello world', 'yaml') # yaml Returns: bool: True for success, False otherwise. r)rrdumps)rformatrrrrr r'ksr')N)NN)iorrpathlibrrrr(rrrrr$r'rrrr s    * 0