o FhjM@sUddlZddlmZmZmZmZmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZmZddlmZmZmZmZeZdedefd d ZGd d d eeZGd ddeZGdddeZGdddeZGdddeZ GdddeZ!GdddeZ"GdddeZ#GdddeZ$GdddeZ%Gdd d eZ&Gd!d"d"eZ'Gd#d$d$eZ(Gd%d&d&eZ)Gd'd(d(eZ*Gd)d*d*eZ+Gd+d,d,eZ,Gd-d.d.eZ-Gd/d0d0eZ.iej/eej/eej/eej/ee j/e e!j/e!e"j/e"e#j/e#e$j/e$e%j/e%e&j/e&e'j/e'e(j/e(e)j/e)e*j/e*e+j/e+e,j/e,e-j/e-e.j/e.iZ0ee ee efe1d1<d2ed3edefd4d5Z2  d^d6e3d7eed8e ed9e edeeddff d:d;Z4dd?ed@edAedefdBdCZ5dZ$d?e%ddfd@dAZ&dS)CTextz Simple text element Ntype_body_paramsbodyparamsrcOs||_||_dSNrselfr!r"rrr__init__*s z Text.__init__textentitiescCs.|tt||rt|ddddSgdS)NcS|jSr#offset)itemrrr7z$Text.from_entities..key)r'r()_unparse_entitiesrsorted)clsr'r(rrr from_entities2szText.from_entitiesrT_offset_sort_collect_entitiesr6r7r8c Csd}g}|}|jD]1}t|tst|}||7}|t|7}q |j|d|d\}} ||7}|t|7}|r:|| q |rL|jrL||j |||d|rX|rX|j ddd||fS)zS Render elements tree as text with entities list :return: Fr5r+lengthcSr)r#r*)entityrrrr-`r.zText.render..r/) r isinstancerstrrrenderextendrappend_render_entitysort) r%r6r7r8r'r(r+node node_text node_entitiesrrrr?;s.       z Text.renderr+r;cCs,|jdus Jdtd|j||d|jS)Nz-Node without type can't be rendered as entity)rr+r;r)rrr )r%r+r;rrrrBdszText._render_entity parse_mode)text_key entities_keyreplace_parse_modeparse_mode_keyrHrIrJrKcCs(|\}}||||i}|rd||<|S)af Render element tree as keyword arguments for usage in an API call, for example: .. code-block:: python entities = Text(...) await message.answer(**entities.as_kwargs()) :param text_key: :param entities_key: :param replace_parse_mode: :param parse_mode_key: :return: N)r?)r%rHrIrJrK text_valueentities_valueresultrrr as_kwargshs zText.as_kwargs)rJcCs|jdd|dS)af Shortcut for :meth:`as_kwargs` for usage with API calls that take ``caption`` as a parameter. .. code-block:: python entities = Text(...) await message.answer_photo(**entities.as_caption_kwargs(), photo=phot) :param replace_parse_mode: Will be passed to :meth:`as_kwargs`. :return: captioncaption_entities)rHrIrJrOr%rJrrras_caption_kwargss  zText.as_caption_kwargscC|jddd|dS)at Shortcut for :meth:`as_kwargs` for usage with method :class:`aiogram.methods.send_poll.SendPoll`. .. code-block:: python entities = Text(...) await message.answer_poll(**entities.as_poll_question_kwargs(), options=options) :param replace_parse_mode: Will be passed to :meth:`as_kwargs`. :return: questionquestion_entitiesquestion_parse_moderHrIrKrJrRrSrrras_poll_question_kwargs  zText.as_poll_question_kwargscCrU)a' Shortcut for :meth:`as_kwargs` for usage with method :class:`aiogram.methods.send_poll.SendPoll`. .. code-block:: python question_entities = Text(...) explanation_entities = Text(...) await message.answer_poll( **question_entities.as_poll_question_kwargs(), options=options, **explanation_entities.as_poll_explanation_kwargs(), ) :param replace_parse_mode: Will be passed to :meth:`as_kwargs`. :return: explanationexplanation_entitiesexplanation_parse_moderYrRrSrrras_poll_explanation_kwargss zText.as_poll_explanation_kwargscCrU)a{ Shortcut for :meth:`as_kwargs` for usage with method :class:`aiogram.methods.send_gift.SendGift`. .. code-block:: python entities = Text(...) await bot.send_gift(gift_id=gift_id, user_id=user_id, **entities.as_gift_text_kwargs()) :param replace_parse_mode: Will be passed to :meth:`as_kwargs`. :return: r' text_entitiestext_parse_moderYrRrSrrras_gift_text_kwargsr[zText.as_gift_text_kwargscC|\}}t||S)z5 Render elements tree as HTML markup )r?runparser%r'r(rrras_html  z Text.as_htmlcCrc)z; Render elements tree as MarkdownV2 markup )r?rrdrerrr as_markdownrgzText.as_markdownr%argskwargscOst||ii|j|Sr#)rr )r%rirjrrrreplacesz Text.replaceFindentcsrdnd}|fdd|jD}|dd|jD}g}|r)|||r0||||}rAtd|dd}t|jd|d S) Nz, z, c3s.|]}t|tr|jdnt|VqdS)rlN)r=ras_pretty_stringrepr.0r,rmrr s  z(Text.as_pretty_string..css*|]\}}|dur|d|VqdS)N=r)rqkvrrrrrs( z ()) joinrr itemsrAtextwraprlr__name__)r%rlsepr!r"riargs_strrrmrrns    zText.as_pretty_stringothercCs~t|tr |j|jkr |j|jkr t|g||Ri|jSt|tur:t|tr:t|g||Ri|jSt||Sr#)r=rrr r>)r%rrrr__add__s " z Text.__add__ccs|jEdHdSr#)r)r%rrr__iter__sz Text.__iter__cCs|jdd\}}t|S)NF)r8)r?r)r%r'_rrr__len__ sz Text.__len__r,c Cst|ts td|jdus|jdkr|jdur|j|jS|jdur%dn|j}|jdur1t|n|j}||kr<|Sg}d}|jD]5}t|}|}||7}||krTqC||krZntd||f} t |||f} || | } | ssqC| | qC|j|S)NzCan only be slicedr) r=slice TypeErrorstartstoprkrrmaxminrA) r%r,rrnodespositionrD node_sizecurrent_positionabnew_noderrr __getitem__ s2      zText.__getitem__)F)'r| __module__ __qualname____doc__rrr r>__annotations__ __slots__NodeTyperr& classmethodrrr4intboolr r?rBrrOrTrZr_rbrfrhr rkrnrrrrrrrrrrr!sj    )     rc4eZdZdZejZdededdffdd Z Z S)HashTaga Hashtag element. .. warning:: The value should always start with '#' symbol Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.HASHTAG` r!r"rNc\t|dkr tdt|dtstd|dds#d|df}tj|i|dS)Nz$Hashtag can contain only one elementrzHashtag can contain only string#r ValueErrorr=r> startswithsuperr&r$ __class__rrr&: zHashTag.__init__) r|rrrr HASHTAGrrrr& __classcell__rrrrr, "rcr)CashTaga Cashtag element. .. warning:: The value should always start with '$' symbol Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.CASHTAG` r!r"rNcr)Nrz$Cashtag can contain only one elementrzCashtag can contain only string$rr$rrrr&RrzCashTag.__init__) r|rrrr CASHTAGrrrr&rrrrrrDrrc@eZdZdZejZdS) BotCommanda Bot command element. .. warning:: The value should always start with '/' symbol Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.BOT_COMMAND` N)r|rrrr BOT_COMMANDrrrrrr\s rc@r)Urlz Url element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.URL` N)r|rrrr URLrrrrrrk rc@r)Emailz Email element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.EMAIL` N)r|rrrr EMAILrrrrrrvrrc@r) PhoneNumberz Phone number element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.PHONE_NUMBER` N)r|rrrr PHONE_NUMBERrrrrrrrrc@r)Boldz Bold element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.BOLD` N)r|rrrr BOLDrrrrrrrrc@r)Italicz Italic element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.ITALIC` N)r|rrrr ITALICrrrrrrrrc@r) Underlinez Underline element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.UNDERLINE` N)r|rrrr UNDERLINErrrrrrrrc@r) Strikethroughz Strikethrough element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.STRIKETHROUGH` N)r|rrrr STRIKETHROUGHrrrrrrrrc@r)Spoilerz Spoiler element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.SPOILER` N)r|rrrr SPOILERrrrrrrrrc@r)Codez Code element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.CODE` N)r|rrrr CODErrrrrrrrc sBeZdZdZejZdddedee de ddffdd Z Z S) Prez Pre element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.PRE` N)languager!rr"rctj|d|i|dS)Nrrr&)r%rr!r"rrrr&z Pre.__init__) r|rrrr PRErrr r>rr&rrrrrrs0rc8eZdZdZejZdedede ddffdd Z Z S) TextLinkz Text link element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.TEXT_LINK` r!urlr"rNcr)Nrr)r%rr!r"rrrr&rzTextLink.__init__) r|rrrr TEXT_LINKrrr>rr&rrrrrr&rcr) TextMentionz Text mention element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.TEXT_MENTION` r!userr"rNcr)Nrr)r%rr!r"rrrr&rzTextMention.__init__) r|rrrr TEXT_MENTIONrrrrr&rrrrrrrrcr) CustomEmojiz Custom emoji element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.CUSTOM_EMOJI` r!custom_emoji_idr"rNcr)Nrr)r%rr!r"rrrr&rzCustomEmoji.__init__) r|rrrr CUSTOM_EMOJIrrr>rr&rrrrrrrrc@r) BlockQuotez Block quote element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.BLOCKQUOTE` N)r|rrrr BLOCKQUOTErrrrrrrrc@r)ExpandableBlockQuotez Expandable block quote element. Will be wrapped into :obj:`aiogram.types.message_entity.MessageEntity` with type :obj:`aiogram.enums.message_entity_type.MessageEntityType.EXPANDABLE_BLOCKQUOTE` N)r|rrrr EXPANDABLE_BLOCKQUOTErrrrrrrr NODE_TYPESr<rcGs*t|jt}||i|jhdddS)zW Apply single entity to text :param entity: :param text: :return: >rr;r+F)excludewarnings)rgetrr model_dump)r<r node_typerrr _apply_entity3s rr'r(r+r;c #sdurd|p t|}t|D]N\}}|jdkrq|jdkr0t||jdV|jd}|jd|jdttfdd||dd}t|gt|||dRVq|krot||VdSdS)Nrrcs|jdpdkS)Nrrr*)er*rrr-Ssz#_unparse_entities..rr:) r enumerater+rr;listfilterrr1)r'r(r+r;indexr<r sub_entitiesrr*rr1As(  " r1rvr9)endr}rzrr}cGs\|r$g}|ddD] }|||gq ||d||t|Sg||}t|S)z Wrap multiple nodes into line with :code:`\\n` at the end of line. :param items: Text or Any :param end: ending of the line, by default is :code:`\\n` :param sep: separator between items, by default is empty string :return: Text Nr@rAr)rr}rzrr,rrras_line]s   r)r}cGs:g}|ddD] }|||gq||dt|S)z^ Wrap each element to separated lines :param items: :param sep: :return: Nrr)r}rzrr,rrras_listqs rz- markerrcstfdd|DS)z~ Wrap elements as marked list :param items: :param marker: line marker, by default is '- ' :return: Text c3s|]}t|VqdSr#rrprrrrrsz!as_marked_list..)r)rrzrrras_marked_listsrrz{}. rfmtrrcstfddt||DS)z Wrap elements as numbered list :param items: :param start: initial number, by default 1 :param fmt: number format, by default '{}. ' :return: Text c3s$|] \}}t||VqdSr#)rformat)rqrr,rrrrrs"z#as_numbered_list..)rr)rrrzrrras_numbered_lists rtitler!cGst|dg|RS)z{ Wrap elements as simple section, section has title and body :param title: :param body: :return: Text rvr)rr!rrr as_sectionrcGst|t|d|iS)zw Wrap elements as section with marked list :param title: :param body: :param marker: :return: r)rr)rrr!rrras_marked_sections rcGst|t|||dS)z Wrap elements as section with numbered list :param title: :param body: :param start: :param fmt: :return: r)rr)rrrr!rrras_numbered_sectionsrr0cCstt|dd|S)z Wrap elements pair as key-value line. (:code:`{key}: {value}`) :param key: :param value: :return: Text : )rr)r0rrrr as_key_valuerr)NN)=r{typingrrrrrrrr r r typing_extensionsr aiogram.enumsr aiogram.typesrraiogram.utils.text_decorationsrrrrrr>rrrrrrrrrrrrrrrrrrrrrrrrrbytesr1rrrrrrrrrrrrs 0                    ""