o 5jO@sUddlmZddlZddlmZmZmZddlmZddlm Z m Z ddl m Z ddl mZddlmZmZdd lmZmZmZmZe ZdlddZGdddeeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZ 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.d/d/eZ(Gd0d1d1eZ)Gd2d3d3eZ*Gd4d5d5eZ+Gd6d7d7eZ,iej-eej-eej-eej-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,iZ.d8e/d9<dmd>d?Z0  dndodHdIZ1dJdKdLdpdPdQZ2dJdRdqdSdTZ3dUdVdrdXdYZ4dZd[d\dsd_d`Z5dtdcddZ6dUdVdudedfZ7dZd[d\dvdgdhZ8dwdjdkZ9dS)x) annotationsN) GeneratorIterableIterator)datetime)AnyClassVar)Self)MessageEntityType) MessageEntityUser)add_surrogateshtml_decorationmarkdown_decorationremove_surrogatesvaluestrreturnintcCst|ddS)Nz utf-16-le)lenencode)rrU/var/www/kodo/Anonymous/send/lib/python3.10/site-packages/aiogram/utils/formatting.pysizeofsrc@seZdZUdZdZded<dZdPd d ZedQddZ dddddRddZ dSd"d#Z dddd$d%dTd+d,Z dd-dUd.d/Z dd-dUd0d1Zdd-dUd2d3Zdd-dUd4d5ZdVd6d7ZdVd8d9ZdWd>d?ZdXdYdBdCZdZdEdFZd[dHdIZd\dJdKZd]dNdOZdS)^Textz Simple text element NzClassVar[str | None]type_body_paramsbodyNodeTypeparamsrrNonecOs||_||_dSNr)selfr r"rrr__init__#s z Text.__init__textrentitieslist[MessageEntity]cCs.|tt||rt|ddddSgdS)NcS|jSr$offset)itemrrr0z$Text.from_entities..key)r'r()_unparse_entitiesr sorted)clsr'r(rrr from_entities+szText.from_entitiesrT_offset_sort_collect_entitiesr7rr8boolr9tuple[str, list[MessageEntity]]c 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: Fr6r,lengthcSr*r$r+)entityrrrr.Yr/zText.render..r0) r isinstancerrrrenderextendrappend_render_entitysort) r%r7r8r9r'r(r,node node_text node_entitiesrrrrA4s.       z Text.renderr,r>r cCs,|jdus Jdtd|j||d|jS)Nz-Node without type can't be rendered as entity)rr,r>r)rr r)r%r,r>rrrrD]szText._render_entity parse_mode)text_key entities_keyreplace_parse_modeparse_mode_keyrJrKrLrMdict[str, Any]cCs(|\}}||||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)rA)r%rJrKrLrM text_valueentities_valueresultrrr as_kwargsas zText.as_kwargs)rLcCs|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)rJrKrLrRr%rLrrras_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_moderJrKrMrLrUrVrrras_poll_question_kwargs  zText.as_poll_question_kwargscCrX)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_moder\rUrVrrras_poll_explanation_kwargss zText.as_poll_explanation_kwargscCrX)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_moder\rUrVrrras_gift_text_kwargsr^zText.as_gift_text_kwargscC|\}}t||S)z5 Render elements tree as HTML markup )rArunparser%r'r(rrras_html  z Text.as_htmlcCrf)z; Render elements tree as MarkdownV2 markup )rArrgrhrrr as_markdownrjzText.as_markdownr%r argskwargscOst||ii|j|Sr$)rr)r%rlrmrrrreplacesz Text.replaceFindentcsrdnd}|fdd|jD}|dd|jD}g}|r)|||r0||||}rAtd|dd}t|jd|d S) Nz, z, c3s.|]}t|tr|jdnt|VqdS)roN)r@ras_pretty_stringrepr.0r-rprr s  z(Text.as_pretty_string..css*|]\}}|dur|d|VqdS)N=r)rtkvrrrrus( z ()) joinrritemsrCtextwrapror__name__)r%rosepr r"rlargs_strrrprrqs    zText.as_pretty_stringothercCs~t|tr |j|jkr |j|jkr t|g||Ri|jSt|tur:t|tr:t|g||Ri|jSt||Sr$)r@rrrr)r%rrrr__add__s " z Text.__add__Iterator[NodeType]ccs|jEdHdSr$)r)r%rrr__iter__sz Text.__iter__cCs|jdd\}}t|S)NF)r9)rAr)r%r'_rrr__len__sz Text.__len__r-slicec Cst|ts d}t||jdus|jdkr |jdur |j|jS|jdur'dn|j}|jdur3t|n|j}||kr>|Sg}d}|jD]5}t|}|} ||7}||krVqE| |kr\ntd|| f} t ||| f} || | } | suqE| | qE|j|S)NzCan only be slicedr) r@r TypeErrorstartstoprnrrmaxminrC) r%r-msgrrnodespositionrF node_sizecurrent_positionabnew_noderrr __getitem__s4      zText.__getitem__r r!r"rrr#)r'rr(r)rr)r7rr8r:r9r:rr;)r,rr>rrr ) rJrrKrrLr:rMrrrN)rLr:rrN)rr)r%r rlrrmrrr )F)ror:rr)rr!rr)rr)rr)r-rrr)r __module__ __qualname____doc__r__annotations__ __slots__r& classmethodr5rArDrRrWr]rbrerirkrnrqrrrrrrrrrs:      )       rc(eZdZdZejZd fdd ZZS) 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!r"rrr#cdt|dkr d}t|t|dtsd}t||dds'd|df}tj|i|dS)Nz$Hashtag can contain only one elementrzHashtag can contain only string#r ValueErrorr@r startswithsuperr&r%r r"r __class__rrr&4 zHashTag.__init__r) rrrrr HASHTAGrr& __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!r"rrr#cr)Nrz$Cashtag can contain only one elementrzCashtag can contain only string$rrrrrr&NrzCashTag.__init__r) rrrrr CASHTAGrr&rrrrrr@rrc@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)rrrrr BOT_COMMANDrrrrrrZs 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)rrrrr URLrrrrrri 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)rrrrr EMAILrrrrrrtrrc@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)rrrrr 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)rrrrr 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)rrrrr 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)rrrrr 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)rrrrr 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)rrrrr 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)rrrrr CODErrrrrrrrcs.eZdZdZejZdddfd d ZZS)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 r!r str | Noner"rrr#ctj|d|i|dS)Nrrr&)r%rr r"rrrr&z Pre.__init__)r r!rrr"rrr#) rrrrr PRErr&rrrrrrsrc(eZdZdZejZd fd d ZZS) 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 r!urlrr"rrr#cr)Nrr)r%rr r"rrrr&rzTextLink.__init__)r r!rrr"rrr#) rrrrr TEXT_LINKrr&rrrrrrrcr) 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 r!userr r"rrr#cr)Nrr)r%rr r"rrrr&rzTextMention.__init__)r r!rr r"rrr#) rrrrr TEXT_MENTIONrr&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 r!custom_emoji_idrr"rrr#cr)Nrr)r%rr r"rrrr&rzCustomEmoji.__init__)r r!rrr"rrr#) rrrrr CUSTOM_EMOJIrr&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)rrrrr 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)rrrrr EXPANDABLE_BLOCKQUOTErrrrrrrrcs*eZdZejZdddfd dZZS)DateTimeN)date_time_formatr r! unix_timeint | datetimerrr"rrr#cs2t|tr t|}tj|||d|dS)N)rr)r@rr timestamprr&)r%rrr r"rrrr&s   zDateTime.__init__) r r!rrrrr"rrr#)rrrr DATE_TIMErr&rrrrrrsrzdict[str | None, type[Text]] NODE_TYPESr?r rr!cGs*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_entityFs rr'bytesr(r)r, int | Noner>Generator[NodeType, None, None]c #sdurd|p t|}t|D]N\}}|jdkrq|jdkr0t||jdV|jd}|jd|jdttfdd||dd}t|gt|||dRVq|krot||VdSdS)Nrrcs|jdpdkS)Nrrr+)er+rrr.gsz#_unparse_entities..rr=) r enumerater,rr>listfilterrr2)r'r(r,r>indexr?r sub_entitiesrr+rr2Us(  " r2ryr<)endrr}rrcGsV|r!g}|ddD] }|||gq ||d|gt|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 N)rBr)rrr}rr-rrras_lineqs  r)rcGs:g}|ddD] }|||gq||dt|S)z^ Wrap each element to separated lines :param items: :param sep: :return: Nr)rBrCr)rr}rr-rrras_lists rz- markerrcstfdd|DS)z~ Wrap elements as marked list :param items: :param marker: line marker, by default is '- ' :return: Text c3s|]}t|VqdSr$rrsrrrrusz!as_marked_list..)r)rr}rrras_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)rtrr-rrrrus"z#as_numbered_list..)rr)rrr}rrras_numbered_lists rtitler cGst|dg|RS)z{ Wrap elements as simple section, section has title and body :param title: :param body: :return: Text ryr)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_sectionsrr1cCstt|dd|S)z Wrap elements pair as key-value line. (:code:`{key}: {value}`) :param key: :param value: :return: Text : )rr)r1rrrr as_key_valuerr)rrrr)r?r rr!rr!)NN) r'rr(r)r,rr>rrr)r}r!rrrrrr)r}r!rrrr)r}r!rrrr)r}r!rrrrrr)rr!r r!rr)rr!r r!rrrr) rr!r r!rrrrrr)r1r!rr!rr): __future__rr~collections.abcrrrrtypingrrtyping_extensionsr aiogram.enumsr aiogram.typesr r aiogram.utils.text_decorationsr rrrr!rrrrrrrrrrrrrrrrrrrrrrrrrr2rrrrrrrrrrrrs