U nh9@s"dZddlmZddlmZddlmZddlmZm Z m Z ddl m Z m Z mZmZddlmZdd lmZmZmZmZmZmZmZmZmZmZmZmZGd d d eZe e e e efd d dZ!edededededededdeddeddeddeddi Z"e eee dddZ#dS)z( Simple HTML -> Telegram entity parser. )deque)escape) HTMLParser)IterableTupleList) add_surrogate del_surrogatewithin_surrogate strip_text)TLObject) MessageEntityBoldMessageEntityItalicMessageEntityCodeMessageEntityPreMessageEntityEmailMessageEntityUrlMessageEntityTextUrlMessageEntityMentionNameMessageEntityUnderlineMessageEntityStrikeMessageEntityBlockquoteTypeMessageEntitycs4eZdZfddZddZddZddZZS) HTMLToTelegramParsercs0td|_g|_i|_t|_t|_dS)N)super__init__textentities_building_entitiesr _open_tags_open_tags_meta)self __class__b/var/www/html/peyman_registration/venvv2/lib/python3.8/site-packages/telethonv2/extensions/html.pyrs  zHTMLToTelegramParser.__init__cCs|j||jdt|}d}i}|dks8|dkr@t}nZ|dksP|dkrXt}nB|dkrht}n2|dksx|dkrt}n|dkrt}n |d krz>|j d }z|d t d d|_ Wnt k rYnXWnt k rt }YnXn|d kr t}d |d<n|dkrz |d}Wnt k r8YdSX|dr\|t dd}t}n(||krpt}nt}t||d<d}|j|j||r||j kr|ft |jdd||j |<dS)NZstrongbZemiudelsZ blockquotecodepreclassz language-rlanguageaZhrefzmailto:urlr)offsetlength)r! appendleftr"dictrrrrrr lenr0KeyErrorrr startswithrZget_starttag_textrrr popleftr)r#tagattrs EntityTypeargsr.r2r&r&r'handle_starttagsb             z$HTMLToTelegramParser.handle_starttagcCsnt|jdkr|jdnd}|dkr6|jd}|r6|}|jD]\}}|jt|7_q@|j|7_dS)Nrrr1)r7r!r"r itemsr4r)r#rZ previous_tagr2r;entityr&r&r' handle_dataYs z HTMLToTelegramParser.handle_datacCsPz|j|jWntk r,YnX|j|d}|rL|j|dSN)r!r:r" IndexErrorr poprappend)r#r;rAr&r&r' handle_endtages z"HTMLToTelegramParser.handle_endtag)__name__ __module__ __qualname__rr?rBrG __classcell__r&r&r$r'rs < r)htmlreturncCsX|s |gfSt}|t|t|j|j}|j|jjdddt||jfS)a Parses the given HTML message and returns its stripped representation plus a list of the MessageEntity's that were found. :param html: the message with HTML to be parsed. :return: a tuple consisting of (clean message, [message entities]). cSs|jSrC)r3)rAr&r&r'zparse..key) rfeedr r rrreversesortr )rLparserrr&r&r'parseps rV)zz )zz)zz)zz)zz)z
z
cCsd|jdfS)Nz-
    
        z{}
)formatr0e_r&r&r'rNs rNcCsd|dfS)NzrWrZtr&r&r'rNrOcCsd|dfSNz r[r\r]r&r&r'rNrOcCsdt|jdfSr_)rWrr2rXr&r&r'rNrOcCsd|jdfS)Nzr[)rWuser_idrXr&r&r'rNrO)rrrMc CsB|s|S|st|St|tr$|f}t|}g}t|D]r\}}|j}|j|j}tt |d}|r8t |r|||||}| |||df| || |dfq8|j dddt |}|r|\} } } t|| r| d7} q|d| | t|| |||d}| }qt|d|||d}t|S)a= Performs the reverse operation to .parse(), effectively returning HTML given a normal text and its MessageEntity's. :param text: the text to be reconverted into HTML. :param entities: the MessageEntity's applied to the text. :return: a HTML representation of the combination of both inputs. NrcSs|d|dfS)Nrrar&)r^r&r&r'rNrOzunparse..rP)r isinstancer r enumerater3r4ENTITY_TO_FORMATTERgettypecallablerFrTr7rEr r ) rr insert_atr)rAr,rY delimiterZnext_escape_boundatrZwhatr&r&r'unparses6     ,rlN)$__doc__ collectionsrrLrZ html.parserrtypingrrrhelpersr r r r tlr tl.typesrrrrrrrrrrrrrstrrVrdrlr&r&r&r'sB    8 \