o Fh)@sfddlZddlZddlmZmZddlmZddlmZmZm Z m Z m Z m Z m Z mZddlmZmZmZmZddlmZddlmZddlmZdd lmZmZmZdd lmZdd l m!Z!dd l"m#Z#dd l$m%Z%dedededdfddZ&de%dej'de e(e)ffddZ*de%de ej'egeeffddZ+GdddeZ,Gddde,Z-Gddde,Z.dS) N)ABCabstractmethod) Transport)Any AwaitableCallableDictOptionalSetTuplecast) JsonPayloadMultipartWriterPayloadweb)Handler) Application) middleware)Bot Dispatcherloggers)TelegramMethod) TelegramType) InputFile)IPFilterapp dispatcherkwargsreturnc sh|dj|dtdtddffdd }dtdtddffdd }|j||j|dS) z This function helps to configure a startup-shutdown process :param app: aiohttp application :param dispatcher: aiogram dispatcher :param kwargs: additional data :return: )rrakwrNcjdiIdHdSN) emit_startuprr r workflow_datar#Q/var/www/html/venv/lib/python3.10/site-packages/aiogram/webhook/aiohttp_server.py on_startup#z%setup_application..on_startupcr!r") emit_shutdownr%r&r#r( on_shutdown&r*z&setup_application..on_shutdown)r'rr)appendr,)rrrr)r,r#r&r(setup_applications  r. ip_filterrequestcCs^|jdd}r|jddd^}}|||vfStt|jd}r-|\}}|||vfSdS)NzX-Forwarded-For,)maxsplitpeername)r1F)headersgetsplitr r transportget_extra_info)r/r0 forwarded_for_ peer_namehostr#r#r(check_ip-s  r?cs$tdtjdtdtffdd }|S)z) :param ip_filter: :return: r0handlerrcs:t|d\}}|stjd|t||IdHS)N)r/r0z,Blocking request from an unauthorized IP: %s)r?rwebhookwarningrHTTPUnauthorized)r0r@ ip_addressacceptr/r#r(_ip_filter_middlewareHs z3ip_filter_middleware.._ip_filter_middleware)rrRequestrr)r/rGr#rFr(ip_filter_middleware?s rIc @s$eZdZ d'dedededdfddZd ed ed eddfd d Z dededdfddZ e d(ddZ e de jdefddZe dededefddZdedeeefddfddZdede jde jfddZded eeedefd!d"Zdede jde jfd#d$Zde jde jfd%d&ZeZdS))BaseRequestHandlerFrhandle_in_backgrounddatarNcKs||_||_||_t|_dS)aZ Base handler that helps to handle incoming request from aiohttp and propagate it to the Dispatcher :param dispatcher: instance of :class:`aiogram.dispatcher.dispatcher.Dispatcher` :param handle_in_background: immediately responds to the Telegram instead of a waiting end of a handler process N)rrKrLset_background_feed_update_tasks)selfrrKrLr#r#r(__init__Ts zBaseRequestHandler.__init__pathrrcKs,|j|j|jjd||jfi|dS)z Register route and shutdown callback :param app: instance of aiohttp Application :param path: route path :param kwargs: POSTN)r,r- _handle_closerouter add_routehandlerOrrQrr#r#r(registergszBaseRequestHandler.registerrr cs|IdHdSN)close)rOrr r#r#r(rSrsz BaseRequestHandler._handle_closecsdSrYr#rOr#r#r(rZuszBaseRequestHandler.closer0csdS)z This method should be implemented in subclasses of this class. Resolve Bot instance from request. :param request: :return: Bot instance Nr#rOr0r#r#r( resolve_botys zBaseRequestHandler.resolve_bottelegram_secret_tokenbotcCsdSrYr#rOr^r_r#r#r( verify_secretsz BaseRequestHandler.verify_secretupdatecsJ|jjd||d|jIdH}t|tr#|jj||dIdHdSdS)Nr_rbr_resultr#)rfeed_raw_updaterL isinstancersilent_call_request)rOr_rbrer#r#r(_background_feed_updates   z*BaseRequestHandler._background_feed_updatecsTt|j||j|jjdIdHd}|j|||jj t j i|jj dS)Nloadsrc)dumps) asyncio create_taskrijsonsession json_loadsrNaddadd_done_callbackdiscardr json_response json_dumps)rOr_r0feed_update_taskr#r#r(_handle_request_backgrounds z-BaseRequestHandler._handle_request_backgroundrecCs|stiStddtdd}||j}|jdddi}|jddD]\}}|j j |||d }|s9q)||}|jd|dq)|D]\}}|| |}|jd||j p^|d qJ|S) Nz form-datawebhookBoundary)boundarymethod)nameF)warnings)r_files)r}filename) r rsecrets token_urlsafer-__api_method__set_content_disposition model_dumpitemsrp prepare_valuereadr)rOr_rewriterpayloadrkeyvaluer#r#r(_build_response_writers.  z)BaseRequestHandler._build_response_writercsH|jj||j|jjdIdHfi|jIdH}tj|j||ddS)Nrjrd)body) rfeed_webhook_updaterorprqrLrResponser)rOr_r0rer#r#r(_handle_requests z"BaseRequestHandler._handle_requestcsd||IdH}||jdd|stjdddS|jr(|j||dIdHS|j||dIdHS)NzX-Telegram-Bot-Api-Secret-Tokenr1 Unauthorizedi)rstatus)r_r0) r]rar6r7rrrKrxr)rOr0r_r#r#r(rVszBaseRequestHandler.handle)FrN)__name__ __module__ __qualname__rboolrrPstrrrXrSrrZrrHrr]rarrirrxr rrrrrrV__call__r#r#r#r(rJSs>       #rJc steZdZ  ddedededeededdf fd d Z d ededefd d Z dddZ de j defddZZS)SimpleRequestHandlerTNrr_rK secret_tokenrLrc s(tjd||d|||_||_dS)aM Handler for single Bot instance :param dispatcher: instance of :class:`aiogram.dispatcher.dispatcher.Dispatcher` :param handle_in_background: immediately responds to the Telegram instead of a waiting end of handler process :param bot: instance of :class:`aiogram.client.bot.Bot` rrKNr#)superrPr_r)rOrr_rKrrL __class__r#r(rPs zSimpleRequestHandler.__init__r^cCs|jr t||jSdSNT)rrcompare_digestr`r#r#r(rasz"SimpleRequestHandler.verify_secretcs|jjIdHdS)z# Close bot session N)r_rprZr[r#r#r(rZszSimpleRequestHandler.closer0cs|jSrY)r_r\r#r#r(r]sz SimpleRequestHandler.resolve_botTNr)rrrrrrr rrrPrarZrrHr] __classcell__r#r#rr(rs& rc seZdZ  ddededeeeefdeddf fdd Z d ed e defd d Z dddZ dede deddffdd Zdejde fddZZS)TokenBasedRequestHandlerTNrrK bot_settingsrLrc s4tjd||d||duri}||_i|_dS)aP Handler that supports multiple bots the context will be resolved from path variable 'bot_token' .. note:: This handler is not recommended in due to token is available in URL and can be logged by reverse proxy server or other middleware. :param dispatcher: instance of :class:`aiogram.dispatcher.dispatcher.Dispatcher` :param handle_in_background: immediately responds to the Telegram instead of a waiting end of handler process :param bot_settings: kwargs that will be passed to new Bot instance rNr#)rrPrbots)rOrrKrrLrr#r(rPs  z!TokenBasedRequestHandler.__init__r^r_cCsdSrr#r`r#r#r(rasz&TokenBasedRequestHandler.verify_secretcs&|jD] }|jIdHqdSrY)rvaluesrprZ)rOr_r#r#r(rZszTokenBasedRequestHandler.closerQrrc s,d|vrtdtj|fd|i|dS)z Validate path, register route and shutdown callback :param app: instance of aiohttp Application :param path: route path :param kwargs: z {bot_token}z,Path should contains '{bot_token}' substringrQN) ValueErrorrrXrWrr#r(rXsz!TokenBasedRequestHandler.registerr0cs:|jd}||jvrtdd|i|j|j|<|j|S)z Get bot token from a path and create or get from cache Bot instance :param request: :return: bot_tokentokenNr#) match_inforrr)rOr0rr#r#r(r]!s    z$TokenBasedRequestHandler.resolve_botrr)rrrrrr rrrrPrrarZrrXrrHr]rr#r#rr(rs$  r)/rmrabcrrrtypingrrrrr r r r aiohttpr rrraiohttp.typedefsraiohttp.web_appraiohttp.web_middlewaresraiogramrrraiogram.methodsraiogram.methods.baser aiogram.typesraiogram.webhook.securityrr.rHrrr?rIrJrrr#r#r#r(s0 (         {$