3 uQh9@s"dZddlmZddlmZddlmZddlmZm Z m Z ddl m Z m Z mZmZddlmZdd lmZmZmZmZmZmZmZmZmZmZmZmZGd d d eZe e e e efd d dZ!ed%ed&ed'ed(ed)ed*eddeddeddedded di Z"e eee d!d"d#Z#d$S)+z( Simple HTML -> Telegram entity parser. )deque)escape) HTMLParser)IterableTupleList) add_surrogate del_surrogatewithin_surrogate strip_text)TLObject) MessageEntityBoldMessageEntityItalicMessageEntityCodeMessageEntityPreMessageEntityEmailMessageEntityUrlMessageEntityTextUrlMessageEntityMentionNameMessageEntityUnderlineMessageEntityStrikeMessageEntityBlockquoteTypeMessageEntitycs4eZdZfddZddZddZddZZS) HTMLToTelegramParsercs0tjd|_g|_i|_t|_t|_dS)N)super__init__textentities_building_entitiesr _open_tags_open_tags_meta)self) __class__|j d }y|d t d d|_ Wnt k rYnXWnt k rt }YnXn|d kr t}d |d<n|dkry |d}Wnt k r6dSX|jdrZ|t dd}t}n(|j|krnt}nt}t||d<d}|jj|jj||r||j kr|ft |jdd||j |<dS)Nstrongbemiudels blockquotecodepreclassz language-rlanguageahrefzmailto:urlr)offsetlength)r! appendleftr"dictrrrrrr lenr2KeyErrorrr startswithrget_starttag_textrrr popleftr)r#tagattrsZ EntityTypeargsr0r5r%r%r&handle_starttags^             z$HTMLToTelegramParser.handle_starttagcCsrt|jdkr|jdnd}|dkr6|jd}|r6|}x(|jjD]\}}|jt|7_qBW|j|7_dS)Nrrr3)r:r!r"r itemsr7r)r#rZ previous_tagr5r?entityr%r%r& handle_dataYs z HTMLToTelegramParser.handle_datac CsPy|jj|jjWntk r,YnX|jj|d}|rL|jj|dS)N)r!r>r" IndexErrorr poprappend)r#r?rDr%r%r& handle_endtages z"HTMLToTelegramParser.handle_endtag)__name__ __module__ __qualname__rrBrErI __classcell__r%r%)r$r&rs < r)htmlreturncCsX|s |gfSt}|jt|t|j|j}|jj|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|jS)N)r6)rDr%r%r&szparse..)key) rfeedr r rrreversesortr )rNparserrr%r%r&parseps rV 
cCsdj|jdfS)Nz-
    
        z{}
)formatr2)e_r%r%r&rPsrPcCsdj|dfS)Nzz)rc)retr%r%r&rPscCsdj|dfS)Nz z)rc)rerfr%r%r&rPscCsdjt|jdfS)Nz z)rcrr5)rdrer%r%r&rPscCsdj|jdfS)Nzz)rcZuser_id)rdrer%r%r&rPs)rrrOc CsN|s|S|st|St|tr$|f}t|}g}x~t|D]r\}}|j}|j|j}tjt |d}|r:t |r|||||}|j |||df|j || |dfq:W|j dddt |}x`|r(|j\} } } xt|| r| d7} qW|d| | t|| |||d}| }qWt|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)Nrrgr%)rfr%r%r&rPszunparse..)rQ)r isinstancer r enumerater6r7ENTITY_TO_FORMATTERgettypecallablerHrTr:rGr r ) rrZ insert_atr*rDr-rd delimiterZnext_escape_boundZatrewhatr%r%r&unparses6     ,rpN)rWrX)rYrZ)r[r\)r]r^)r_r`)rarb)$__doc__ collectionsrrNr html.parserrtypingrrrhelpersr r r r tlr Ztl.typesrrrrrrrrrrrrrstrrVrjrpr%r%r%r&s*    8 \