o +eD@sdZddlmZddlZddlmmZddlmZm Z ddl m Z ddl m Z er1ddl mZdddZdddZGddde jZGdddeZGdddeZGdddeZdS) z Tree processors manipulate the tree created by block processors. They can even create an entirely new `ElementTree` object. This is an excellent place for creating summaries, adding collected references, or last minute adjustments. ) annotationsN) TYPE_CHECKINGAny)util)inlinepatterns)Markdownmdrkwargsrreturnutil.Registry[Treeprocessor]cKsBt}|t|dd|t|dd|t|dd|S)z3 Build the default `treeprocessors` for Markdown. ZinlineZprettify unescaper)rZRegistryregisterInlineProcessorPrettifyTreeprocessorUnescapeTreeprocessor)r r Ztreeprocessorsr7lib/python3.10/site-packages/markdown/treeprocessors.pybuild_treeprocessors's rsobjectboolcCst|tjs t|tSdS)z_ Return `True` if object is a string but not an [`AtomicString`][markdown.util.AtomicString]. F) isinstancer AtomicStringstr)rrrrisString0s  rc@seZdZdZd ddZdS) Treeprocessora  `Treeprocessor`s are run on the `ElementTree` object before serialization. Each `Treeprocessor` implements a `run` method that takes a pointer to an `Element` and modifies it as necessary. `Treeprocessors` must extend `markdown.Treeprocessor`. root etree.Elementr etree.Element | NonecCsdS)a Subclasses of `Treeprocessor` should implement a `run` method, which takes a root `Element`. This method can return another `Element` object, and the existing root `Element` will be replaced, or it can modify the current tree and return `None`. Nr)selfrrrrrunAszTreeprocessor.runN)rr r r!)__name__ __module__ __qualname____doc__r#rrrrr7s rc@seZdZdZd8ddZd9d d Zd:ddZd;ddZdd?d d!Z d>d@d%d&Z d.linkTextrrSFN)rHrPr rG)findr)!_InlineProcessor__findPlaceholderr2getrrrKrIstrip$_InlineProcessor__processElementTextrHappendr.r+rr) r"r5rQrFrVZ strartIndexr6r3Z phEndIndexr=rHchildr;rrTrZ__processPlaceholderssN           *z%InlineProcessor.__processPlaceholderspatterninlinepatterns.PatternrCtuple[str, bool, int]cCst|tj}|jD]}||jvr|ddfSq |rEd}|||D]}|||\}} } | dus9| durC|| d7}d}q&n| ||d}|d|} |s]|ddfS|sn||}| d} | d} |durw|d| fSt|t st|j tjs|gt|D],} t|s| j r|j| j|| j |d| _ |j| jr|| j|| _q|||} |rd|d| | || dddfSd| |d| |dddfS) a Check if the line fits the pattern, create the necessary elements, add it to `stashed_nodes`. Arguments: data: The text to be processed. pattern: The pattern to be checked. patternIndex: Index of current pattern. startIndex: String index, from which we start searching. Returns: String with placeholders instead of `ElementTree` elements. FrNTrz{}{}{}z{}{}{}{}rS)rrrZANCESTOR_EXCLUDESlowerr.ZgetCompiledRegExpfinditerZ handleMatchr;matchstartrrHrrrKrr]tag_InlineProcessor__handleInlinepoprI_InlineProcessor__stashNoder0formatr:groups)r"r_r5rArCZ new_styleZexcluderdr=rer;ZleftDatar^r@rrrZ__applyPatternsj             zInlineProcessor.__applyPatternr!parents list[str]cCsNg}|dur|dur||j|j|}|dus|||dS)zBuild the ancestor list.N)r]rfrb parent_maprZrLextend)r"rQrlr.rrrZ__build_ancestorsMs z!InlineProcessor.__build_ancestorsNtreer.list[str] | NonecCsi|_|dur gn|dd}dd|D|_||fg}|r|\}}||_|||jg}|D]}|jrtt|jtj st|j |j |j} d|_| || |} | D] } ||j| d<qZ|| 7}| || f|j|jr||j} td} d|_| | | d}| jr| j|_t||d}||D]}||j|d<|||dqt|r||j|<| ||jddfq4|D]\}} t| D]\}}|d}|||qq|s |S)aApply inline patterns to a parsed Markdown tree. Iterate over `Element`, find elements with inline tag, apply inline patterns and append newly created Elements to tree. To avoid further processing of string with inline patterns, instead of normal string, use subclass [`AtomicString`][markdown.util.AtomicString]: node.text = markdown.util.AtomicString("This will not be processed.") Arguments: tree: `Element` object, representing Markdown tree. ancestors: List of parent tag names that precede the tree node (if needed). Returns: An element tree object with applied inline patterns. NcSsi|] }|D]}||qqSrr).0pcrrr osz'InlineProcessor.run..rdFr)r2iterrnrhr.!_InlineProcessor__build_ancestorsrHrrrr]rfrbrJrgrIetreeZElementrKr6rLrMr+ enumerate)r"rpr.Z tree_parentsstackZ currElementrlZ insertQueuer^rHZlstitemrIZdumbyZ tailResultrNrOelementiobjrrrr#Ws^        *zInlineProcessor.run)r r)r0rr r1)r5rr6r7r r8)r=r>r0rr r)r)r5rrAr7r r)T)r=r rEr rFrr rG)r5rPrQr rFrr rR) r_r`r5rrAr7rCr7r ra)rQr!rlrmr rGN)rpr r.rqr r )r$r%r&r'r/r?rYrirgr\rJrBrxr#rrrrrKs     ! V P rc@s$eZdZdZd ddZd d d Zd S)rz' Add line breaks to the html document. elemr r rGcCsd}|j|jr7|jdvr7|jr|js&t|r&|j|djr&||_|D]}|j|jr6||q(|jr?|jsD||_dSdS)z8 Recursively add line breaks to `ElementTree` children.  )codeprerN)r Zis_block_levelrfrHr[r+_prettifyETreerI)r"rr~errrrs  z$PrettifyTreeprocessor._prettifyETreercCs|||d}|D]}|jr|jsd|_q d|j|_q |d}|D]%}t|rM|djdkrM|d}t|sM|jdurMt|j d|_q(dS)z7 Add line breaks to `Element` object and its children. brrz %srrrN) rrwrIr[r+rfrHrrrstrip)r"rZbrsrZpresrrrrrr#s   zPrettifyTreeprocessor.runN)rr r rGrr r rG)r$r%r&r'rr#rrrrrs rc@sDeZdZdZedejej Z dddZ dd d Z dddZ dS)rz Restore escaped chars z {}(\d+){}r< re.Match[str]r rcCstt|dS)Nr)chrr7r:)r"r<rrr _unescapeszUnescapeTreeprocessor._unescaperHcCs|j|j|Sr)REsubr)r"rHrrrrszUnescapeTreeprocessor.unescaperr rGcCsh|D]-}|jr|jdks||j|_|jr||j|_|D] \}}||||q#qdS)z/ Loop over all elements and unescape all text. rN)rwrHrfrrIitemsset)r"rrkeyvaluerrrr#s zUnescapeTreeprocessor.runN)r<rr r)rHrr rr)r$r%r&r'recompilerjrZSTXr*rrrr#rrrrrs   r)r rr rr r )rrr r)r'Z __future__rrZxml.etree.ElementTreeryZ ElementTreetypingrrrWrrZmarkdownrrrZ Processorrrrrrrrrs"      V'