o LDi)@sfddlZddlZddlmZmZddlmZddlmZmZddl m Z m Z m Z ddl mZmZmZmZddlmZddlmZdd lmZdd lmZmZmZdd lmZdd lmZdd l m!Z!e rjddl"m#Z#dedede ddfddZ$de!dej%de&e'e(ffddZ)de!deej%egee ffddZ*GdddeZ+Gddde+Z,Gdd d e+Z-dS)!N)ABCabstractmethod) Transport) AwaitableCallable) TYPE_CHECKINGAnycast) JsonPayloadMultipartWriterPayloadweb)Handler) Application) middleware)Bot Dispatcherloggers)TelegramMethod) TelegramType)IPFilter) InputFileapp 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 j/var/www/www-root/data/www/77.83.87.30/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_ip0s  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<rwebhookwarningr HTTPUnauthorized)r-r= ip_addressacceptr,r r%_ip_filter_middlewareKs z3ip_filter_middleware.._ip_filter_middleware)rr Requestrr)r,rDr rCr%ip_filter_middlewareBs rFc @s$eZdZ d'dedededdfddZd ed ed eddfd d Z dededdfddZ e d(ddZ e de jdefddZe dededefddZdedeeefddfddZdede jde jfddZded eedBdefd!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)rrHrIset_background_feed_update_tasks)selfrrHrIr r r%__init__Ws 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_routehandle)rLrrNrr r r%registerjszBaseRequestHandler.registerrrcs|IdHdSN)close)rLrrr r r%rPusz BaseRequestHandler._handle_closecsdSrUr rLr r r%rVxszBaseRequestHandler.closer-csdS)z This method should be implemented in subclasses of this class. Resolve Bot instance from request. :param request: :return: Bot instance Nr rLr-r r r% resolve_bot|szBaseRequestHandler.resolve_bottelegram_secret_tokenbotcCsdSrUr rLrZr[r r r% verify_secretsz BaseRequestHandler.verify_secretupdatecsJ|jjd||d|jIdH}t|tr#|jj||dIdHdSdS)Nr[r^r[resultr )rfeed_raw_updaterI isinstancersilent_call_request)rLr[r^rar r r%_background_feed_updates   z*BaseRequestHandler._background_feed_updatecsTt|j||j|jjdIdHd}|j|||jj t j i|jj dS)Nloadsr_)dumps) asyncio create_taskrejsonsession json_loadsrKaddadd_done_callbackdiscardr json_response json_dumps)rLr[r-feed_update_taskr r r%_handle_request_backgrounds z-BaseRequestHandler._handle_request_backgroundracCs|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)ryfilename) r r secrets token_urlsafer*__api_method__set_content_disposition model_dumpitemsrl prepare_valuereadr|)rLr[rawriterpayloadr{keyvaluer r r%_build_response_writers.  z)BaseRequestHandler._build_response_writercsH|jj||j|jjdIdHfi|jIdH}tj|j||ddS)Nrfr`)body) rfeed_webhook_updaterkrlrmrIr Responser)rLr[r-rar 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-) rYr]r3r4r rrHrtr)rLr-r[r r r%rSszBaseRequestHandler.handle)FrN)__name__ __module__ __qualname__rboolrrMstrrrTrPrrVr rErrYr]dictrerrtrrr rrrS__call__r r r r%rGVs>       %rGc steZdZ  ddededededBdeddf fd d Zd ededefd d Z dddZ de j defddZ ZS)SimpleRequestHandlerTNrr[rH secret_tokenrIrc 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` rrHNr )superrMr[r)rLrr[rHrrI __class__r r%rMs zSimpleRequestHandler.__init__rZcCs|jr t||jSdSNT)rr}compare_digestr\r r r%r]sz"SimpleRequestHandler.verify_secretcs|jjIdHdS)z# Close bot session N)r[rlrVrWr r r%rVszSimpleRequestHandler.closer-cs|jSrU)r[rXr r r%rYsz SimpleRequestHandler.resolve_botTNr)rrrrrrrrrMr]rVr rErY __classcell__r r rr%rs& rc seZdZ  ddededeeefdBdeddf fdd Zd ed e defd d Z dddZ dede deddffdd Z dejde fddZZS)TokenBasedRequestHandlerTNrrH bot_settingsrIrc 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 )rrMrbots)rLrrHrrIrr r%rMs  z!TokenBasedRequestHandler.__init__rZr[cCsdSrr r\r r r%r]sz&TokenBasedRequestHandler.verify_secretcs&|jD] }|jIdHqdSrU)rvaluesrlrV)rLr[r r r%rVszTokenBasedRequestHandler.closerNrrc s0d|vr d}t|tj|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}' substringrNN) ValueErrorrrT)rLrrNrmsgrr r%rTsz!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)rLr-rr r r%rY's    z$TokenBasedRequestHandler.resolve_botrr)rrrrrrrrrMrr]rVrrTr rErYrr r rr%rs$  r).rir}abcrrrcollections.abcrrtypingrrr aiohttpr r r r aiohttp.typedefsraiohttp.web_appraiohttp.web_middlewaresraiogramrrraiogram.methodsraiogram.methods.baseraiogram.webhook.securityr aiogram.typesrr+rEtuplerrr<rFrGrrr r r r%s4          }$