o Fhc@sddlmZddlZddlZddlZddlZddlmZmZmZm Z ddl m Z ddl m Z mZmZmZmZmZmZmZddlmZddlmZdd lmZdd lmZdd lmZmZdd l m!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/ddl0m1Z1ddl2m3Z3m4Z4ddl5m6Z6m7Z7ddl8m9Z9ddl:m;Z;ddlm?Z?e4dddddZ@Gdd d e?ZAdS)!) annotationsN)CancelledErrorEventFutureLock)suppress)AnyAsyncGenerator AwaitableDictListOptionalSetUnion)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_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_fsmnamer0Optional[BaseStorage]r1rr2Optional[BaseEventIsolation]r3boolr4 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 )r4z4FSM storage should be instance of 'BaseStorage' not update)router event_name)r0strategyr2N)superr/__init__ isinstancer TypeErrortype__name__r'r< observersregister_listen_updateouter_middlewarer(r)rrrfsmshutdownclose workflow_datar _running_lock _stop_signal_stopped_signalset_handle_update_tasks)selfr0r1r2r3r4r9 __class__P/var/www/html/venv/lib/python3.10/site-packages/aiogram/dispatcher/dispatcher.pyrA&s0 zDispatcher.__init__itemstrcCs |j|SNrM)rSrXrVrVrW __getitem__ds zDispatcher.__getitem__keyvaluecCs||j|<dSrZr[)rSr]r^rVrVrW __setitem__gzDispatcher.__setitem__cCs |j|=dSrZr[)rSr]rVrVrW __delitem__js zDispatcher.__delitem__default Optional[Any]cCs|j||SrZ)rMget)rSr]rbrVrVrWrdmr`zDispatcher.getrcCs|jjSrZ)rJr0rSrVrVrWr0pszDispatcher.storageOptional[Router]cCsdS)z} Dispatcher has no parent router and can't be included to any other routers or dispatchers :return: NrVrerVrVrW 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)rSr^rVrVrWrg}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: FricontextNiz/Update id=%s is %s. Duration %d ms by bot id=%dhandledz not handled)asyncioget_running_looptimerirmodel_validate model_dumpr<wrap_outer_middlewaretriggerrMr%reventinfo update_idid) rSrir<r9looprl start_timeresponse finish_timedurationrVrVrW 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: rirjrir<NrV)rrp_feed_webhook_update)rSrir<r9 parsed_updaterVrVrWfeed_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 dispatchererrorrDrEwarning next_delaycounterrwasleepruresetrvoffset) clsrirrrbackoff get_updatesr9failedupdateser<rVrVrW_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_typertrV) event_typertr!warningswarnmodel_dump_jsonRuntimeWarningr&r<propagate_event)rSr<r9rrtrrVrVrWrHs     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)rrrtrrUrE)rrirrrVrVrWsilent_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 Nrir>Cause exception while process update id=%d by bot id=%d %s: %sT) r}rBrrr%rrrt exceptionrvrwrUrE)rSrir<rr9rzrrVrVrW_process_update(s"  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)rSrrrVrVrW_process_with_semaphoreDs z"Dispatcher._process_with_semaphorehandle_as_taskstasks_concurrency_limit Optional[int]c 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)rrrrz&Polling stopped for bot @%s id=%d - %rrV)merrruusernamerw full_namerm Semaphorerracquire create_taskrrRaddadd_done_callbackdiscard) rSrirrrrrr9userrr<rhandle_update_taskrVrVrW_pollingSs@      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}rrrtrrvrwrUrE)rSrir<r9rrVrVrWrszDispatcher._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)Nrirj_rr:r;css ddSdSrZ)done set_result)r)waiterrVrWrelease_waitersz6Dispatcher.feed_webhook_update..release_waiterrtask 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) rrrrrrBrrm ensure_futurer)rrr)rirSrVrWprocess_responses  z8Dispatcher.feed_webhook_update..process_response)rrr:r;rV)rrr:r;)rBrrp contextvars copy_contextrmrn create_future call_laterrrrrremove_done_callbackcancelrrr) rSrir<rr9ctxrxrtimeout_handleprocess_updatesrrzrV)rirSrrWfeed_webhook_updatesF        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)rNlockedrhrOrPrQwaitrerVrVrW stop_pollings   zDispatcher.stop_pollingsigsignal.SignalscCs6|jsdStjd|j|jsdS|jdS)NzReceived %s signal)rNrrrrr4rOrQ)rSrrVrVrW_signal_stop_pollings zDispatcher._signal_stop_polling rrrrhandle_signalsclose_bot_sessionrbots&Optional[Union[List[str], UNSET_TYPE]]rrcs|stdd| vrtdj4IdHYjdur!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|rLtjd d |DIdHn|r^tjd d |DIdHwwjwWdIdHdS1IdHswwYdS)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 pollingrizbKeyword argument 'bot' is not acceptable, the bot instance should be passed as positional argumentN)rrz Start pollingc s0g|]}tjd|dqS))rirrrrrrV)rmrr.0rirrrrrSrrMrVrW 8s z,Dispatcher.start_polling..) return_whenzPolling stoppedcss|]}|jVqdSrZ)rrLrrVrVrW Wsz+Dispatcher.start_polling..rV) ValueErrorrNrOrrPrresolve_used_update_typesclearrmrnrNotImplementedErroradd_signal_handlersignalSIGTERMrSIGINTrMpop emit_startuprrruappendrrFIRST_COMPLETEDrrgather emit_shutdownrQ)rSrrrrrrrrr9rxtasksrpendingrrVrrW start_pollings                  0zDispatcher.start_pollingc OsVttt|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 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: rN)rKeyboardInterruptrmrunr) rSrrrrrrrrr9rVrVrW run_pollingZs" $zDispatcher.run_polling)r0r5r1rr2r6r3r7r4r8r9rr:r;)rXrYr:r)r]rYr^rr:r;)r]rYr:r;rZ)rbrcr]rYr:rc)r:r)r:rf)r^r*r:r;)rirr<rr9rr:r)rirr<r~r9rr:r) rirrrrr#rrr:r)r<rr9rr:r)rirrrr:r;)T) rirr<rrr7r9rr:r7)rrrrr:r7)rirrrrr7rr#rrrrr9rr:r;)r) rirr<rrrr9rr:r)r:r;)rrr:r;)rrrrrr7rr#rrrr7rr7rrr9rr:r;)!rE __module__ __qualname____doc__r USER_IN_CHATrAr\r_rardpropertyr0rgsetterr}r classmethodDEFAULT_BACKOFF_CONFIGrrHrrrrrrrrrrr __classcell__rVrVrTrWr/!st >       -  9   < ;  mr/)B __future__rrmrrrrrrr contextlibrtypingrr r r r r rrr 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*r r/rVrVrVrWs8  (