o HEDi@sNddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZddlmZddlm Z ddl!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)dd l*m+Z+d d l,m-Z-d d l.m/Z/d d l0m1Z1m2Z2d dl3m4Z4m5Z5e6e7Z8dZ9dddZ:Gddde5e4Z;dS)N)sentinel)renamed_argument)CommandContentTypeFilterExceptionsFilterFiltersFactoryHashTagRegexpRegexpCommandsFilter StateFilterTextIDFilter AdminFilter IsReplyFilterForwardedMessageFilterIsSenderContactChatTypeFilterMediaGroupFilterAbstractFilter)Handler)MiddlewareManager) BaseStorageDELTADisabledStorageEXCEEDED_COUNT FSMContext LAST_CALL RATE_LIMITRESULT) BaseResponse)types)Bot)TelegramAPIError Throttled)ContextInstanceMixin DataMixin皙?xasyncio.AbstractEventLoopcCs*t|tjsJdtjdt|dS)Nz#Loop must be the implementation of z, not ) isinstanceasyncioAbstractEventLooptype)r)r/l/var/www/www-root/data/www/ovozai.pdev.uz/venv/lib/python3.10/site-packages/aiogram/dispatcher/dispatcher.py _ensure_loops r1c @seZdZdZdddeddfdejedefddZ e deje j fd d Z e dd d ZddZddZddZddefddZdejfddZddefddZddZ     ! dded"ed#ejejefd$d%Zddefd&d'Zd(d)Zd*d+Zd,d-Zdddddd.d/d0Z dddddd.d1d2Z!dddddd.d3d4Z"dddddd.d5d6Z#dddddd.d7d8Z$dddddd.d9d:Z%dddddd.d;d<Z&dddddd.d=d>Z'ddd?d@dAZ(ddd?dBdCZ)ddd?dDdEZ*ddd?dFdGZ+ddd?dHdIZ,ddd?dJdKZ-ddd?dLdMZ.ddd?dNdOZ/ddd?dPdQZ0ddd?dRdSZ1ddTdUdVZ2ddTdWdXZ3ddTdYdZZ4ddTd[d\Z5ddTd]ej6d^ejeddfd_d`Z7ddTdadbZ8ddTd]ej6d^ejeddfdcddZ9ddTdedfZ:ddTd]ej6d^ejeddfdgdhZ;ddTdidjZdddpdqej?eedfdrej?eedfde@fdsdtZAeBdrdudvdwdxeBdqdydvdzdxddddd{defd|d}ZCeBdrdudvdwdxeBdqdydvdzdxdd~dZDeBdrdudvdwdxeBdqdydvdzdxdddZEddZFddeGfddZH   ddejej6fddZI   dd]ej?ej6eJfdejej6dejejeKdejejLeKfddZMd]ej?ej6eJffddZNddZOdS) Dispatcherz Simple Updates dispatcher It will process incoming updates: messages, edited messages, channel posts, edited channel posts, inline queries, chosen inline results, callback queries, shipping queries, pre-checkout queries. NFstoragerun_tasks_by_defaultcCst|tstdt|jd|durt}|durt|}||_||_||_ ||_ ||_ ||_ ||_ t|dd|_t|dd|_t|dd|_t|dd|_t|dd|_t|d d|_t|d d|_t|d d|_t|d d|_t|d d|_t|dd|_t|dd|_t|dd|_t|dd|_t|dd|_t|ddd|_t||_ |j!|j"d|_#d|_$d|_%|&dS)Nz0Argument 'bot' must be an instance of Bot, not ''update)middleware_keymessageedited_message channel_postedited_channel_post inline_querychosen_inline_resultcallback_queryshipping_querypre_checkout_querypoll poll_answermy_chat_member chat_memberchat_join_requestFerror)oncer7T)'r+r# TypeErrorr.__name__rrbot _main_loopr3r4throttling_rate_limitno_throttle_errorfilters_factoryrupdates_handlermessage_handlersedited_message_handlerschannel_post_handlersedited_channel_post_handlersinline_query_handlerschosen_inline_result_handlerscallback_query_handlersshipping_query_handlerspre_checkout_query_handlers poll_handlerspoll_answer_handlersmy_chat_member_handlerschat_member_handlerschat_join_request_handlerserrors_handlersr middlewareregisterprocess_update_polling_closed_dispatcher_close_waiter_setup_filters)selfrJloopr3r4rLrMrNr/r/r0__init__-sF   zDispatcher.__init__returncC|jSN)rKrfr/r/r0rg^zDispatcher.loopasyncio.FuturecCs|jdur t|_|jSrk)rdr,get_event_loop create_futurerlr/r/r0 _close_waiterfs zDispatcher._close_waiterc Cs|j}|jt|j|j|jgd|jt|j|j|j |j gdf|jt |j|jgd|jt |j|j|j |j |j |j|jgd|jt|j|j|j |j gd|jt|j|j|j |j |j |j|jgd|jt|j|jgd|jt|jgd|jt|j|j|j |j |j |j|j|jgd|jt|j|j|j |j |j |j|j|j|jg d|jt|j|j|j |j gd|jt|j|j|j |j gd|jt|j|j |j |j gd|jt|j|j|j |j |j |j|j|jgd|jt|j|j |j |j gddS)N)exclude_event_handlers)event_handlers)rNbindr r^rYrZrrPrQrRrSrr rVrTr r r rrr\r]rr[rrrrr)rfrNr/r/r0rels       zDispatcher._setup_filterscCs |dSrk) stop_pollingrlr/r/r0__del__s zDispatcher.__del__cs|jjddIdHdS)z You can skip old incoming updates from queue. This method is not recommended for using in production. Note that the webhook will be deleted! T)drop_pending_updatesN)rJdelete_webhookrlr/r/r0 skip_updatesszDispatcher.skip_updatesTfastcsR|rfdd|D}tj|IdHSg}|D]}|j|IdHq|S)zh Process list of updates :param updates: :param fast: :return: csg|]}j|qSr/)rOnotify).0r6rlr/r0 sz.Dispatcher.process_updates..N)r,gatherappendrOr{)rfupdatesrztasksresultsr6r/rlr0process_updatesszDispatcher.process_updatesr6c stj|z|jr.tj|jtj|jjtj|jj|j |jIdHWS|j rStj|j tj|j jtj|j j|j |j IdHWS|j rptj|j tj|j j|j |j IdHWS|jrtj|jtj|jj|j |jIdHWS|jrtj|jtj|jj|j |jIdHWS|jrtj|jtj|jj|j |jIdHWS|jrtj|j|jjrtj|jjjtj|jj|j |jIdHWS|jrtj|jtj|jj|j |jIdHWS|jr-tj|jtj|jj|j |jIdHWS|j rCtj!|j |j" |j IdHWS|j#ratj$|j#tj|j#j%|j& |j#IdHWS|j'rtj(|j'tj|j'j|j) |j'IdHWS|j*rtj(|j*tj|j*j|j+ |j*IdHWS|j,rtj-|j,tj|j,jtj|j,j|j. |j,IdHWSWdSt/y}z|j0 ||IdH}|r|WYd}~Sd}~ww)zW Process single update object :param update: :return: N)1r"Update set_currentr8MessageUser from_userChatchatrPr{r9rQr:rRr;rSr< InlineQueryrTr=ChosenInlineResultrUr> CallbackQueryrVr? ShippingQueryrWr@PreCheckoutQueryrXrAPollrYrB PollAnsweruserrZrCChatMemberUpdatedr[rDr\rEChatJoinRequestr] Exceptionr^)rfr6eerrr/r/r0ras zDispatcher.process_updatecs0|r|jIdH}|jsdS|jIdHS)z] Reset webhook :param check: check before deleting :return: NF)rJget_webhook_infourlrx)rfcheckwhr/r/r0 reset_webhook>s zDispatcher.reset_webhookcCs t|Srk)r, create_task)rfcoror/r/r0_loop_create_taskLs zDispatcher._loop_create_taskr( error_sleepallowed_updatesc s|jrtdtdt|t|j|dur%|jddIdH|r0|jddIdHd|_d}z|jj } | t urN|durNt j | j |pJdd} nd} |jrz%|j| |jj||||d IdH} Wdn1srwYWn(tjyYnety} ztd t|IdHWYd} ~ qPd} ~ ww| rtd t| d | d jd}t|| ||rt|IdH|jsSW|jdtddSW|jdtddSW|jdtddS|jdtdw)z Start long-polling :param timeout: :param relax: :param limit: :param reset_webhook: :param fast: :param error_sleep: :param allowed_updates: :return: zPolling already startedzStart polling.NF)rTr)total)limitoffsettimeoutrz&Cause exception while getting updates.z Received z updates.zPolling is stopped.)rb RuntimeErrorloginfor2rr#rJrrraiohttp ClientTimeoutrrequest_timeout get_updatesr,CancelledErrorr exceptionsleepdebuglen update_idr_process_polling_updatesrq set_resultwarning) rfrrelaxrrrzrrrcurrent_request_timeoutrrrr/r/r0 start_pollingOsj         zDispatcher.start_pollingcsg}tj|||IdHD]}|D]}t|tsq|||jqq|rAzt j |WdSt y@t dYdSwdS)z| Process updates received from long-polling. :param updates: list of updates. :param fast: Nz)Cause exception while processing updates.) itertoolschain from_iterablerr+r rexecute_responserJr,r~r$rr)rfrrz need_to_call responsesresponser/r/r0rs  z#Dispatcher._process_polling_updatescCs,t|dr|jrtdd|_dSdSdS)z? Break long-polling process. :return: rbzStop polling...FN)hasattrrbrrrlr/r/r0rus  zDispatcher.stop_pollingcst|jIdHdS)zF Wait for the long-polling to close :return: N)r,shieldrqrlr/r/r0 wait_closedszDispatcher.wait_closedcCrj)z? Check if polling is enabled :return: )rbrlr/r/r0 is_pollingrmzDispatcher.is_pollingcommandsregexp content_typesstaterun_taskc O@|jj|jg|R||||d|} |j|||| dS)a Register handler for message .. code-block:: python3 # This handler works only if state is None (by default). dp.register_message_handler(cmd_start, commands=['start', 'about']) dp.register_message_handler(entry_point, commands=['setup']) # This handler works only if current state is "first_step" dp.register_message_handler(step_handler_1, state="first_step") # If you want to handle all states by one handler, use `state="*"`. dp.register_message_handler(cancel_handler, commands=['cancel'], state="*") dp.register_message_handler(cancel_handler, lambda msg: msg.text.lower() == 'cancel', state="*") :param callback: :param commands: list of commands :param regexp: REGEXP :param content_types: List of content types. :param custom_filters: list of custom filters :param kwargs: :param state: :return: decorated function rrrrN)rNresolverPr`_wrap_async_task rfcallbackrrrrrcustom_filterskwargs filters_setr/r/r0register_message_handlers z#Dispatcher.register_message_handlerc fdd}|S)ab Decorator for message handler Examples: Simple commands handler: .. code-block:: python3 @dp.message_handler(commands=['start', 'welcome', 'about']) async def cmd_handler(message: types.Message): Filter messages by regular expression: .. code-block:: python3 @dp.message_handler(regexp='^[a-z]+-[0-9]+') async def msg_handler(message: types.Message): Filter messages by command regular expression: .. code-block:: python3 @dp.message_handler(filters.RegexpCommandsFilter(regexp_commands=['item_([0-9]*)'])) async def send_welcome(message: types.Message): Filter by content type: .. code-block:: python3 @dp.message_handler(content_types=ContentType.PHOTO | ContentType.DOCUMENT) async def audio_handler(message: types.Message): Filter by custom function: .. code-block:: python3 @dp.message_handler(lambda message: message.text and 'hello' in message.text.lower()) async def text_handler(message: types.Message): Use multiple filters: .. code-block:: python3 @dp.message_handler(commands=['command'], content_types=ContentType.TEXT) async def text_handler(message: types.Message): Register multiple filters set for one handler: .. code-block:: python3 @dp.message_handler(commands=['command']) @dp.message_handler(lambda message: demojize(message.text) == ':new_moon_with_face:') async def text_handler(message: types.Message): This handler will be called if the message starts with '/command' OR is some emoji By default content_type is :class:`ContentType.TEXT` :param commands: list of commands :param regexp: REGEXP :param content_types: List of content types. :param custom_filters: list of custom filters :param kwargs: :param state: :param run_task: run callback in task (no wait results) :return: decorated function c(j|gRd|SNr)rrrrrrrrrfrr/r0 decorator/sz-Dispatcher.message_handler..decoratorr/ rfrrrrrrrrr/rr0message_handlersGzDispatcher.message_handlerc Or)a Register handler for edited message :param callback: :param commands: list of commands :param regexp: REGEXP :param content_types: List of content types. :param state: :param custom_filters: list of custom filters :param run_task: run callback in task (no wait results) :param kwargs: :return: decorated function rN)rNrrQr`rrr/r/r0register_edited_message_handler7 z*Dispatcher.register_edited_message_handlerc r)aZ Decorator for edited message handler You can use combination of different handlers .. code-block:: python3 @dp.message_handler() @dp.edited_message_handler() async def msg_handler(message: types.Message): :param commands: list of commands :param regexp: REGEXP :param content_types: List of content types. :param state: :param custom_filters: list of custom filters :param run_task: run callback in task (no wait results) :param kwargs: :return: decorated function crr)rrrr/r0rf z4Dispatcher.edited_message_handler..decoratorr/rr/rr0edited_message_handlerOsz!Dispatcher.edited_message_handlerc Or)a Register handler for channel post :param callback: :param commands: list of commands :param regexp: REGEXP :param content_types: List of content types. :param state: :param custom_filters: list of custom filters :param run_task: run callback in task (no wait results) :param kwargs: :return: decorated function rN)rNrrRr`rrr/r/r0register_channel_post_handlermrz(Dispatcher.register_channel_post_handlerc r)ay Decorator for channel post handler :param commands: list of commands :param regexp: REGEXP :param content_types: List of content types. :param state: :param custom_filters: list of custom filters :param run_task: run callback in task (no wait results) :param kwargs: :return: decorated function crr)rrrr/r0rrz2Dispatcher.channel_post_handler..decoratorr/rr/rr0channel_post_handlerszDispatcher.channel_post_handlerc Os@|jj|jg|R||||d|} |j|||| dS)a Register handler for edited channel post :param callback: :param commands: list of commands :param regexp: REGEXP :param content_types: List of content types. :param state: :param custom_filters: list of custom filters :param run_task: run callback in task (no wait results) :param kwargs: :return: decorated function rN)rNrrQrSr`rrr/r/r0$register_edited_channel_post_handlerrz/Dispatcher.register_edited_channel_post_handlerc r)a Decorator for edited channel post handler :param commands: list of commands :param regexp: REGEXP :param content_types: List of content types. :param custom_filters: list of custom filters :param state: :param run_task: run callback in task (no wait results) :param kwargs: :return: decorated function crr)rrrr/r0rs z9Dispatcher.edited_channel_post_handler..decoratorr/rr/rr0edited_channel_post_handlersz&Dispatcher.edited_channel_post_handlerrrcOF|durg}|jj|jg|Rd|i|}|j||||dS)a Register handler for inline query Example: .. code-block:: python3 dp.register_inline_handler(some_inline_handler, lambda inline_query: True) :param callback: :param custom_filters: list of custom filters :param state: :param run_task: run callback in task (no wait results) :param kwargs: :return: decorated function Nr)rNrrTr`rrfrrrrrrr/r/r0register_inline_handler z"Dispatcher.register_inline_handlercfdd}|S)a Decorator for inline query handler Example: .. code-block:: python3 @dp.inline_handler(lambda inline_query: True) async def some_inline_handler(inline_query: types.InlineQuery) :param state: :param custom_filters: list of custom filters :param run_task: run callback in task (no wait results) :param kwargs: :return: decorated function c"j|gRd|SNr)rrrrrrfrr/r0rz,Dispatcher.inline_handler..decoratorr/rfrrrrrr/rr0inline_handlerzDispatcher.inline_handlercOr)a Register handler for chosen inline query Example: .. code-block:: python3 dp.register_chosen_inline_handler(some_chosen_inline_handler, lambda chosen_inline_result: True) :param callback: :param state: :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: :return: Nr)rNrrUr`rrr/r/r0register_chosen_inline_handlerrz)Dispatcher.register_chosen_inline_handlercr)a Decorator for chosen inline query handler Example: .. code-block:: python3 @dp.chosen_inline_handler(lambda chosen_inline_result: True) async def some_chosen_inline_handler(chosen_inline_result: types.ChosenInlineResult) :param state: :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: :return: crr)rrrr/r0r&rz3Dispatcher.chosen_inline_handler..decoratorr/rr/rr0chosen_inline_handlerrz Dispatcher.chosen_inline_handlercO:|jj|jg|Rd|i|}|j||||dS)ar Register handler for callback query Example: .. code-block:: python3 dp.register_callback_query_handler(some_callback_handler, lambda callback_query: True) :param callback: :param state: :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: rN)rNrrVr`rrr/r/r0register_callback_query_handler, z*Dispatcher.register_callback_query_handlercr)a Decorator for callback query handler Example: .. code-block:: python3 @dp.callback_query_handler(lambda callback_query: True) async def some_callback_handler(callback_query: types.CallbackQuery) :param state: :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: crr)rrrr/r0rSrz4Dispatcher.callback_query_handler..decoratorr/rr/rr0callback_query_handlerBz!Dispatcher.callback_query_handlercOr)ax Register handler for shipping query Example: .. code-block:: python3 dp.register_shipping_query_handler(some_shipping_query_handler, lambda shipping_query: True) :param callback: :param state: :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: rN)rNrrWr`rrr/r/r0register_shipping_query_handlerYs z*Dispatcher.register_shipping_query_handlercr)a Decorator for shipping query handler Example: .. code-block:: python3 @dp.shipping_query_handler(lambda shipping_query: True) async def some_shipping_query_handler(shipping_query: types.ShippingQuery) :param state: :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: crr)rrrr/r0rrz4Dispatcher.shipping_query_handler..decoratorr/rr/rr0shipping_query_handlerprz!Dispatcher.shipping_query_handlercOr)a Register handler for pre-checkout query Example: .. code-block:: python3 dp.register_pre_checkout_query_handler(some_pre_checkout_query_handler, lambda shipping_query: True) :param callback: :param state: :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: rN)rNrrXr`rrr/r/r0#register_pre_checkout_query_handlerrz.Dispatcher.register_pre_checkout_query_handlercr)a Decorator for pre-checkout query handler Example: .. code-block:: python3 @dp.pre_checkout_query_handler(lambda shipping_query: True) async def some_pre_checkout_query_handler(shipping_query: types.ShippingQuery) :param state: :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: crr)rrrr/r0rsz8Dispatcher.pre_checkout_query_handler..decoratorr/rr/rr0pre_checkout_query_handlersz%Dispatcher.pre_checkout_query_handler)rcO6|jj|jg|Ri|}|j||||dS)a' Register handler for poll Example: .. code-block:: python3 dp.register_poll_handler(some_poll_handler) :param callback: :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: N)rNrrYr`rrfrrrrrr/r/r0register_poll_handler z Dispatcher.register_poll_handlercfdd}|S)a0 Decorator for poll handler Example: .. code-block:: python3 @dp.poll_handler() async def some_poll_handler(poll: types.Poll) :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: c j|gRdi|SNr)rrrrrrfr/r0rz*Dispatcher.poll_handler..decoratorr/rfrrrrr/rr0 poll_handlerzDispatcher.poll_handlercOr)a< Register handler for poll_answer Example: .. code-block:: python3 dp.register_poll_answer_handler(some_poll_answer_handler) :param callback: :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: N)rNrrZr`rrr/r/r0register_poll_answer_handlerrz'Dispatcher.register_poll_answer_handlercr)aR Decorator for poll_answer handler Example: .. code-block:: python3 @dp.poll_answer_handler() async def some_poll_answer_handler(poll_answer: types.PollAnswer) :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: crr)r rrr/r0rrz1Dispatcher.poll_answer_handler..decoratorr/rr/rr0poll_answer_handlerr zDispatcher.poll_answer_handlerrrcO8|jj|jg|Ri|}|jj||||ddS)aE Register handler for my_chat_member Example: .. code-block:: python3 dp.register_my_chat_member_handler(some_my_chat_member_handler) :param callback: :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: handlerfiltersN)rNrr[r`rrr/r/r0register_my_chat_member_handler   z*Dispatcher.register_my_chat_member_handlercr)aV Decorator for my_chat_member handler Example: .. code-block:: python3 @dp.my_chat_member_handler() async def some_handler(my_chat_member: types.ChatMemberUpdated) :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: crr)rrrr/r0r8z4Dispatcher.my_chat_member_handler..decoratorr/rr/rr0my_chat_member_handler( z!Dispatcher.my_chat_member_handlercOr )a< Register handler for chat_member Example: .. code-block:: python3 dp.register_chat_member_handler(some_chat_member_handler) :param callback: :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: rN)rNrr\r`rrr/r/r0register_chat_member_handlerCrz'Dispatcher.register_chat_member_handlercr)aM Decorator for chat_member handler Example: .. code-block:: python3 @dp.chat_member_handler() async def some_handler(chat_member: types.ChatMemberUpdated) :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: crr)rrrr/r0rprz1Dispatcher.chat_member_handler..decoratorr/rr/rr0chat_member_handler`rzDispatcher.chat_member_handlercOr )a> Register handler for chat_join_request Example: .. code-block:: python3 dp.register_chat_join_request(some_chat_join_request) :param callback: :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: rN)rNrr]r`rrr/r/r0"register_chat_join_request_handler{rz-Dispatcher.register_chat_join_request_handlercr)aO Decorator for chat_join_request handler Example: .. code-block:: python3 @dp.chat_join_request() async def some_handler(chat_member: types.ChatJoinRequest) :param custom_filters: :param run_task: run callback in task (no wait results) :param kwargs: crr)rrrr/r0rrz7Dispatcher.chat_join_request_handler..decoratorr/rr/rr0chat_join_request_handlerrz$Dispatcher.chat_join_request_handler)rrcOr)z Register handler for errors :param callback: :param exception: you can make handler for specific errors type :param run_task: run callback in task (no wait results) rN)rNrr^r`r)rfrrrrrrr/r/r0register_errors_handlers z"Dispatcher.register_errors_handlercr)z Decorator for errors handler :param exception: you can make handler for specific errors type :param run_task: run callback in task (no wait results) :return: cs(j|gRdi|S)Nr)rrrrrrrrfr/r0rsz,Dispatcher.errors_handler..decoratorr/)rfrrrrrr/rr0errors_handlers zDispatcher.errors_handlerrrrrcCsP|durtj}|r|jnd}|dur tj}|r|jnd}t|j||dS)aZ Get current state for user in chat as context .. code-block:: python3 with dp.current_state(chat=message.chat.id, user=message.user.id) as state: pass state = dp.current_state() state.set_state('my_state') :param chat: :param user: :return: N)r3rr)r"r get_currentidrrr3)rfrrchat_objuser_objr/r/r0 current_states  zDispatcher.current_stateuser_idz3.0)old_namenew_name until_version stacklevelchat_idrater#r)no_errorcs`|js td|dur|j}|dur|j}|dur8|dur8tj}|r*|jnd}tj }|r6|jnd}t }|jj ||dIdH} | durO|ii} || vrWi| |<| |} | t |} || } | |kpl| dk} | | t<|| t<|| t <| | t<| s| td7<nd| t<| || |jj||| dIdH| s|std|||d| | S) a Execute throttling manager. Returns True if limit has not exceeded otherwise raises ThrottleError or returns False :param key: key in storage :param rate: limit (by default is equal to default rate limit) :param user_id: user id :param chat_id: chat id :param no_error: return boolean value instead of raising error :return: bool *This storage does not provide Leaky BucketNrrrrrbucketkeyrrr/)r3 has_bucketrrMrLr"rrrrtime get_bucketgetrrrrrr6 set_bucketr%)rfr2r,r#r)r-r r!nowr0datacalleddeltaresultr/r/r0throttlesD    zDispatcher.throttlecs|js td|dur*|dur*tj}|r|jnd}tj}|r(|jnd}|jj||dIdH}| |i}t d|||d|S)z Get information about key in bucket :param key: :param chat_id: :param user_id: :return: r.Nrr1r/) r3r3rr"rrrrr5r6r%)rfr2r)r#r r!r0r9r/r/r0 check_key*s    zDispatcher.check_keycs|js td|dur*|dur*tj}|r|jnd}tj}|r(|jnd}|jj||dIdH}|rL||vrL||=|jj |||dIdHdSdS)z{ Release blocked key :param key: :param chat_id: :param user_id: :return: r.Nrr/TF) r3r3rr"rrrrr5r7)rfr2r)r#r r!r0r/r/r0 release_keyCs    zDispatcher.release_keycs(fddtfdd}|S)a Execute handler as task and return None. Use this decorator for slow handlers (with timeouts) .. code-block:: python3 @dp.message_handler(commands=['command']) @dp.async_task async def cmd_with_timeout(message: types.Message): await asyncio.sleep(120) return SendMessage(message.chat.id, 'KABOOM').reply(message) :param func: :return: c srz|}Wn ty&}ztjtj|WYd}~dSd}~wwt |t r7t| j dSdSrk) r<rr,rr^r{r"rrr+r rrJ)taskrrrlr/r0process_responseps  z/Dispatcher.async_task..process_responsecs$t|i|}|dSrk)r,radd_done_callback)argsrr@)funcrAr/r0wrapperzsz&Dispatcher.async_task..wrapper functoolswraps)rfrDrEr/)rDrArfr0 async_task_s  zDispatcher.async_taskcCs |dur|j}|r||S|Srk)r4rI)rfrrr/r/r0rs  zDispatcher._wrap_async_task on_throttledcsfdd}|S)a Meta-decorator for throttling. Invokes on_throttled if the handler was throttled. Example: .. code-block:: python3 async def handler_throttled(message: types.Message, **kwargs): await message.answer("Throttled!") @dp.throttled(handler_throttled) async def some_handler(message: types.Message): await message.answer("Didn't throttled!") :param on_throttled: the callable object that should be either a function or return a coroutine :param key: key in storage :param rate: limit (by default is equal to default rate limit) :param user_id: user id :param chat_id: chat id :return: decorator cs&tfdd}|S)Ncsjdur njddIdH}|r!|i|IdHS|dr]tr>|i|IdHdS|dtitjg|Ri|}td|dSdS)NTr+)r,r2r#r)rg) r=rIr6r,iscoroutinefunctionget_running_looprGpartialrun_in_executor)rCris_not_throttled partial_func)r)rDr2rJr,rfr#r/r0wrappeds<  z8Dispatcher.throttled..decorator..wrappedrF)rDrQr)r2rJr,rfr#)rDr0rsz'Dispatcher.throttled..decoratorr/)rfrJr2r,r#r)rr/rRr0 throttleds zDispatcher.throttled validatorrsrrcCs|jj||||ddS)a0 Register filter :param callback: callable or subclass of :obj:`AbstractFilter` :param validator: custom validator. :param event_handlers: list of instances of :obj:`Handler` :param exclude_event_handlers: list of excluded event handlers (:obj:`Handler`) )rrTrsrrN)rNrt)rfrrTrsrrr/r/r0 bind_filters  zDispatcher.bind_filtercCs|jj|ddS)zk Unregister filter :param callback: callable of subclass of :obj:`AbstractFilter` rN)rNunbind)rfrr/r/r0 unbind_filterszDispatcher.unbind_filtercCs|j|dS)zO Setup middleware :param middleware: :return: N)r_setup)rfr_r/r/r0setup_middlewareszDispatcher.setup_middleware)rirn)T)rr(NNTrN)NNrk)NNNNN)NNN)PrI __module__ __qualname____doc__DEFAULT_RATE_LIMITtypingOptionalrboolrhpropertyr,r-rgrqrervryrr"rrarrintListstrrrrurrrrrrrrrrrrrrrrrrrrrr r r CallablerrrrrrrrUnionrr"rr=r>r?rIcallablerrSrrIterablerUrWrYr/r/r/r0r2%s 1 i L H  $ O          <"=   r2)r)r*)s,  L  (