o 5jn6@sddlmZddlmZddlmZddlmZddlmZ ddl m Z m Z m Z mZmZddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZe rXdd lm Z m!Z!ed eeZ"ed Z#Gd d d ee e"Z$dddZ%Gddde$eZ&Gddde$eZ'dS)) annotations)ABC)deepcopy)chain)cycle) TYPE_CHECKINGAnyGenericTypeVarcast) CallbackData) CallbackGameCopyTextButtonInlineKeyboardButtonInlineKeyboardMarkupKeyboardButtonKeyboardButtonPollTypeKeyboardButtonRequestChatKeyboardButtonRequestUsersLoginUrlReplyKeyboardMarkupSwitchInlineQueryChosenChat WebAppInfo) GeneratorIterable ButtonTypeTc@seZdZUdZdZded<dZded<dZded< d?d@ddZe dAddZ dBddZ dCddZ dDddZ dEd d!ZdFd$d%ZdGd&d'ZdHd)d*Zdd+dId.d/Zd0d1dJd4d5ZdKd7d8ZdLd:d;ZdMd=d>ZdS)NKeyboardBuilderz Generic keyboard builder that helps to adjust your markup with defined shape of lines. Works both of InlineKeyboardMarkup and ReplyKeyboardMarkup. rint max_width min_width max_buttonsN button_typetype[ButtonType]markuplist[list[ButtonType]] | NonereturnNonecCsFt|ttfsd|d}t|||_|r||ng}||_dS)Nz Button type z are not allowed here) issubclassrr ValueError _button_type_validate_markup_markup)selfr"r$msgr/S/var/www/kodo/Anonymous/send/lib/python3.10/site-packages/aiogram/utils/keyboard.py__init__+s   zKeyboardBuilder.__init__!Generator[ButtonType, None, None]ccst|EdHdS)zB Get flatten set of all buttons :return: N)r from_iterableexportr-r/r/r0buttons:szKeyboardBuilder.buttonsbuttonrboolcCs8|j}t||s|d|jdt|j}t|dS)zb Check that button item has correct type :param button: :return: z should be type z not T)r* isinstance__name__typer))r-r7allowedr.r/r/r0_validate_buttonCs  z KeyboardBuilder._validate_buttonr6cGstt|j|S)zi Check that all passed button has correct type :param buttons: :return: )allmapr=)r-r6r/r/r0_validate_buttonsPsz!KeyboardBuilder._validate_buttonsrowlist[ButtonType]cCsjt|tsd|d|jjdt|j}t|t||jkr.d|d|jd}t||j|dS)z Check that row of buttons are correct Row can be only list of allowed button types and has length 0 <= n <= 8 :param row: :return: zRow z should be type 'list[z ]' not type z is too long (max width: )T) r9listr*r:r;r)lenrr@)r-rAr.r/r/r0 _validate_rowYs  zKeyboardBuilder._validate_rowlist[list[ButtonType]]cCsrd}t|tsd|jjdt|j}t||D] }|||t|7}q||jkr7d|j}t|dS)z Check that passed markup has correct data structure Markup is list of lists of buttons :param markup: :return: rz!Markup should be type 'list[list[z ]]' not type z.Too much buttons detected Max allowed count - T) r9rDr*r:r;r)rFrEr!)r-r$countr.rAr/r/r0r+ms     z KeyboardBuilder._validate_markupsizercCsTt|ts d}t||t|j|jdvr(d|d|jd|jd}t||S)z[ Validate that passed size is legit :param size: :return: zOnly int sizes are allowedz Row size z is not allowed, range: [z, ])r9rr)ranger r)r-rIr.r/r/r0_validate_sizes zKeyboardBuilder._validate_sizecCs t|jS)aB Export configured markup as list of lists of buttons .. code-block:: python >>> builder = KeyboardBuilder(button_type=InlineKeyboardButton) >>> ... # Add buttons to builder >>> markup = InlineKeyboardMarkup(inline_keyboard=builder.export()) :return: )rr,r5r/r/r0r4s zKeyboardBuilder.exportKeyboardBuilder[ButtonType]cGs|j||}|r1t|d|jkr1|d}|jt|}|d|||d}}|||jdkrQ|rP|d|j||jd}}|t||s8n|t|||_|S)z^ Add one or many buttons to markup. :param buttons: :return: Nr)r@r4rErextendappendrDr,)r-r6r$last_rowposheadrAr/r/r0adds   zKeyboardBuilder.add)widthrV int | NonecsNdur|j||j|jfddtdtD|S)z Add row to markup When too much buttons is passed it will be separated to many rows :param buttons: :param width: :return: Nc3s$|] }t||VqdSN)rD).0rSr6rVr/r0 s z&KeyboardBuilder.row..r)rrMr@r,rPrLrE)r-rVr6r/rZr0rAs   zKeyboardBuilder.rowF)repeatsizesr\c Gs|s|jf}t|j|}|rt|nt|}t|}g}g}|jD]}t||kr4||t|}g}||q!|rA||||_ |S)a Adjust previously added buttons to specific row sizes. By default, when the sum of passed sizes is lower than buttons count the last one size will be used for tail of the markup. If repeat=True is passed - all sizes will be cycled when available more buttons count than all sizes :param sizes: :param repeat: :return: ) rr?rM repeat_all repeat_lastnextr6rErQr,) r-r\r]validated_sizes sizes_iterrIr$rAr7r/r/r0adjusts"       zKeyboardBuilder.adjustkwargscKs:t|d}tr||d<|jdi|}||S)zO Add button to markup :param kwargs: :return: callback_dataNr/)r9getr packr*rU)r-rdrer7r/r/r0_buttons  zKeyboardBuilder._button*InlineKeyboardMarkup | ReplyKeyboardMarkupcKsR|jturtttt|}tdd|i|Stttt|}t|dS)Nkeyboardinline_keyboardr/)r*rr rDr4rrr)r-rdrjrlr/r/r0 as_markups  zKeyboardBuilder.as_markupbuildercCsbt|tsdt|j}t||j|jur'd|jjd|jj}t||j||S)Nz*Only KeyboardBuilder can be attached, not z9Only builders with same button type can be attached, not z and ) r9rr;r:r)r*r,rPr4)r-rnr.r/r/r0attachs  zKeyboardBuilder.attachrX)r"r#r$r%r&r')r&r2)r7rr&r8)r6rr&r8)rArBr&r8)r$rGr&r8)rIrr&r)r&rG)r6rr&rN)r6rrVrWr&rN)r]rr\r8r&rN)rdrr&rN)rdrr&ri)rnrNr&rN)r: __module__ __qualname____doc__r__annotations__r r!r1propertyr6r=r@rFr+rMr4rUrArcrhrmror/r/r/r0r s*            ! ritems Iterable[T]r&Generator[T, None, None]ccsht|}zt|}Wn tyYdSw|Vd} |s0zt|}Wn ty/d}Ynw|Vq)NFT)iterr` StopIteration)ru items_itervaluefinishedr/r/r0r_s$    r_c seZdZUdZdZded<dZded<dZded<d d d d d d d d d d d d d d5d$d%Zd6fd'd( Z d7d8fd,d- Z d9d/d0Z e d:d3d4Z ZS);InlineKeyboardBuilderzK Inline keyboard builder inherits all methods from generic builder rrrJr dr!N) icon_custom_emoji_idstyleurlreweb_app login_urlswitch_inline_query switch_inline_query_current_chatswitch_inline_query_chosen_chat copy_text callback_gamepaytextstrr str | Nonerrrestr | CallbackData | NonerWebAppInfo | NonerLoginUrl | Nonerrr"SwitchInlineQueryChosenChat | NonerCopyTextButton | NonerCallbackGame | Noner bool | Nonerdrr&c Ks2tt|jd||||||||| | | | | d |S)N) rrrrrerrrrrrrrr/)r r}rh)r-rrrrrerrrrrrrrrdr/r/r0r7.s(zInlineKeyboardBuilder.buttonrc tttjdi|S)z!Construct an InlineKeyboardMarkupNr/)r rsuperrmr-rd __class__r/r0rmTzInlineKeyboardBuilder.as_markupr$'list[list[InlineKeyboardButton]] | Noner'ctjt|ddSN)r"r$)rr1rr-r$rr/r0r1XzInlineKeyboardBuilder.__init__r-cCt|dSzQ Make full copy of current builder with markup :return: r$)r}r4r5r/r/r0copy[zInlineKeyboardBuilder.copyclstype[InlineKeyboardBuilder]cC ||jdSz^ Create builder from existing markup :param markup: :return: rrkrr$r/r/r0 from_markupcs z!InlineKeyboardBuilder.from_markup)rrrrrrrrrerrrrrrrrrrrrrrrrrrdrr&r})rdrr&rrX)r$rr&r')r-r}r&r})rrr$rr&r}r:rprqrrrrsr r!r7rmr1r classmethodr __classcell__r/r/rr0r}%s.    & r}c seZdZUdZdZded<dZded<dZded<d d d d d d d d d d-ddZd.fd!d" Z d/d0fd&d' Z d1d)d*Z e d2d+d,Z ZS)3ReplyKeyboardBuilderzJ Reply keyboard builder inherits all methods from generic builder rrrJr i,r!N)rr request_users request_chatrequest_contactrequest_location request_pollrrrrrrr!KeyboardButtonRequestUsers | Noner KeyboardButtonRequestChat | NonerrrrKeyboardButtonPollType | Nonerrrdrr&c Ks*tt|jd||||||||| d | S)N) rrrrrrrrrr/)r rrh) r-rrrrrrrrrrdr/r/r0r7zs  zReplyKeyboardBuilder.buttonrc r)zConstruct a ReplyKeyboardMarkupNr/)r rrrmrrr/r0rmrzReplyKeyboardBuilder.as_markupr$!list[list[KeyboardButton]] | Noner'crr)rr1rrrr/r0r1rzReplyKeyboardBuilder.__init__r-cCrr)rr4r5r/r/r0rrzReplyKeyboardBuilder.copycCrr)rjrr/r/r0rs z ReplyKeyboardBuilder.from_markup)rrrrrrrrrrrrrrrrrrrdrr&r)rdrr&rrX)r$rr&r')r-rr&r)r$rr&rrr/r/rr0rqs&     rN)rurvr&rw)( __future__rabcrrr itertoolsrrr^typingrrr r r aiogram.filters.callback_datar aiogram.typesr rrrrrrrrrrrcollections.abcrrrrrr_r}rr/r/r/r0s"     8  tL