HR-e6 dZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZgdZgd Ze d d d dddejddddddf dZdZddZejddfdZddZdZdZdS)zP This file contains a contains the high-level functions to read a VOTable file. N)data)deprecated_renamed_argument) iterparser) exceptionstree)parseparse_single_table from_tablewritetovalidatereset_vo_warnings)ignorewarn exceptionpedanticverifyz5.0)sincerFc ddlm} |}|dvrtd|&| j} | dvr| dk}n| }t |t r|rdnd }n3|tvr*td d t| i} |||||||| d } t |trtj |}|t |tr|| d <tj|| 5}tj| d|| cdddS#1swxYwYdS)af Parses a VOTABLE_ xml file (or file-like object), and returns a `~astropy.io.votable.tree.VOTableFile` object. Parameters ---------- source : path-like or file-like Path or file-like object containing a VOTABLE_ xml file. If file, must be readable. columns : sequence of str, optional List of field names to include in the output. The default is to include all fields. invalid : str, optional One of the following values: - 'exception': throw an exception when an invalid value is encountered (default) - 'mask': mask out invalid values 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'. .. versionchanged:: 4.0 ``verify`` replaces the ``pedantic`` argument, which will be deprecated in future. .. versionchanged:: 5.0 The ``pedantic`` argument is deprecated. chunk_size : int, optional The number of rows to read before converting to an array. Higher numbers are likely to be faster, but will consume more memory. table_number : int, optional The number of table in the file to read in. If `None`, all tables will be read. If a number, 0 refers to the first table in the file, and only that numbered table will be parsed and read in. Should not be used with ``table_id``. table_id : str, optional The ID of the table in the file to read in. Should not be used with ``table_number``. filename : str, optional A filename, URL or other identifier to use in error messages. If *filename* is None and *source* is a string (i.e. a path), then *source* will be used as a filename for error messages. Therefore, *filename* is only required when source is a file-like object. unit_format : str, astropy.units.format.Base instance or None, optional The unit format to use when parsing unit attributes. If a string, must be the name of a unit formatter. The built-in formats include ``generic``, ``fits``, ``cds``, and ``vounit``. A custom formatter may be provided by passing a `~astropy.units.UnitBase` instance. If `None` (default), the unit format to use will be the one specified by the VOTable specification (which is ``cds`` up to version 1.3 of VOTable, and ``vounit`` in more recent versions of the spec). datatype_mapping : dict, optional A mapping of datatype names (`str`) to valid VOTable datatype names (str). For example, if the file being read contains the datatype "unsignedInt" (an invalid datatype in VOTable), include the mapping ``{"unsignedInt": "long"}``. Returns ------- votable : `~astropy.io.votable.tree.VOTableFile` object See Also -------- astropy.io.votable.exceptions : The exceptions this function may raise. r)conf)rmaskzBaccepted values of ``invalid`` are: ``'exception'`` or ``'mask'``.N)falsetruerrrzverify should be one of /)columnsinvalidr chunk_size table_numberfilename unit_formatdatatype_mappingr)_debug_python_based_parser)rr)configpos)rlower ValueErrorr isinstanceboolVERIFY_OPTIONSjoinstrospath expanduserrget_xml_iteratorr VOTableFiler )sourcerrrrrtable_idrr r!r"rconf_verify_lowercaser#iterators 8lib/python3.11/site-packages/astropy/io/votable/table.pyr r "s@mmooG+++ P   ~ $ 1 1 3 3 !$5 5 5*f4FF*F&$P &2F ~ % %NCHH^4L4LNNOOO $",  F&#,##F++Jvs33#z  $+E   S v6:::@@6RRSSSSSSSSSSSSSSSSSSs*EEEc x|dd|d<t|fi|}|S)a Parses a VOTABLE_ xml file (or file-like object), reading and returning only the first `~astropy.io.votable.tree.Table` instance. See `parse` for a description of the keyword arguments. Returns ------- votable : `~astropy.io.votable.tree.Table` object rNr)getr get_first_table)r2kwargsvotables r6r r sGzz.!!)!"~F%%f%%G  " " $ $$cddlm}t||r tj|}n)t|tjst d|||ddS)a Writes a `~astropy.io.votable.tree.VOTableFile` to a VOTABLE_ xml file. Parameters ---------- table : `~astropy.io.votable.tree.VOTableFile` or `~astropy.table.Table` instance. file : str or writable file-like Path or file object to write to tabledata_format : str, optional Override the format of the table(s) data to write. Must be one of ``tabledata`` (text representation), ``binary`` or ``binary2``. By default, use the format that was specified in each ``table`` object as it was created or read in. See :ref:`astropy:astropy:votable-serialization`. r)TablezPfirst argument must be astropy.io.vo.VOTableFile or astropy.table.Table instanceT)tabledata_formatr"N) astropy.tabler>r(rr1r TypeErrorto_xml)tablefiler?r>s r6r r s$$#####%  ++E22 t/ 0 0  +    LL /Dr<cddlm}m}d}|tj}d}g}d}t t |trtj |}tj |d5} | } dddn #1swxYwYtj| } | d|Jt |tr|}n2t!|dr|j}nt!|d r|j}nd }t'jd 5} t'jt'jd t.jd  t3| d|}n9#t4$r,} |t| Yd} ~ nd} ~ wwxYwdddn #1swxYwYd| D|z}| d|d|dt;|rt=j| }|D]}t/j |}|ds1||d|dn||ddz }|d}|drd}nd}||dddd|pd|ddtCj"|ddd !#|"|||d#|"|d$n|d%d}|rtj $|rdd&l%m&}|d'}n|j'}|(||\}}}|dkr>|d(||)d)n|d*|r|*St;|dko|dkS)+a: Prints a validation report for the given file. Parameters ---------- source : path-like or file-like Path to a VOTABLE_ xml file or `~pathlib.Path` object having Path to a VOTABLE_ xml file. If file-like object, must be readable. output : file-like, optional Where to output the report. Defaults to ``sys.stdout``. If `None`, the output will be returned as a string. Must be writable. xmllint : bool, optional When `True`, also send the file to ``xmllint`` for schema and DTD validation. Requires that ``xmllint`` is installed. The default is `False`. ``source`` must be a file on the local filesystem in order for ``xmllint`` to work. filename : str, optional A filename to use in the error messages. If not provided, one will be automatically determined from ``source``. Returns ------- is_valid : bool or str Returns `True` if no warnings were found. If ``output`` is `None`, the return value will be a string. r) color_printprint_code_lineFNTbinary)encodingnameurlz )recordalways)appendr)rrctg|]5}t|jtj!t |j6S) issubclasscategoryr VOWarningr,message).0xs r6 zvalidate...sE    aj*"6 7 7  AI   r<zValidation report for z is_somethingrTnlinerwarning is_warningyellowreddz: r%EXCz z )initial_indentsubsequent_indent)rDnchar z)astropy.io.votable found no violations. )xmlutilz1.1zxmllint schema violations: zutf-8zxmllint passed )+astropy.utils.consolerFrGioStringIOrr(r,r-r.r/rget_readable_fileobjreadBytesIOseekhasattrrJrKwarningscatch_warnings resetwarnings simplefilterrrSr r'rNwritelenr xml_readlinesparse_vowarningtextwrapfilllstripexistsr%rdversionvalidate_schemadecodegetvalue)r2outputxmllintrrFrG return_as_strlinesr;fdcontentcontent_buffer warning_linese xml_linesrZwlinecolorsuccessrdrystdoutstderrs r6r r s@CBBBBBBBM ~ EG&#,##F++ "6H = = =''))Z((N fc " " #HH VV $ $ #{HH VU # # #zHH"H   - - -!   h (>>> LL    9 <  BCCCG -27>>(++ - ?GGoG")"9"9(G"L"L a<< LL9 : : : LLw// 0 0 0 0 LL+ , , ,!   u::? +w!|+sN>BB#&B#45G*E=<G= F3"F.)G.F33GG GcDtj||S)a Given an `~astropy.table.Table` object, return a `~astropy.io.votable.tree.VOTableFile` file structure containing just that single table. Parameters ---------- table : `~astropy.table.Table` instance table_id : str, optional If not `None`, set the given id on the returned `~astropy.io.votable.tree.Table` instance. Returns ------- votable : `~astropy.io.votable.tree.VOTableFile` instance )r3)rr1r )rCr3s r6r r os $   & &ux & @ @@r<cft|trtj|} t j|5}|D]\}}}}|dkrddddS|D]\}}}}|dkrddddS ddddS#1swxYwYdS#t$rYdSwxYw)aW Reads the header of a file to determine if it is a VOTable file. Parameters ---------- source : path-like or file-like Path or file object containing a VOTABLE_ xml file. If file, must be readable. Returns ------- is_votable : bool Returns `True` if the given file is a VOTable file. xmlNFVOTABLET)r(r,r-r.r/rr0r')r2r5starttagr^r$s r6 is_votablersm&#,##F++  ( 0 0 H&.  "sAs%<<         &.  "sAs)##                            uusMB" B B"(B9 B"B B"BB"BB"" B0/B0cbddlm}m}|tt|fD]} |`#t $rYwxYwdS)z Resets all of the vo warning state so that warnings that have already been emitted will be emitted again. This is used, for example, by `validate` which must emit all warnings each time it is called. r) convertersrdN)r%rrdrr__warningregistry__AttributeError)rrdmodules r6rrsn&%%%%%%%z49 **    D s  ,,)N)__doc__rfr-sysrurm astropy.utilsrastropy.utils.decoratorsrastropy.utils.xmlrr%rr__all__r*DEFAULT_CHUNK_SIZEr r r rr r rrrPr<r6rs  @@@@@@((((((   100Z???   &  $NSNSNS@?NSb%%%(@ JD,D,D,D,NAAAA*Dr<