a ߙfb@sddlZddlmZmZddlmZmZddlm Z ddl mZddl m Z ddlmZdd lmZd d Zdd dZdddZe deee deee deedS)N)parse from_table) VOTableFileTable)registry)r) BaseColumn)Quantity)NOT_OVERWRITING_MSGc Osxddlm}|dkrp|durDz||}W|dn |d0|S|durT||St|dttfrjdSdSndSdS)a7 Reads the header of a file to determine if it is a VOTable file. Parameters ---------- origin : str or readable file-like Path or file object containing a VOTABLE_ xml file. Returns ------- is_votable : bool Returns `True` if the given file is a VOTable file. r) is_votablereadNrTF)r seek isinstancerVOTable)originfilepathZfileobjargskwargsr resultr9lib/python3.9/site-packages/astropy/io/votable/connect.pyr s  r FcKs<t|ttfs$t|f||d|}t}g}t|tr |D]"}|jdurZ|||j<||qBt|dkr|durt d d | t|nbt|t r||vr||}qt d|dn4t|tr|t|kr||}ntd |t|n t|dkr|d }nt d nt|tr0|}|j|d S) a Read a Table object from an VO table file Parameters ---------- input : str or `~astropy.io.votable.tree.VOTableFile` or `~astropy.io.votable.tree.Table` If a string, the filename to read the table from. If a :class:`~astropy.io.votable.tree.VOTableFile` or :class:`~astropy.io.votable.tree.Table` object, the object to extract the table from. table_id : str or int, optional The table to read in. If a `str`, it is an ID corresponding to the ID of the table in the file (not all VOTable files assign IDs to their tables). If an `int`, it is the index of the table in the file, starting at 0. use_names_over_ids : bool, optional When `True` use the ``name`` attributes of columns as the names of columns in the `~astropy.table.Table` instance. Since names are not guaranteed to be unique, this may cause some columns to be renamed by appending numbers to the end. Otherwise (default), use the ID attributes as the column names. verify : {'ignore', 'warn', 'exception'}, optional When ``'exception'``, raise an error when the file violates the spec, otherwise either issue a warning (``'warn'``) or silently continue (``'ignore'``). Warnings may be controlled using the standard Python mechanisms. See the `warnings` module in the Python standard library for more information. When not provided, uses the configuration setting ``astropy.io.votable.verify``, which defaults to ``'ignore'``. **kwargs Additional keyword arguments are passed on to :func:`astropy.io.votable.table.parse`. )table_idverifyNrzMultiple tables found: table id should be set via the table_id= argument. The available tables are {}, or integers less than {}.z, zNo tables with id=z foundz/Table index {} is out of range. {} tables foundrzNo table found)use_names_over_ids)rrrrdictZ iter_tablesZIDappendlen ValueErrorformatjoinkeysstrint IndexErrorZto_table)inputrrrrZtable_id_mappingZtablestablerrrread_table_votable0sH&               r'cCs|jttf}|r0dd|D}td|t|trdtj |rd|rVt |nt t |t||d}|j||ddS)a Write a Table object to an VO table file Parameters ---------- input : Table The table to write out. output : str The filename to write the table to. table_id : str, optional The table ID to use. If this is not specified, the 'ID' keyword in the ``meta`` object of the table will be used. overwrite : bool, optional Whether to overwrite any existing file without warning. tabledata_format : str, optional The format of table data to write. Must be one of ``tabledata`` (text representation), ``binary`` or ``binary2``. Default is ``tabledata``. See :ref:`astropy:votable-serialization`. cSsg|] }|jjqSr)infoname).0colrrr z'write_table_votable..z5cannot write table with mixin column(s) {} to VOTable)r)tabledata_formatN)columnsZnot_isinstancerr rrrr"ospathexistsremoveOSErrorr rZto_xml)r%outputrZ overwriter.Zunsupported_colsZunsupported_namesZ table_filerrrwrite_table_votables  r6Zvotable)NFN)NFN)r0r rrZtreerrrZ astropy.iorZ io_registryZ astropy.tableZastropy.table.columnrZ astropy.unitsr Zastropy.utils.miscr r r'r6Zregister_readerZregister_writerZregister_identifierrrrrs"       R /