o @j(@s^ddlZddlZddlmZmZddlmZddlmZmZm Z m Z m Z m Z m Z mZddlmZmZddlmZddlmZddlmZdd lmZmZmZdd lmZdd lmZdd 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)MultipartWriterweb) Application)Handler) 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_startuprrr workflow_datar!m/var/www/www-root/data/www/ai-bot.pdev.uz/venv/lib/python3.10/site-packages/aiogram/webhook/aiohttp_server.py on_startup#z%setup_application..on_startupcrr ) 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)r/F)headersgetsplitr r transportget_extra_info)r-r. forwarded_for_ peer_namehostr!r!r&check_ip-s  r=cs$tdtjdtdtffdd }|S)z) :param ip_filter: :return: r.handlerrcs:t|d\}}|stjd|t||IdHS)N)r-r.z,Blocking request from an unauthorized IP: %s)r=rwebhookwarningrHTTPUnauthorized)r.r> ip_addressacceptr-r!r&_ip_filter_middlewareHs z3ip_filter_middleware.._ip_filter_middleware)rrRequestrr)r-rEr!rDr&ip_filter_middleware?s rGc @s eZdZ d%dedededdfddZd ed ed eddfd d Z d eddfddZ e d&ddZ e de jdefddZe dededefddZdedeeefddfddZdede jde jfddZdedeeedefdd 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)rrIrJset_background_feed_update_tasks)selfrrIrJr!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_routehandlerMrrOrr!r!r&registergszBaseRequestHandler.registercs|IdHdSN)close)rMrr!r!r&rQrsz BaseRequestHandler._handle_closecsdSrWr!rMr!r!r&rXuszBaseRequestHandler.closer.csdS)z This method should be implemented in subclasses of this class. Resolve Bot instance from request. :param request: :return: Bot instance Nr!rMr.r!r!r& resolve_botys zBaseRequestHandler.resolve_bottelegram_secret_tokenbotcCsdSrWr!rMr\r]r!r!r& verify_secretsz BaseRequestHandler.verify_secretupdatecsJ|jjd||d|jIdH}t|tr#|jj||dIdHdSdS)Nr]r`r]resultr!)rfeed_raw_updaterJ isinstancersilent_call_request)rMr]r`rcr!r!r&_background_feed_updates   z*BaseRequestHandler._background_feed_updatecsTt|j||j|jjdIdHd}|j|||jj t j i|jj dS)Nloadsra)dumps) asyncio create_taskrgjsonsession json_loadsrLaddadd_done_callbackdiscardr json_response json_dumps)rMr]r.feed_update_taskr!r!r&_handle_request_backgrounds z-BaseRequestHandler._handle_request_backgroundrccCstddtdd}|s|S||j}|jdddi}|jddD]\}}|jj |||d }|s7q'||}|jd|dq'|D]\}}|| |}|jd||j p\|d qH|S) Nz form-datawebhookBoundary)boundarymethod)nameF)warnings)r]files)r{filename) r secrets token_urlsafer+__api_method__set_content_disposition model_dumpitemsrn prepare_valuereadr~)rMr]rcwriterpayloadr}keyvaluer!r!r&_build_response_writers.  z)BaseRequestHandler._build_response_writercsH|jj||j|jjdIdHfi|jIdH}tj|j||ddS)Nrhrb)body) rfeed_webhook_updatermrnrorJrResponser)rMr]r.rcr!r!r&_handle_requests z"BaseRequestHandler._handle_requestcsd||IdH}||jdd|stjdddS|jr(|j||dIdHS|j||dIdHS)NzX-Telegram-Bot-Api-Secret-Tokenr/ Unauthorizedi)rstatus)r]r.) r[r_r4r5rrrIrvr)rMr.r]r!r!r&rTszBaseRequestHandler.handle)FrN)__name__ __module__ __qualname__rboolrrNstrrrVrQrrXrrFrr[r_rrgrrvr rrr rrrT__call__r!r!r!r&rHSs>       rHc steZdZ  ddedededeededdf fd d Z d ededefd d Z dddZ de j defddZZS)SimpleRequestHandlerTNrr]rI secret_tokenrJrc 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` rrINr!)superrNr]r)rMrr]rIrrJ __class__r!r&rNs zSimpleRequestHandler.__init__r\cCs|jr t||jSdSNT)rrcompare_digestr^r!r!r&r_sz"SimpleRequestHandler.verify_secretcs|jjIdHdS)z# Close bot session N)r]rnrXrYr!r!r&rXszSimpleRequestHandler.closer.cs|jSrW)r]rZr!r!r&r[sz SimpleRequestHandler.resolve_botTNr)rrrrrrr rrrNr_rXrrFr[ __classcell__r!r!rr&rs& rc seZdZ  ddededeeeefdeddf fdd Z d ed e defd d Z dddZ dede deddffdd Zdejde fddZZS)TokenBasedRequestHandlerTNrrI bot_settingsrJrc 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!)rrNrbots)rMrrIrrJrr!r&rNs  z!TokenBasedRequestHandler.__init__r\r]cCsdSrr!r^r!r!r&r_ sz&TokenBasedRequestHandler.verify_secretcs&|jD] }|jIdHqdSrW)rvaluesrnrX)rMr]r!r!r&rX szTokenBasedRequestHandler.closerOrrc 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}' substringrON) ValueErrorrrVrUrr!r&rVsz!TokenBasedRequestHandler.registerr.cs:|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)rMr.rr!r!r&r[s    z$TokenBasedRequestHandler.resolve_botrr)rrrrrr rrrrNrr_rXrrVrrFr[rr!r!rr&rs$  r)-rkrabcrrrtypingrrrrr r r r aiohttpr r aiohttp.abcraiohttp.typedefsraiohttp.web_middlewaresraiogramrrraiogram.methodsraiogram.methods.baser aiogram.typesraiogram.webhook.securityrr,rFrrr=rGrHrrr!r!r!r&s0 (         w$