jnddlmZmZmZmZddlmZddlmZddl m Z m Z ddl m Z ddlmZeejejdfZGdd e ZdS) )AnyListOptionalUnion)etree)Self)DocumentPage) read_txt_file)loggerNceZdZdZ ddeedeffd Zdee fdZ e dee ffd Z d efd Z e dd ededeed edef d Ze ddededeedeed edef dZxZS) XMLDocumentzClass for handling .xml documents. This class uses rules based parsing to identify sections of interest within the document.N stylesheetparserc|s,|rtjdntjd}||_||_d|_t dS)aClass for parsing XML documents. XML documents are parsed using lxml. Parameters ---------- filename: The name of the XML file to read stylesheet: An XLST stylesheet that can be applied to transform the XML file parser: The lxml parser to use with the file. The HTML parser is used by default because it is more tolerant of special characters and malformed XML. If you are using a stylesheet, you likely want the XMLParser. T)remove_commentsN)r XMLParser HTMLParserrr document_treesuper__init__)selfrr __class__s d/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/unstructured/documents/xml.pyrzXMLDocument.__init__ss$ <5555%d;;;  % ! returnct)N)NotImplementedError)rs r_parse_pages_from_element_treez*XMLDocument._parse_pages_from_element_tree/s!!rcj|j||_tjS)z1Gets all elements from pages in sequential order.)_pagesrrpages)rrs rr"zXMLDocument.pages2s, ; ==??DKww}rcontentc\t|jtj}|r|ds|rd|z}|j tj||j}|tdn<#t$r/tj||j}YnwxYw|j rkt|jtjrtj dtj |j }tj |}||}||_|jS)zDReads in an XML file and converts it to an lxml element tree object. Nzdocument_tree is NonezYou are using the HTML parser with an XSLT stylesheet. Stylesheets are more commonly parsed with the XMLParser. If your HTML does not display properly, try `import lxml.etree as etree` and setting `parser=etree.XMLParser()` instead.) isinstancerrr startswithr fromstring ValueErrorencoderr warningparseXSLT)rr#is_html_parserrxslt transforms r _read_xmlzXMLDocument._read_xml9sC $DK1ABB  %7--d33 % %WnG   % P % 0$+ F F  ($%<===) P P P % 01A1A4; O O  P 9dk5+;<<N>{4?33!Jt,, ) - 8 8 !.D !!s+A226B+*B+textkwargsc ntjd|d||d|}|||S)zFSupports reading in an XML file as a raw string rather than as a file.z Reading document from string ...rr)r infor1)clsr2rrr3docs r from_stringzXMLDocument.from_string`sH  6777cAJAA&AA d rfilenameencodingc Lt||\}}|j|f||d|S)N)r;r<r5)r r:)r8r;rrr<r3_r#s r from_filezXMLDocument.from_filens<#HxHHH 7swWv*WWPVWWWr)NN)NNN)__name__ __module__ __qualname____doc__rstr VALID_PARSERSrrr rpropertyr"r1 classmethodrrr:r? __classcell__)rs@rrr s11 %) $SM<"T """"tDzX %"%"%"%"%"N!%$(    SM       [ !%$("& X X X XSM X 3- X  X  X X X[ X X X X Xrr)typingrrrrlxmlrtyping_extensionsrunstructured.documents.baser r unstructured.file_utils.encodingr unstructured.loggerr rrrErr6rrrOs------------""""""66666666::::::&&&&&&e&=> lXlXlXlXlX(lXlXlXlXlXr