U e{j@sdZddlZddlZddlZddlmZdddgZejddZdd d Z e Z zd 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|jVW5QRXdS)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. Nbinaryencodingwin) callablerget_readable_fileobjsysplatform startswithread isinstanceioFileIO)fdZnew_fdr[/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/utils/xml/iterparser.py_convert_to_fd_or_read_functions)   rc#sddlm}t|s|j}n|}ggfdd}fdd}|d_|_|_j_ j }||}|r||dEdHdd=||}qx|d dEdHdS) Nr)expatcs(d||jjffdd=dS)NT)appendCurrentLineNumberCurrentColumnNumber)nameattrparserqueuetextrrstartSsz_fast_iterparse..startcs(d|djjffdS)NF)rjoinstriprr)rrrrendYs  z_fast_iterparse..endTFr#) xml.parsersrr r ParserCreatespecified_attributesStartElementHandlerEndElementHandlerrCharacterDataHandlerParse)rZ buffersizerrr"r&r-rrrr_fast_iterparseHs,       r.) _iterparserFc cs8t|&}|rt|}nt|}t|VW5QRXdS)af 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 ---------- source : path-like, readable file-like, or callable Handle that contains the data or function that reads it. If a function or callable object, it must directly read from a stream. Non-callable objects must define a ``read`` method. 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_parserrcontextrrrrs %  c CsFt|*}t|\}}}}|r&|dkr.tdW5QRX|dpDdS)a Determine the encoding of an XML file by reading its header. Parameters ---------- source : path-like, readable file-like, or callable Handle that contains the data or function that reads it. If a function or callable object, it must directly read from a stream. Non-callable objects must define a ``read`` method. Returns ------- encoding : str xmlzInvalid XML filerzutf-8)rnextOSErrorget)r3iteratorr"tagrposrrrrs   c Cs8t|}tj||d}|d|}W5QRX|S)a Get the lines from a given XML file. Correctly determines the encoding and always returns unicode. Parameters ---------- source : path-like, readable file-like, or callable Handle that contains the data or function that reads it. If a function or callable object, it must directly read from a stream. Non-callable objects must define a ``read`` method. Returns ------- lines : list of unicode rr)rrr seek readlines)r3rinputZ xml_linesrrrrs  )r)F)__doc__ contextlibrr Z astropy.utilsr__all__contextmanagerrr.r1r#r0Z IterParser ImportErrorrrrrrrrs$   6 /   ,