B dvi @sjddlZddlZddlZddlZddlZddlmZddlmZddl Z dddddd d d d d dddg Z dgZ eZ eZeZeejfZGdd d eZeZejddZd1dd ZddZd2ddZddZd3ddZddZd4ddZddZ d dZ!d!d"Z"d#dZ#d$d Z$ddd%d&d Z%dd'd(d Z&d)d*Z'd+d,Z(d-d.Z)Gd/ddZ*Gd0ddZ+dS)5N) OrderedDict) itemgetterregister_readerregister_writerregister_identifieridentify_format get_reader get_writerreadwrite get_formatsIORegistryErrordelay_doc_updatesUnifiedReadWriteMethodUnifiedReadWritec@seZdZdZdS)r z'Custom error for registry clashes. N)__name__ __module__ __qualname____doc__rr`/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/astropy/io/registry.pyr sccs2t|dVt|t|dt|ddS)aContextmanager to disable documentation updates when registering reader and writer. The documentation is only built once when the contextmanager exits. .. versionadded:: 1.3 Parameters ---------- cls : class Class for which the documentation updates should be delayed. Notes ----- Registering multiple readers and writers can cause significant overhead because the documentation of the corresponding ``read`` and ``write`` methods are build every time. .. warning:: This contextmanager is experimental and may be replaced by a more general approach. Examples -------- see for example the source code of ``astropy.table.__init__``. Nr r )_delayed_docs_classesadddiscard_update__doc__)clsrrrr)s    c Csbddlm}tttttBtdd}g}x|D]}|dk rPt||d|sPq2|tkr\dnd}|tkrldnd}|tkr|dnd}d|d|df} | |krdnd } | |dj |d|||| fq2W|dk r |d krd d |D}n"|d krdd |D}nt d|r0t t t|tdddd} nd} || dd} t| ddks^| d| S)a Get the list of registered I/O formats as a Table. Parameters ---------- data_class : class, optional Filter readers/writer to match data class (default = all classes). readwrite : str or None, optional Search only for readers (``"Read"``) or writers (``"Write"``). If None search for both. Default is None. .. versionadded:: 1.3 Returns ------- format_table : :class:`~astropy.table.Table` Table of available I/O formats. r)Table)keyNYesZNozascii.ReadcSsg|]}|ddkr|qS)rr).0rowrrr zszget_formats..WritecSsg|]}|ddkr|qS)rr)r#r$rrrr%|szQunrecognized value for "readwrite": {0}. Allowed are "Read" and "Write" and None.)z Data classZFormatr!r&z Auto-identify Deprecated)namesr))Z astropy.tablersortedset_readers_writersr_is_best_match _identifiersappendr ValueErrorlistzipnpany remove_column) data_class readwriterformat_classesrowsZ format_classZhas_readZ has_writeZ has_identifyZascii_format_class deprecateddata format_tablerrrr Ms6       c sldt||}t|jtsdS|j}fddt|D}|rV|d}|d|}dd|ddD}dtd d |Dt||}| d |j d d d}t dd|d} | |d<| d| || d|jkr|dddgdg|}|fdd|Dt|tr2d||j_n6yd||_Wn$tk rfd||j_YnXdS)z Update the docstring to include all the available readers / writers for the ``data_class.read`` or ``data_class.write`` functions (respectively). z#The available built-in formats are:Ncsg|]\}}|kr|qSrr)r#iiline) FORMATS_TEXTrrr%sz"_update__doc__..rcSsg|]}td|qS)z(\S))research)r#r@rrrr%sr css|]}|r|VqdS)N)start)r#matchrrr sz!_update__doc__..z Data classP) max_lines max_width-=r)r zZDeprecated format names like ``aastex`` will be removed in a future version. Use the full z%name (e.g. ``ascii.aastex``) instead.csg|] }|qSrr)r#r@) left_indentrrr%s )getattr isinstancerstr splitlines enumerateminr capitalizer7pformatrBsubinsertr1Zcolnamesextendrjoin __class__AttributeError__func__) r8r9Zclass_readwrite_funclinesZ sep_indicesZ chop_indexmatchesr> new_linesZ table_rst_sepr)rArNrrs<          rFcCsJ||ftks|r"||ft||f<ntd||j|tkrFt|ddS)a Register a reader function. Parameters ---------- data_format : str The data format identifier. This is the string that will be used to specify the data type when reading. data_class : class The class of the object that the reader produces. function : function The function to read in a data object. force : bool, optional Whether to override any existing function if already present. Default is ``False``. priority : int, optional The priority of the reader, used to compare possible formats when trying to determine the best reader to use. Higher priorities are preferred over lower priorities, with the default priority being 0 (negative numbers are allowed though). z8Reader for format '{}' and class '{}' is already definedr N)r-r formatrrr) data_formatr8functionforcepriorityrrrrs  cCsD||ftkrt||fntd||j|tkr@t|ddS)z Unregister a reader function Parameters ---------- data_format : str The data format identifier. data_class : class The class of the object that the reader produces. z0No reader defined for format '{}' and class '{}'r N)r-popr rbrrr)rcr8rrrunregister_readers  rhcCsJ||ftks|r"||ft||f<ntd||j|tkrFt|ddS)a Register a table writer function. Parameters ---------- data_format : str The data format identifier. This is the string that will be used to specify the data type when writing. data_class : class The class of the object that can be written. function : function The function to write out a data object. force : bool, optional Whether to override any existing function if already present. Default is ``False``. priority : int, optional The priority of the writer, used to compare possible formats when trying to determine the best writer to use. Higher priorities are preferred over lower priorities, with the default priority being 0 (negative numbers are allowed though). z8Writer for format '{}' and class '{}' is already definedr N)r.r rbrrr)rcr8rdrerfrrrrs  cCsD||ftkrt||fntd||j|tkr@t|ddS)z Unregister a writer function Parameters ---------- data_format : str The data format identifier. data_class : class The class of the object that can be written. z0No writer defined for format '{}' and class '{}'r N)r.rgr rbrrr)rcr8rrrunregister_writer's  ricCs4||ftks|r|t||f<ntd||jdS)a/ Associate an identifier function with a specific data type. Parameters ---------- data_format : str The data format identifier. This is the string that is used to specify the data type when reading/writing. data_class : class The class of the object that can be written. identifier : function A function that checks the argument specified to `read` or `write` to determine whether the input can be interpreted as a table of type ``data_format``. This function should take the following arguments: - ``origin``: A string ``"read"`` or ``"write"`` identifying whether the file is to be opened for reading or writing. - ``path``: The path to the file. - ``fileobj``: An open file object to read the file's contents, or `None` if the file could not be opened. - ``*args``: Positional arguments for the `read` or `write` function. - ``**kwargs``: Keyword arguments for the `read` or `write` function. One or both of ``path`` or ``fileobj`` may be `None`. If they are both `None`, the identifier will need to work from ``args[0]``. The function should return True if the input can be identified as being of format ``data_format``, and False otherwise. force : bool, optional Whether to override any existing function if already present. Default is ``False``. Examples -------- To set the identifier based on extensions, for formats that take a filename as a first argument, you can do for example:: >>> def my_identifier(*args, **kwargs): ... return isinstance(args[0], str) and args[0].endswith('.tbl') >>> register_identifier('ipac', Table, my_identifier) zformat_table_strrrr_get_format_table_strs  rscsbfddtD}xJ|D]$\}}t|||rt||fdSqWt|d}td|j|dS)aGet reader for ``data_format``. Parameters ---------- data_format : str The data format identifier. This is the string that is used to specify the data type when reading/writing. data_class : class The class of the object that can be written. Returns ------- reader : callable The registered reader function for this format and class. cs g|]\}}|kr||fqSrr)r#fmtr)rcrrr%szget_reader..rr!zQNo reader defined for format '{}' and class '{}'. The available formats are: {}N)r-r/rsr rbr)rcr8ZreadersZ reader_formatZ reader_classrrr)rcrrs  csbfddtD}xJ|D]$\}}t|||rt||fdSqWt|d}td|j|dS)aGet writer for ``data_format``. Parameters ---------- data_format : str The data format identifier. This is the string that is used to specify the data type when reading/writing. data_class : class The class of the object that can be written. Returns ------- writer : callable The registered writer function for this format and class. cs g|]\}}|kr||fqSrr)r#rtr)rcrrr%szget_writer..rr&zQNo writer defined for format '{}' and class '{}'. The available formats are: {}N)r.r/rsr rbr)rcr8ZwritersZ writer_formatZ writer_classrrr)rcrr s  )rbcachec Osd}z`|dkrd}d}t|rt|dtrtj|dsddlm}t|dtjrxt |df|dd}|d}y||dd|d}| }Wn,t k rYqt k rd}YqX|gt |dd}nt|ddrd}|d}td|||||}t||} | ||} t| |sdy || } Wn&t k rbtd|jYnXWd|dk r|jtX| S) zV Read in data. The arguments passed to this method depend on the format. Nr)get_readable_fileobjrbinary)encodingrur z,could not convert reader output to {} class.)lenrQ PATH_TYPESosrmisdirZastropy.utils.datarvPathLikefspath __enter__OSError Exceptionr3hasattr_get_valid_formatr TypeErrorrbr__exit__sysexc_info) rrbrurorpctxrmrnrvreaderr=rrrr sF         )rbcOs|dkrd}d}t|rzt|dtr`t|dtjrRt|df|dd}|d}d}nt|ddrzd}|d}td|j||||}t ||j}||f||dS)zX Write out data. The arguments passed to this method depend on the format. Nrrr r ) ryrQrzr{r}r~rrr\r )r=rbrorprmrnwriterrrrr !s  cCsDt||r@dd|D}x&|jD]}||kr0dS||kr dSq WdS)ad Determine if class2 is the "best" match for class1 in the list of classes. It is assumed that (class2 in classes) is True. class2 is the the best match if: - ``class1`` is a subclass of ``class2`` AND - ``class2`` is the nearest ancestor of ``class1`` that is in classes (which includes the case that ``class1 is class2``) cSsh|] \}}|qSrr)r#rtrrrr Hsz!_is_best_match..TF) issubclass__mro__)Zclass1Zclass2r:classesparentrrrr/=s  r/cCs\t||||||}t|dkrsL     $ D> ! ! 5&7#]