B q\O@sdZddlZddlZddlZddlmZdddgZejddZdd d Z e Z yd d l m Z e j Z Wnek rxYnXejdddZddZddZdS)z8 This module includes a fast iterator-based XML parser. N)dataget_xml_iteratorget_xml_encoding xml_readlinesc csbt|r|VdStj|dd8}tjdr8|jVnt|tj rL|Vn|jVWdQRXdS)a Returns a function suitable for streaming input, or a file object. This function is only useful if passing off to C code where: - If it's a real file object, we want to use it as a real C file object to avoid the Python overhead. - If it's not a real file object, it's much handier to just have a Python function to call. This is somewhat quirky behavior, of course, which is why it is private. For a more useful version of similar behavior, see `astropy.utils.misc.get_readable_fileobj`. Parameters ---------- fd : object May be: - a file object. If the file is uncompressed, this raw file object is returned verbatim. Otherwise, the read method is returned. - a function that reads from a stream, in which case it is returned verbatim. - a file path, in which case it is opened. Again, like a file object, if it's uncompressed, a raw file object is returned, otherwise its read method. - an object with a :meth:`read` method, in which case that method is returned. Returns ------- fd : context-dependent See above. NZbinary)encodingwin) callablerget_readable_fileobjsysplatform startswithread isinstanceioFileIO)fdZnew_fdr;lib/python3.7/site-packages/astropy/utils/xml/iterparser.py_convert_to_fd_or_read_functions)   rc #sddlm}t|s|j}n|}ggfdd}fdd}|d_|_|_j_ j }||}x8|r||dxD] }|VqWdd=||}qzW|d dxD] }|VqWdS) Nr)expatcs(d||jjffdd=dS)NT)appendCurrentLineNumberCurrentColumnNumber)nameattr)parserqueuetextrrstartTs z_fast_iterparse..startcs(d|djjffdS)NF)rjoinstriprr)r)rrrrrendYsz_fast_iterparse..endTFr ) Z xml.parsersrrr Z ParserCreateZspecified_attributesZStartElementHandlerZEndElementHandlerrZCharacterDataHandlerParse) rZ buffersizerr rr#r$relemr)rrrr_fast_iterparseIs0        r&) _iterparserFc cs8t|&}|rt|}nt|}t|VWdQRXdS)a Returns an iterator over the elements of an XML file. The iterator doesn't ever build a tree, so it is much more memory and time efficient than the alternative in ``cElementTree``. Parameters ---------- fd : readable file-like object or read function Returns ------- parts : iterator The iterator returns 4-tuples (*start*, *tag*, *data*, *pos*): - *start*: when `True` is a start element event, otherwise an end element event. - *tag*: The name of the element - *data*: Depends on the value of *event*: - if *start* == `True`, data is a dictionary of attributes - if *start* == `False`, data is a string containing the text content of the element - *pos*: Tuple (*line*, *col*) indicating the source of the event. N)r_slow_iterparser&iter)sourceZ_debug_python_based_parserrcontextrrrr{s "  c CsFt|*}t|\}}}}|r&|dkr.tdWdQRX|dpDdS)z Determine the encoding of an XML file by reading its header. Parameters ---------- source : readable file-like object, read function or str path Returns ------- encoding : str ZxmlzInvalid XML fileNrzutf-8)rnextOSErrorget)r+iteratorrtagrposrrrrs  c Cs8t|}tj||d}|d|}WdQRX|S)a Get the lines from a given XML file. Correctly determines the encoding and always returns unicode. Parameters ---------- source : readable file-like object, read function or str path Returns ------- lines : list of unicode )rrN)rrr seek readlines)r+rinputZ xml_linesrrrrs  )r)F)__doc__ contextlibrr Z astropy.utilsr__all__contextmanagerrr&r)r r(Z IterParser ImportErrorrrrrrrrs"  7 *   )