B }d@sdZddlZddlZddlZddlmZddlmZdZddl m Z e ej ej fZ dZdd Zd d Zdd dZddZddZdS)z#Utilities for unified input/output N)wraps)urlparsez(Duncan Macleod )IOBasescsfdd}|S)aFactory function to create I/O identifiers for a set of extensions The returned function is designed for use in the unified I/O registry via the `astropy.io.registry.register_identifier` hool. Parameters ---------- extensions : `str` one or more file extension strings Returns ------- identifier : `callable` an identifier function that tests whether an incoming file path carries any of the given file extensions (using `str.endswith`) cst|to|S)z.identifyr)r rr)r ridentify_factory(s rcOsjt|dr tj|f||St|f||}|d}|d|tkrf|tj|f||S|S)aROpen a file handling optional gzipping If ``name`` ends with ``'.gz'``, or if the GZIP file signature is found at the beginning of the file, the file will be opened with `gzip.open`, otherwise a regular file will be returned from `open`. Parameters ---------- name : `str`, `pathlib.Path` path or name of file to open. *args, **kwargs other arguments to pass to either `open` for regular files, or `gzip.open` for gzipped files. Returns ------- file : `io.TextIoBase`, `file`, `gzip.GzipFile` the open file object z.gzr)rrgzipopenreadseekGZIP_SIGNATUREclose)namer r fobjsigrrrgopenDs  rrcsfdd}|r||S|S)aDecorate a function to ensure the chosen argument is an open file Parameters ---------- func : `callable` the function to decorate mode : `str`, optional the mode with which to open files pos : `int`, optional which argument to look at Examples -------- To ensure that the first argument is an open read-only file, just use the decorator without functional parentheses or arguments: >>> @with_open >>> def my_func(pathorfile, *args, **kwargs) >>> ... To ensure that the second argument (position 1) is a file open for writing: >>> @with_open(mode="w", pos=1) >>> def my_func(stuff, pathorfile, *args, **kwargs) >>> stuff.write_to(pathorfile, *args, **kwargs) cstfdd}|S)Nc sJt|ts@t|d}t|}||<||SQRX||S)N)mode)r FILE_LIKErlist)r r r)funcrposrr wrapped_funcs z3with_open.._decorator..wrapped_func)r)r!r#)rr")r!r _decorators zwith_open.._decoratorr)r!rr"r$r)rr"r with_opengsr%c Cst|tr(|dr(ddlm}||St|tr<|dSt|ttfrXttt |Sy t |gSt k r}zd|df|_ Wdd}~XYnXdS)aParse a number of possible input types into a list of filepaths. Parameters ---------- flist : `file-like` or `list-like` iterable the input data container, normally just a single file path, or a list of paths, but can generally be any of the following - `str` representing a single file path (or comma-separated collection) - open `file` or `~gzip.GzipFile` object - :class:`~lal.utils.CacheEntry` - `str` with ``.cache`` or ``.lcf`` extension - simple `list` or `tuple` of `str` paths Returns ------- files : `list` `list` of `str` file paths Raises ------ ValueError if the input `flist` cannot be interpreted as any of the above inputs )z.cachez.lcfz.ffl) read_cache,zCould not parse input z! as one or more file-like objectsN) rrrcacher'splitr tuplemap file_path ValueErrorr )flistr'excrrr file_lists      r1cCs~t|tr|drt|jSt|ttjfr6t|St|trPt|drP|j Sy|jSt k rxt d|YnXdS)aDetermine the path of a file. This doesn't do any sanity checking to check that the file actually exists, or is readable. Parameters ---------- fobj : `file`, `str`, `CacheEntry`, ... the file object or path to parse Returns ------- path : `str` the path of the underlying file Raises ------ ValueError if a file path cannnot be determined Examples -------- >>> from gwpy.io.utils import file_path >>> import pathlib >>> file_path("test.txt") 'test.txt' >>> file_path(pathlib.Path('dir') / 'test.txt') 'dir/test.txt' >>> file_path(open("test.txt", "r")) 'test.txt' >>> file_path("file:///home/user/test.txt") '/home/user/test.txt' zfile:rzCannot parse file name for N) rr startswithrpathosPathLikerhasattrrAttributeErrorr.)rrrrr-s" r-)Nrr)__doc__rr4tempfile functoolsr urllib.parser __author__iorGzipFile_TemporaryFileWrapperrrrrr%r1r-rrrrs   # 34