U {hy@sdZddlZddlmZddlmZmZmZddlm Z e dZ e dZ e dZ eed d d ZGd d d ZGdddZGdddeZGdddeZGdddZdS)z= Utility classes for rewriting elements of HTML-like strings N) defaultdict)CallableDictList)cached_propertyz ]*)>z]*)/>z([\w-]+)\="([^"]*)") attr_stringreturncCsHi}t|D]4\}}|dddddddd}|||<q|S) zR helper method to extract tag attributes, as a dict of un-escaped strings z<z""z&&) FIND_ATTRSfindallreplace)r attributesnamevalrc/var/www/dating/data/www/fatepal.com/env/lib/python3.8/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__#szTagMatch.__init__cCst|jdS)N)rrgrouprrrrattrs'szTagMatch.attrscCs |jSr)rstartrrrrr +szTagMatch.startcCs |jSr)rendrrrrr!/sz TagMatch.endN) __name__ __module__ __qualname____doc__rrrpropertyr r!rrrrr s  rc@sreZdZdddZddZddZdd Zeed d d Zee ee e fd d dZ e e dddZddZdS) TagRewriterNcCs"|pi|_|pi|_|pi|_dSr)rules bulk_rulesreference_extractors)rr(r)r*rrrr5s  zTagRewriter.__init__cCstdSrNotImplementedErrorrrrrget_opening_tag_regex:sz!TagRewriter.get_opening_tag_regexcCstdS)z;Given a dict of attributes from a tag, return the tag type.Nr+rrrrrget_tag_type_from_attrs=sz#TagRewriter.get_tag_type_from_attrscCstdS)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. Nr+)rtag_type attrs_listrrrget_tag_replacementsAsz TagRewriter.get_tag_replacements)htmlrc Cs||}g}|D]L\}}dd|D}|||}|s>qt||D]\}} | |_||qHq|jdddd} |D]H}|d|j| |j||j| d}| t |j|j|j7} q||S)NcSsg|] }|jqSr)r).0rrrr Osz(TagRewriter.__call__..cSs|jSr)r )rrrrZz&TagRewriter.__call__..)keyr) extract_tagsitemsr2ziprappendsortr r!len) rr3matches_by_tag_typeZmatches_to_replacer0Z tag_matchesZ attr_dictsZ 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-finditerrr/rr<)rr3r? re_patternZre_matchZ tag_matchr0rrrr9hs zTagRewriter.extract_tags)rulercsfdd}|S)Ncstt|Sr)rBmap)argsrErr bulk_rule{sz8TagRewriter.convert_rule_to_bulk_rule..bulk_ruler)rrErIrrHrconvert_rule_to_bulk_rulezs z%TagRewriter.convert_rule_to_bulk_ruleccsN|}||D]6}t|}||}||jkr4q|j||EdHqgSr)r-rrr/r*)rr3rDrrr0rrrextract_referencess  zTagRewriter.extract_references)NNN)r"r#r$rr-r/r2strrArrrr9rrJrKrrrrr'4s r'c@s(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. cCstSr)FIND_EMBED_TAGrrrrr-sz#EmbedRewriter.get_opening_tag_regexcCs |dS)NZ embedtype)getr.rrrr/sz%EmbedRewriter.get_tag_type_from_attrscCstz|j|}Wntk r&d}YnX|sZz|j|}Wntk rNYn X||}|rf||Sdgt|S)Nr)KeyErrorr(rJr>rr0r1rErrrr2s  z"EmbedRewriter.get_tag_replacementsNr"r#r$r%r-r/r2rrrrrMsrMc@s(eZdZdZddZddZddZdS) 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). cCstSr) FIND_A_TAGrrrrr-sz"LinkRewriter.get_opening_tag_regexcCsdz |dWStk r^|dd}|rZ|dr:YdS|drJYdS|drZYdSYnXdS) Nlinktypehref)zhttp:zhttps:externalzmailto:email#anchor)rRrO startswith)rrrXrrrr/s     z$LinkRewriter.get_tag_type_from_attrscCs|sgSz|j|}Wntk r.d}YnX|srz|j|}Wn$tk rf|dkrbgYSYn X||}|r~||Sdgt|S)N)rZrYr\zrQrSrrrr2s  z!LinkRewriter.get_tag_replacementsNrTrrrrrUsrUc@s(eZdZdZddZddZddZdS) MultiRuleRewriterz:Rewrites HTML by applying a sequence of rewriter functionscCs ||_dSr rewriters)rr`rrrrszMultiRuleRewriter.__init__cCs|jD] }||}q|Srr_)rr3ZrewriterrrrAs  zMultiRuleRewriter.__call__ccs |jD]}||EdHqdSr)r`rK)rr3ZrewriterrrrrKs z$MultiRuleRewriter.extract_referencesN)r"r#r$r%rrArKrrrrr^sr^)r%re collectionsrtypingrrrZdjango.utils.functionalrcompilerVrNr rLdictrrr'rMrUr^rrrrs     Z8