o 4jHf@sddlmZddlZddlZddlZddlZddlZddlmZmZm Z m Z ddl m Z m Z ddlmZddlmZmZddlmZddlmZdd lmZdd lmZmZdd lmZmZdd lm Z dd l!m"Z"m#Z#ddl$m%Z%m&Z&ddl'm(Z(m)Z)ddl*m+Z+ddl,m-Z-m.Z.ddl/m0Z0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9erddl:m;Z;ddlGdd d e9Z?dS)!) annotationsN)CancelledErrorEventFutureLock)AsyncGenerator Awaitable)suppress) TYPE_CHECKINGAny)loggers)TelegramAPIError)FSMContextMiddleware)BaseEventIsolation BaseStorage)DisabledEventIsolation MemoryStorage) FSMStrategy) GetUpdatesTelegramMethod)UpdateUser)UNSET UNSET_TYPE)UpdateTypeLookupError)Backoff BackoffConfig) UNHANDLED SkipHandler)TelegramEventObserver)ErrorsMiddleware)UserContextMiddleware)Router)Bot) TelegramTypeg?g@g?g?) min_delay max_delayfactorjittercs\eZdZdZdejdddddlfddZdmddZdnddZdoddZ dpdqd!d"Z e drd$d%Z e dsd'd(Z e jdtd*d(Z dud/d0Zdvd2d3Zed4edfdwdd?ZedydBdCZ Ddzd{dFdGZd|dLdMZd4dDeddfd}dQdRZdudSdTZ Ud~ddZd[Zdd\d]Zdd`daZdbdDeedDdDddcddhdiZdbdDeedDdDddcddjdkZZ S) Dispatcherz Root router NF)storage fsm_strategyevents_isolation disable_fsmnamer+BaseStorage | Noner,rr-BaseEventIsolation | Noner.boolr/ str | Nonekwargsr returnNonec stj|d|rt|tsdt|j}t|t|dd|_|j d<|j |j |j t ||j tt|pAt||pFtd|_|sS|j |j|j |jj||_t|_d|_d|_t|_dS)a Root router :param storage: Storage for FSM :param fsm_strategy: FSM strategy :param events_isolation: Events isolation :param disable_fsm: Disable FSM, note that if you disable FSM then you should not use storage and events isolation :param kwargs: Other arguments, will be passed as keyword arguments to handlers )r/z4FSM storage should be instance of 'BaseStorage' not update)router event_name)r+strategyr-N)super__init__ isinstancertype__name__ TypeErrorr r7 observersregister_listen_updateouter_middlewarer!r"rrrfsmshutdownclose workflow_datar _running_lock _stop_signal_stopped_signalset_handle_update_tasks)selfr+r,r-r.r/r4msg __class__Z/var/www/kodo/Anonymous/send/lib/python3.10/site-packages/aiogram/dispatcher/dispatcher.pyr<+s0 zDispatcher.__init__itemstrcCs |j|SNrH)rNrTrRrRrS __getitem__is zDispatcher.__getitem__keyvaluecCs||j|<dSrVrW)rNrYrZrRrRrS __setitem__lzDispatcher.__setitem__cCs |j|=dSrVrW)rNrYrRrRrS __delitem__os zDispatcher.__delitem__default Any | NonecCs|j||SrV)rHget)rNrYr^rRrRrSr`rr\zDispatcher.getrcCs|jjSrV)rEr+rNrRrRrSr+uszDispatcher.storage Router | NonecCsdS)z} Dispatcher has no parent router and can't be included to any other routers or dispatchers :return: NrRrarRrRrS parent_routeryszDispatcher.parent_routerr#cCs d}t|)z Dispatcher is root Router then configuring parent router is not allowed :param value: :return: z1Dispatcher can not be attached to another Router.) RuntimeError)rNrZrOrRrRrSrcsbotr$r7rc st}d}|}|j|krtj|d|id}z9|j|jj |d|i|j |d|iIdH}|t u}|W|}||d} t j d|j|rNdnd | |jS|}||d} t j d|j|ridnd | |jw) z Main entry point for incoming updates Response of this method can be used as Webhook response :param bot: :param update: Frecontext dispatcherNiz/Update id=%s is %s. Duration %d ms by bot id=%dhandledz not handled)asyncioget_running_looptimerermodel_validate model_dumpr7wrap_outer_middlewaretriggerrHrr eventinfo update_idid) rNrer7r4loopri start_timeresponse finish_timedurationrRrRrS feed_updatesN       zDispatcher.feed_updatedict[str, Any]cs0tj|d|id}|jd||d|IdHS)z Main entry point for incoming updates with automatic Dict->Update serializer :param bot: :param update: :param kwargs: rerfrer7NrR)rrm_feed_webhook_update)rNrer7r4 parsed_updaterRrRrSfeed_raw_updateszDispatcher.feed_raw_updatepolling_timeoutintbackoff_configrallowed_updateslist[str] | NoneAsyncGenerator[Update, None]c Cst|d}t||d}i}|jjrt|jj||d<d} z ||fi|IdH} Wn2ty^} z&d}tjdt | j | tj d|j |j |j|IdHWYd} ~ qd} ~ ww|rqtjd |j |j|d}| D] } | V| jd |_qsq) z Endless updates reader with correctly handling any server-side or connection errors. So you may not worry that the polling will stop working. )config)timeoutrrequest_timeoutFTNz Failed to fetch updates - %s: %szASleep for %f seconds and try again... (tryings = %d, bot id = %d)z2Connection established (tryings = %d, bot id = %d)r)rrsessionrr Exceptionr rherrorr>r?warning next_delaycounterrtasleeprrresetrsoffset) clsrerrrbackoff get_updatesr4failedupdateser7rRrRrS_listen_updatessF  zDispatcher._listen_updatesc szz|j}|j}Wnty(}ztjd|jddtddt|d}~ww|j|d|j d ||d|IdHS) a Main updates listener Workflow: - Detect content type and propagate to observers in current router - If no one filter is pass - propagate update to child routers as Update :param update: :param kwargs: :return: zDetected unknown update type. Seems like Telegram Bot API was updated and you have installed not latest version of aiogram framework Update: T) exclude_unset stacklevelN) event_update) update_typerqrR) event_typerqrwarningswarnmodel_dump_jsonRuntimeWarningrr7propagate_event)rNr7r4rrqrrRrRrSrCs"    zDispatcher._listen_updateresultTelegramMethod[Any]c sRz ||IdHWdSty(}ztjd|jj|WYd}~dSd}~ww)zk Simulate answer into WebHook :param bot: :param result: :return: NzFailed to make answer: %s: %s)r r rqrrQr?)rrerrrRrRrSsilent_call_requests "zDispatcher.silent_call_requestT call_answerc sz-|j||fi|IdH}|r%t|tr*|j||dIdHW|t uSW|t uSW|t uStyO}ztjd|j|j |j j |WYd}~dSd}~ww)aL Propagate update to event listeners :param bot: instance of Bot :param update: instance of Update :param call_answer: need to execute response as Telegram method (like answer into webhook) :param kwargs: contextual data for middlewares, filters and handlers :return: status Nrer>Cause exception while process update id=%d by bot id=%d %s: %sT) rzr=rrrr rq exceptionrsrtrQr?r)rNrer7rr4rwrrRrRrS_process_update0s* zDispatcher._process_update handle_updateAwaitable[bool] semaphoreasyncio.Semaphorecs"z |IdHW|S|w)a Process update with semaphore to limit concurrent tasks :param handle_update: Coroutine that processes the update :param semaphore: Semaphore to limit concurrent tasks :return: bool indicating the result of the update processing N)release)rNrrrRrRrS_process_with_semaphoreRs z"Dispatcher._process_with_semaphorehandle_as_taskstasks_concurrency_limit int | Nonec s|IdH}tjd|j|j|jd} |dur!|r!t|} zX|j ||||d2z>3dHW} |j d|| d|} |rd| rQ| IdHt | | | } nt | } |j| | |jjq+| IdHq+6Wtjd|j|j|jdStjd|j|j|jw)a Internal polling process :param bot: :param polling_timeout: Long-polling wait time :param handle_as_tasks: Run task for each event and no wait result :param backoff_config: backoff-retry config :param allowed_updates: List of the update types you want your bot to receive :param tasks_concurrency_limit: Maximum number of concurrent updates to process (None = no limit), used only if handle_as_tasks is True :param kwargs: :return: Nz"Run polling for bot @%s id=%d - %r)rrrr|z&Polling stopped for bot @%s id=%d - %rrR)mer rhrrusernamert full_namerj Semaphorerracquire create_taskrrMaddadd_done_callbackdiscard) rNrerrrrrr4userrr7rhandle_update_taskrRrRrS_pollingcsR      zDispatcher._pollingc sVz|j||fi|IdHWSty*}ztjd|j|j|jj|d}~ww)zg The same with `Dispatcher.process_update()` but returns real response instead of bool Nr) rzrr rqrrsrtrQr?)rNrer7r4rrRrRrSr}szDispatcher._feed_webhook_update7Update | dict[str, Any]_timeoutfloat#TelegramMethod[TelegramType] | Nonec s&t|tstj|did}t}t}|dfdd }|||}t j d|d |} | j ||ddfd d } zHzIdHWnt y^| || w| rt| } t| trs| W|Sn| || j | |dW|dSW|dS|w)Nrerf_r r5r6css ddSdSrV)done set_result)r)waiterrRrSrelease_waitersz6Dispatcher.feed_webhook_update..release_waiterr|task Future[Any]cs>tjdtdd|}t|trtj|ddSdS)NzDetected slow response into webhook. Telegram is waiting for response only first 60 seconds and then re-send update. For preventing this situation response into webhook returned immediately and handler is moved to background and still processing update.rrr) rrrrr=rrj ensure_futurer)rr)rerNrRrSprocess_responses z8Dispatcher.feed_webhook_update..process_response)rr r5r6rR)rrr5r6)r=rrm contextvars copy_contextrjrk create_future call_laterrr}rrremove_done_callbackcancelrrr) rNrer7rr4ctxrurtimeout_handleprocess_updatesrrwrR)rerNrrSfeed_webhook_updatesF         zDispatcher.feed_webhook_updatecsF|js d}t||jr|jsdS|j|jIdHdS)zd Execute this method if you want to stop polling programmatically :return: zPolling is not startedN)rIlockedrdrJrKrLwait)rNrOrRrRrS stop_pollings   zDispatcher.stop_pollingsigsignal.SignalscCs6|jsdStjd|j|jsdS|jdS)NzReceived %s signal)rIrr rhrr/rJrL)rNrrRrRrS_signal_stop_pollings zDispatcher._signal_stop_polling rrrrhandle_signalsclose_bot_sessionrbotslist[str] | UNSET_TYPE | Nonerrcs|s d} t| d| vrd} t| j4IdHYjdur%t_jdur.t_tur6jj|rnt } t t | t jjt j| t jjt jWdn1siwY|dj| dvrdjdd|diIdHtjdzfdd |D} | tjtj| tjd IdH\} }|D]}|t t |IdHWdn1swYqtj| IdHWtjd z!jdd|diIdHW|rtjd d |DIdHn|r"tjd d |DIdHwwjn@tjd z!jdd|diIdHW|rPtjd d |DIdHn|rbtjd d |DIdHwwjwWdIdHdS1IdHs{wYdS)a Polling runner :param bots: Bot instances (one or more) :param polling_timeout: Long-polling wait time :param handle_as_tasks: Run task for each event and no wait result :param backoff_config: backoff-retry config :param allowed_updates: List of the update types you want your bot to receive By default, all used update types are enabled (resolved from handlers) :param handle_signals: handle signals (SIGINT/SIGTERM) :param close_bot_session: close bot sessions on shutdown :param tasks_concurrency_limit: Maximum number of concurrent updates to process (None = no limit), used only if handle_as_tasks is True :param kwargs: contextual data :return: z6At least one bot instance is required to start pollingrezbKeyword argument 'bot' is not acceptable, the bot instance should be passed as positional argumentN)rhrz Start pollingc s0g|]}tjd|dqS))rerrrrrrR)rjrr.0rerrrrrNrrHrRrS Ws z,Dispatcher.start_polling..) return_whenzPolling stoppedcss|]}|jVqdSrV)rrGrrRrRrS vsz+Dispatcher.start_polling..rR) ValueErrorrIrJrrKrresolve_used_update_typesclearrjrkr NotImplementedErroradd_signal_handlersignalSIGTERMrSIGINTrHpop emit_startupr rhrrappendrrFIRST_COMPLETEDrrgather emit_shutdownrL)rNrrrrrrrrr4rOrutasksrpendingrrRrrS start_polling s               0zDispatcher.start_pollingc Osttp|j|i| |||||||d} zddl} Wnty2t| YWdSwtjdkr\tj | j d} | | WdWdS1sVwYn| t| WdSWddS1swwYdS)a Run many bots with polling :param bots: Bot instances (one or more) :param polling_timeout: Long-polling wait time :param handle_as_tasks: Run task for each event and no wait result :param backoff_config: backoff-retry config :param allowed_updates: List of the update types you want your bot to receive :param handle_signals: handle signals (SIGINT/SIGTERM) :param close_bot_session: close bot sessions on shutdown :param tasks_concurrency_limit: Maximum number of concurrent updates to process (None = no limit), used only if handle_as_tasks is True :param kwargs: contextual data :return: rrN) ) loop_factory) r KeyboardInterruptruvloop ImportErrorrjrunsys version_infoRunnernew_event_loopinstall) rNrrrrrrrrr4cororrunnerrRrRrS run_pollingys>      "zDispatcher.run_polling)r+r0r,rr-r1r.r2r/r3r4r r5r6)rTrUr5r )rYrUrZr r5r6)rYrUr5r6rV)r^r_rYrUr5r_)r5r)r5rb)rZr#r5r6)rer$r7rr4r r5r )rer$r7r{r4r r5r ) rer$rrrrrrr5r)r7rr4r r5r )rer$rrr5r6)T) rer$r7rrr2r4r r5r2)rrrrr5r2)rer$rrrr2rrrrrrr4r r5r6)r) rer$r7rrrr4r r5r)r5r6)rrr5r6)rr$rrrr2rrrrrr2rr2rrr4r r5r6)!r? __module__ __qualname____doc__r USER_IN_CHATr<rXr[r]r`propertyr+rcsetterrzr classmethodDEFAULT_BACKOFF_CONFIGrrCrrrrr}rrrrrr  __classcell__rRrRrPrSr*&st >       .  9  " B =  sr*)@ __future__rrjrrrrrrrrcollections.abcrr contextlibr typingr r aiogramr aiogram.exceptionsr aiogram.fsm.middlewareraiogram.fsm.storage.baserraiogram.fsm.storage.memoryrraiogram.fsm.strategyraiogram.methodsrr aiogram.typesrraiogram.types.baserraiogram.types.updateraiogram.utils.backoffrr event.basesrrevent.telegramr middlewares.errorr!middlewares.user_contextr"r8r#aiogram.client.botr$aiogram.methods.baser%rr*rRrRrRrSs>