U h9@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__|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)Nstrongbemiudels blockquotecodepreclassz language-rlanguageahrefzmailto:urlr)offsetlength)r! appendleftr"dictrrrrrr lenr3KeyErrorrr startswithrget_starttag_textrrr popleftr)r#tagattrsZ EntityTypeargsr1r6r&r&r'handle_starttagsb             z$HTMLToTelegramParser.handle_starttagcCsnt|jdkr|jdnd}|dkr6|jd}|r6|}|jD]\}}|jt|7_q@|j|7_dS)Nrrr4)r;r!r"r itemsr8r)r#rZ previous_tagr6r@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@rEr&r&r' handle_endtages z"HTMLToTelegramParser.handle_endtag)__name__ __module__ __qualname__rrCrFrK __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|jSrG)r7)rEr&r&r'zparse..key) rfeedr r rrreversesortr )rPparserrr&r&r'parseps rZ)zz )zz)zz)zz)zz)z
z
cCsd|jdfS)Nz-
    
        z{}
)formatr3e_r&r&r'rRs rRcCsd|dfS)Nzr[r^tr&r&r'rRrScCsd|dfSNz r_r`rar&r&r'rRrScCsdt|jdfSrc)r[rr6r\r&r&r'rRrScCsd|jdfS)Nzr_)r[Zuser_idr\r&r&r'rRrS)rrrQc 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)Nrrdr&)rbr&r&r'rRrSzunparse..rT)r isinstancer r enumerater7r8ENTITY_TO_FORMATTERgettypecallablerJrXr;rIr r ) rrZ insert_atr+rEr.r] delimiterZnext_escape_boundatr^whatr&r&r'unparses6     ,rnN)$__doc__ collectionsrrPr html.parserrtypingrrrZhelpersr r r r tlr Ztl.typesrrrrrrrrrrrrrstrrZrgrnr&r&r&r'sB    8 \