o hm@sdZddlZddlmZddlmZddlmZedZ edZ edZ d e d e fd d ZGd ddZGdddZGdddeZGdddeZGdddZdS)z= Utility classes for rewriting elements of HTML-like strings N) defaultdict)Callable)cached_propertyz ]*)>z]*)/>z([\w-]+)\="([^"]*)" attr_stringreturncCsHi}t|D]\}}|dddddddd}|||<q|S) zR helper method to extract tag attributes, as a dict of un-escaped strings z<z""z&&) FIND_ATTRSfindallreplace)r attributesnamevalr[/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/rich_text/rewriters.py extract_attrss  rc@s<eZdZdZddZeddZeddZedd Z d S) TagMatchz5Represents a single matched tag in a rich text stringcCs||_d|_dSN)match replacement)selfrrrr__init__#s zTagMatch.__init__cCst|jdS)N)rrgrouprrrrattrs'szTagMatch.attrscC |jSr)rstartrrrrr+ zTagMatch.startcCrr)rendrrrrr!/r z TagMatch.endN) __name__ __module__ __qualname____doc__rrrpropertyrr!rrrrr s  rc@sxeZdZdddZddZddZdd Zd ed efd d Zd ed e ee e ffddZ de d e fddZddZdS) TagRewriterNcCs"|pi|_|pi|_|p i|_dSr)rules bulk_rulesreference_extractors)rr(r)r*rrrr5s  zTagRewriter.__init__cCtrNotImplementedErrorrrrrget_opening_tag_regex:z!TagRewriter.get_opening_tag_regexcCr+)z;Given a dict of attributes from a tag, return the tag type.r,rrrrrget_tag_type_from_attrs=sz#TagRewriter.get_tag_type_from_attrscCr+)zGiven a list of attribute dicts, all taken from tags of the same type, return a corresponding list of replacement strings to replace the tags with. Return an empty list for cases when you don't want any replacements made. r,)rtag_type attrs_listrrrget_tag_replacementsAsz TagRewriter.get_tag_replacementshtmlrc Cs||}g}|D]&\}}dd|D}|||}|sq t||D] \}} | |_||q$q |jdddd} |D]$}|d|j| |j||j| d}| t |j|j|j7} q>|S)NcSsg|]}|jqSr)r).0rrrr Osz(TagRewriter.__call__..cSs|jSr)r)rrrrZsz&TagRewriter.__call__..)keyr) extract_tagsitemsr4ziprappendsortrr!len) rr5matches_by_tag_typematches_to_replacer2 tag_matches attr_dicts replacementsrroffsetrrr__call__Is,   zTagRewriter.__call__cCsFtt}|}||D]}t|}||j}|||q |S)zHelper method to extract and group HTML tags and their attributes. Returns a dict of TagMatch objects, mapping tag types to a list of all TagMatch objects of that tag type. )rlistr.finditerrr1rr=)rr5r@ re_patternre_match tag_matchr2rrrr:hs zTagRewriter.extract_tagsrulecsfdd}|S)Ncstt|Sr)rGmap)argsrLrr bulk_rule{sz8TagRewriter.convert_rule_to_bulk_rule..bulk_ruler)rrLrPrrOrconvert_rule_to_bulk_rulezs z%TagRewriter.convert_rule_to_bulk_ruleccsP|}||D]}t|}||}||jvrq |j||EdHq gSr)r.r rr1r*)rr5rIrrr2rrrextract_referencess  zTagRewriter.extract_references)NNN)r"r#r$rr.r1r4strrFdictrGrr:rrQrRrrrrr'4s  r'c@(eZdZdZddZddZddZdS) EmbedRewriterz Rewrites tags within rich text into the HTML fragment given by the embed rule for 'foo'. Each embed rule is a function that takes a dict of attributes and returns the HTML fragment. cCtSr)FIND_EMBED_TAGrrrrr.r/z#EmbedRewriter.get_opening_tag_regexcCs |dS)N embedtype)getr0rrrr1 z%EmbedRewriter.get_tag_type_from_attrscCspz|j|}Wn tyd}Ynw|s+z|j|}Wn ty%Ynw||}|r1||Sdgt|S)Nr)KeyErrorr(rQr?rr2r3rLrrrr4s   z"EmbedRewriter.get_tag_replacementsNr"r#r$r%r.r1r4rrrrrVs  rVc@rU) LinkRewriterz Rewrites tags within rich text into the HTML fragment given by the rule for 'foo'. Each link rule is a function that takes a dict of attributes and returns the HTML fragment for the opening tag (only). cCrWr) FIND_A_TAGrrrrr.r/z"LinkRewriter.get_opening_tag_regexcCsfz|dWSty2|dd}|r,|drYdS|dr$YdS|dr/YdSYdSYdSw) Nlinktypehref)zhttp:zhttps:externalzmailto:email#anchor)r^rZ startswith)rrrdrrrr1s      z$LinkRewriter.get_tag_type_from_attrscCs|sgSz|j|}Wn tyd}Ynw|s7z|j|}Wnty1|dvr/gYSYnw||}|r=||Sdgt|S)N)rfrerhzr]r_rrrr4s"   z!LinkRewriter.get_tag_replacementsNr`rrrrras  rac@rU) MultiRuleRewriterz:Rewrites HTML by applying a sequence of rewriter functionscCs ||_dSr rewriters)rrlrrrrr[zMultiRuleRewriter.__init__cCs|jD]}||}q|Srrk)rr5rewriterrrrFs  zMultiRuleRewriter.__call__ccs"|jD] }||EdHqdSr)rlrR)rr5rewriterrrrrRs z$MultiRuleRewriter.extract_referencesN)r"r#r$r%rrFrRrrrrrjs  rj)r%re collectionsrtypingrdjango.utils.functionalrcompilerbrXr rSrTrrr'rVrarjrrrrs      Z8