IR-e(dZddlZddlZ ddlmZejZejZn#e $r dZdZYnwxYwGddZ dS) zZ Contains a class that makes it simple to stream out well-formed and nicely-indented XML. N) _iterparserc|dd}|dd}|dd}|S)z< Escapes &, < and > in an XML CDATA string. &&<<>>replacess 8lib/python3.11/site-packages/astropy/utils/xml/writer.pyxml_escape_cdatarsB IIc7 # # IIc6 " " IIc6 " "c|dd}|dd}|dd}|dd}|d d }|S) zE Escapes &, ', ", < and > in an XML attribute value. rr'z'"z"rr r r r rs r xml_escapersh IIc7 # # IIc8 $ $ IIc8 $ $ IIc6 " " IIc6 " "rceZdZdZdZddZifdZejddZ ejifd Z d Z d Z dd Z dZd difdZdZdZddZedZd S) XMLWriteraN A class to write well-formed and nicely indented XML. Use like this:: w = XMLWriter(fh) with w.tag('html'): with w.tag('body'): w.data('This is the content') Which produces:: This is the content c|j|_t|dr |j|_d|_g|_g|_d|_t|_t|_dS)zQ Parameters ---------- file : writable file-like flushrz@ N) writehasattrr_open_tags_data _indentationrr)selffiles r__init__zXMLWriter.__init__<sX Z 4 ! ! $DJ   $ 0$rTFcp|jr4|r|dn|dd|_|jrd|j}|r|d}t j|||}|d||||d||n(|||g|_dSdS) z) Flush internal buffers. > r rr)initial_indentsubsequent_indent N)rrrjoinget_indentation_spacestextwrapfillr)r!indentwrapdatas r_flushzXMLWriter._flushMs2 :   5!!!! 3DJ : 774:&&D 844Q77}6 4    40066777 4    466889999 40066777DJJJ  rc \|g|_|j|||d|d||s|r|}||t| }| |D]8\}}|1| |}|d|d|d9d|_ t|jS)am Opens a new element. Attributes can be given as keyword arguments, or as a string/string dictionary. The method returns an opaque identifier that can be passed to the :meth:`close` method, to close all open elements up to and including this one. Parameters ---------- tag : str The element name attrib : dict of str -> str Attribute dictionary. Alternatively, attributes can be given as keyword arguments. Returns ------- id : int Returns an element identifier. rN z="rr)r1rrappendrr+copyupdatelistitemssortrrlen)r!tagattribextrakvs rstartzXMLWriter.startfs,   # 4..r22333 9s99  .U .[[]]F MM% &,,..))F KKMMM . .1=**AJJ}1}}}}}--- 4:r escape_xmlc+K|j}|dkr3 ddln#t$rtdwxYwifd|_n$|dkr d|_n|dkrtd dV||_dS) a[Context manager to control how XML data tags are cleaned (escaped) to remove potentially unsafe characters or constructs. The default (``method='escape_xml'``) applies brute-force escaping of certain key XML characters like ``<``, ``>``, and ``&`` to ensure that the output is not valid XML. In order to explicitly allow certain XML tags (e.g. link reference or emphasis tags), use ``method='bleach_clean'``. This sanitizes the data string using the ``clean`` function of the `bleach `_ package. Any additional keyword arguments will be passed directly to the ``clean`` function. Finally, use ``method='none'`` to disable any sanitization. This should be used sparingly. Example:: w = writer.XMLWriter(ListWriter(lines)) with w.xml_cleaning_method('bleach_clean'): w.start('td') w.data('google.com') w.end() Parameters ---------- method : str Cleaning method. Allowed values are "escape_xml", "bleach_clean", and "none". **clean_kwargs : keyword args Additional keyword args that are passed to the bleach.clean() function. bleach_cleanrNzTbleach package is required when HTML escaping is disabled. Use "pip install bleach".c j|fiSN)clean)xbleach clean_kwargss rz/XMLWriter.xml_cleaning_method..slfl1.M.M .M.Mrnonec|SrF)rHs rrKz/XMLWriter.xml_cleaning_method..sarrBzEallowed values of method are "escape_xml", "bleach_clean", and "none")rrI ImportError ValueError)r!methodrJcurrent_xml_escape_cdatarIs ` @rxml_cleaning_methodzXMLWriter.xml_cleaning_methodsJ$(#8 ^ # #      0  #! $M$M$M$M$MD ! ! v  $/KD ! ! | # #W   8s2c+ZK|j||fi|dV||dS)aT A convenience method for creating wrapper elements using the ``with`` statement. Examples -------- >>> with writer.tag('foo'): # doctest: +SKIP ... writer.element('bar') ... # is implicitly closed here ... Parameters are the same as to `start`. N)rAend)r!r<r=r>s rr<z XMLWriter.tags@  3((%(((   rc||||d||ddS)z Adds a comment to the output stream. Parameters ---------- comment : str Comment text, as a Unicode string. z N)r1rr+r)r!comments rrWzXMLWriter.commentsa  4..00111 A40099AAABBBBBrc:|j|dS)z Adds character data to the output stream. Parameters ---------- text : str Character data, as a Unicode string. N)rr5)r!texts rr0zXMLWriter.datas  $rNc |rL|jstd|d||jdkr td|jdd|n|jstd|j}|jr|||n%|jrd|_|dd S|r'|||d |d d S) a Closes the current element (opened by the most recent call to `start`). Parameters ---------- tag : str Element name. If given, the tag must match the start tag. If omitted, the current element is closed. zunbalanced end()r3z expected end(z), got zunbalanced end()rz/> Nzs relementzXMLWriter.element(sS  3((%(((   IIdOOO D)))))rcdSrFrNr!s rrzXMLWriter.flush3s rc*t|jS)z\ Returns the number of indentation levels the file is currently in. )r;rrcs rget_indentationzXMLWriter.get_indentation6s 4:rrcJ|jdt|j|zS)z` Returns a string of spaces that matches the current indentation level. N)r r;r)r!offsets rr+z XMLWriter.get_indentation_spaces=s%  !;3tz??V#;!;<rzs (0#3'JJ1     6uuuuuuuuuus ! //