o @jZ@sddlmZddlZddlZddlZddlZddlmZmZmZm Z ddl m Z ddl m Z mZmZmZmZmZmZddlmZddlmZdd lmZdd lmZdd lmZmZdd lm Z m!Z!dd l"m#Z#ddl$m%Z%m&Z&ddl'm(Z(ddl)m*Z*m+Z+ddl,m-Z-m.Z.ddl/m0Z0ddl1m2Z2m3Z3ddl4m5Z5m6Z6ddl7m8Z8ddl9m:Z:ddl;mZ>e3dddddZ?Gdd d e>Z@dS)!) annotationsN)CancelledErrorEventFutureLock)suppress)AnyAsyncGeneratorDictListOptionalSetUnion)loggers)Bot)TelegramAPIError)FSMContextMiddleware)BaseEventIsolation BaseStorage)DisabledEventIsolation MemoryStorage) FSMStrategy) GetUpdatesTelegramMethod) TelegramType)UpdateUser)UNSET UNSET_TYPE)UpdateTypeLookupError)Backoff BackoffConfig) UNHANDLED SkipHandler)TelegramEventObserver)ErrorsMiddleware)UserContextMiddleware)Routerg?g@g?g?) min_delay max_delayfactorjittercsLeZdZdZdejddddddfddZdeddZdfddZdgddZ dhdid!d"Z e djd$d%Z e dkd'd(Z e jdld*d(Z dmd/d0Zdnd2d3Zed4edfdodd?ZedqdBdCZ DdrdsdFdGZd4dDedfdtdIdJZdmdKdLZ MdudvdRdSZdwdTdUZdxdXdYZdZdDeedDdDd[dyd`daZdZdDeedDdDd[dydbdcZZS)z Dispatcherz Root router NF)storage fsm_strategyevents_isolation disable_fsmnamer/Optional[BaseStorage]r0rr1Optional[BaseEventIsolation]r2boolr3 Optional[str]kwargsrreturnNonec stt|j|d|rt|tstdt|jt|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 )r3z4FSM storage should be instance of 'BaseStorage' not update)router event_name)r/strategyr1N)superr.__init__ isinstancer TypeErrortype__name__r&r; observersregister_listen_updateouter_middlewarer'r(rrrfsmshutdownclose workflow_datar _running_lock _stop_signal_stopped_signalset_handle_update_tasks)selfr/r0r1r2r3r8 __class__l/var/www/www-root/data/www/ai-bot.pdev.uz/venv/lib/python3.10/site-packages/aiogram/dispatcher/dispatcher.pyr@&s0 zDispatcher.__init__itemstrcCs |j|SNrL)rRrWrUrUrV __getitem__ds zDispatcher.__getitem__keyvaluecCs||j|<dSrYrZ)rRr\r]rUrUrV __setitem__gzDispatcher.__setitem__cCs |j|=dSrYrZ)rRr\rUrUrV __delitem__js zDispatcher.__delitem__default Optional[Any]cCs|j||SrY)rLget)rRr\rarUrUrVrcmr_zDispatcher.getrcCs|jjSrY)rIr/rRrUrUrVr/pszDispatcher.storageOptional[Router]cCsdS)z} Dispatcher has no parent router and can't be included to any other routers or dispatchers :return: NrUrdrUrUrV parent_routertszDispatcher.parent_routerr)cCstd)z Dispatcher is root Router then configuring parent router is not allowed :param value: :return: z1Dispatcher can not be attached to another Router.) RuntimeError)rRr]rUrUrVrf}sbotrr;rc st}d}|}|j|krtj|d|id}z7|j|jj |i|j |d|iIdH}|t u}|W|}||d} t j d|j|rLdnd| |jS|}||d} t j d|j|rgdnd| |jw) z Main entry point for incoming updates Response of this method can be used as Webhook response :param bot: :param update: FrhcontextNiz/Update id=%s is %s. Duration %d ms by bot id=%dhandledz not handled)asyncioget_running_looptimerhrmodel_validate model_dumpr;wrap_outer_middlewaretriggerrLr$reventinfo update_idid) rRrhr;r8looprk start_timeresponse finish_timedurationrUrUrV feed_updatesL       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: rhrirhr;NrU)rro_feed_webhook_update)rRrhr;r8 parsed_updaterUrUrVfeed_raw_updateszDispatcher.feed_raw_updatepolling_timeoutintbackoff_configr"allowed_updatesOptional[List[str]]AsyncGenerator[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#)r!rsessionrr Exceptionr dispatchererrorrCrDwarning next_delaycounterrvasleeprtresetruoffset) clsrhrrrbackoff get_updatesr8failedupdateser;rUrUrV_listen_updatessF  zDispatcher._listen_updatesc svz|j}|j}Wnty&}ztd|jddtt|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_unsetN) event_update) update_typersrU) event_typersr warningswarnmodel_dump_jsonRuntimeWarningr%r;propagate_event)rRr;r8rrsrrUrUrVrGs     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)rrrsrrTrD)rrhrrrUrUrVsilent_call_requests "zDispatcher.silent_call_requestT call_answerc sz#|j||fi|IdH}|r t|tr |j||dIdH|tuWStyE}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 Nrhr>Cause exception while process update id=%d by bot id=%d %s: %sT) r|rArrr$rrrs exceptionrurvrTrD)rRrhr;rr8ryrrUrUrV_process_update(s"  zDispatcher._process_updatehandle_as_tasksc s|IdH}tjd|j|j|jzE|j||||d2z+3dHW}|jd||d|} |rDt | } |j | | |j jq| IdHq6Wtjd|j|j|jdStjd|j|j|jw)zg Internal polling process :param bot: :param kwargs: :return: Nz"Run polling for bot @%s id=%d - %r)rrrr~z&Polling stopped for bot @%s id=%d - %rrU)merrrtusernamerv full_namerrrl create_taskrQaddadd_done_callbackdiscard) rRrhrrrrr8userr; handle_updatehandle_update_taskrUrUrV_pollingDs0   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) r|rrrsrrurvrTrD)rRrhr;r8rrUrUrVrkszDispatcher._feed_webhook_update7Union[Update, Dict[str, Any]]_timeoutfloat&Optional[TelegramMethod[TelegramType]]c 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)Nrhri_rr9r:css ddSdSrY)done set_result)r)waiterrUrVrelease_waitersz6Dispatcher.feed_webhook_update..release_waiterr~task Future[Any]c s\tdtz|}Wnty}z|d}~wwt|tr,tj |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.r) rrrrrrArrl ensure_futurer)rrr)rhrRrUrVprocess_responses  z8Dispatcher.feed_webhook_update..process_response)rrr9r:rU)rrr9r:)rArro contextvars copy_contextrlrm create_future call_laterrrrrremove_done_callbackcancelrrr) rRrhr;rr8ctxrwrtimeout_handleprocess_updatesrryrU)rhrRrrVfeed_webhook_update{sF        zDispatcher.feed_webhook_updatecsB|js td|jr|jsdS|j|jIdHdS)zd Execute this method if you want to stop polling programmatically :return: zPolling is not startedN)rMlockedrgrNrOrPwaitrdrUrUrV stop_pollings   zDispatcher.stop_pollingsigsignal.SignalscCs6|jsdStjd|j|jsdS|jdS)NzReceived %s signal)rMrrrrr3rNrP)rRrrUrUrV_signal_stop_pollings zDispatcher._signal_stop_polling rrrrhandle_signalsclose_bot_sessionbots&Optional[Union[List[str], UNSET_TYPE]]rrcs|stdd|vrtdj4IdHXjdur!t_jdur*t_tur2jj|rjt } t t | t jjt j| t jjt jWdn1sewY|dj|dvr}djdd|diIdHtjdzfdd |D} | tjtj| tjd IdH\} } | D]} | t t | IdHWdn1swYqtj| IdHWtjd z!jdd|diIdHW|r tjd d |DIdHn|rtjd d |DIdHwwjn@tjd z!jdd|diIdHW|rKtjd d |DIdHn|r]tjd d |DIdHwwjwWdIdHdS1IdHsvwYdS)al 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 kwargs: contextual data :return: z6At least one bot instance is required to start pollingrhzbKeyword argument 'bot' is not acceptable, the bot instance should be passed as positional argumentN)rrz Start pollingc s.g|]}tjd|dqS))rhrrrrrU)rlrr.0rhrrrrrRrLrUrV s z,Dispatcher.start_polling..) return_whenzPolling stoppedcss|]}|jVqdSrY)rrKrrUrUrV /sz+Dispatcher.start_polling..rU) ValueErrorrMrNrrOrresolve_used_update_typesclearrlrmrNotImplementedErroradd_signal_handlersignalSIGTERMrSIGINTrLpop emit_startuprrrtappendrrFIRST_COMPLETEDrrgather emit_shutdownrP)rRrrrrrrrr8rwtasksrpendingrrUrrV start_pollings                   0zDispatcher.start_pollingc OsTttt|j|i|||||||dWdS1s#wYdS)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 kwargs: contextual data :return: rN)rKeyboardInterruptrlrunr) rRrrrrrrrr8rUrUrV run_polling2s $zDispatcher.run_polling)r/r4r0rr1r5r2r6r3r7r8rr9r:)rWrXr9r)r\rXr]rr9r:)r\rXr9r:rY)rarbr\rXr9rb)r9r)r9re)r]r)r9r:)rhrr;rr8rr9r)rhrr;r}r8rr9r) rhrrrrr"rrr9r)r;rr8rr9r)rhrrrr9r:)T) rhrr;rrr6r8rr9r6)rhrrrrr6rr"rrr8rr9r:)r) rhrr;rrrr8rr9r)r9r:)rrr9r:)rrrrrr6rr"rrrr6rr6r8rr9r:) rD __module__ __qualname____doc__r USER_IN_CHATr@r[r^r`rcpropertyr/rfsetterr|r classmethodDEFAULT_BACKOFF_CONFIGrrGrrrrrrrrrr __classcell__rUrUrSrVr.!sl >       -  9   ' ;  ir.)A __future__rrlrrrrrrr contextlibrtypingrr r r r r rr client.botr exceptionsrfsm.middlewarerfsm.storage.baserrfsm.storage.memoryrr fsm.strategyrmethodsrr methods.basertypesrr types.baserr types.updater utils.backoffr!r" event.basesr$r%event.telegramr&middlewares.errorr'middlewares.user_contextr(r<r)rr.rUrUrUrVs8  $