o Zzj2@sddlZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl ZddlmZz dd lmZd ZWn eyKd ZYnwdd lmZdd lmZddlmZe rfddlmZeeddZGdddZGdddejj Z!Gdddejj"Z#dS)N) HTTPStatus)Path)socket) SSLContext) TracebackType) TYPE_CHECKING) HTTPServer)bind_unix_socketTF)Update) get_logger)ExtBot)BotUpdater) class_namec @sjeZdZdZdZ ddededddedBd eeBe BdBf d d Z dd e j dBd dfddZ dddZdS) WebhookServerz6Thin wrapper around ``tornado.httpserver.HTTPServer``.) _http_server _server_lock_shutdown_lock is_runninglistenportunixNrr webhook_appWebhookAppClassssl_ctxrcCsv|rtstdt||d|_||_||_d|_d|_|r&t|t r&||_n |r/t t ||_t |_t |_dS)Nz.This OS does not support binding unix sockets.) ssl_optionsF)UNIX_AVAILABLE RuntimeErrorrrrrrr isinstancerr strasyncioLockrr)selfrrrrrr#]/var/www/kodo/tsheker/venv/lib/python3.10/site-packages/telegram/ext/_utils/webhookhandler.py__init__As zWebhookServer.__init__readyreturnc s|j4IdH1|jr|j|jn |jj|j|jdd|_|dur)|t dWdIdHdS1IdHs?wYdS)N)addressTzWebhook Server started.) rrr add_socketrrrset_LOGGERdebug)r"r&r#r#r$ serve_foreverWs .zWebhookServer.serve_foreverc s|j4IdH5|jstd WdIdHdSd|_|j|jIdHtdWdIdHdS1IdHsCwYdS)Nz.Webhook Server is already shut down. ReturningFzWebhook Server stopped)rrr+r,rstopclose_all_connectionsr"r#r#r$shutdownds   .zWebhookServer.shutdownNr'N)__name__ __module__ __qualname____doc__ __slots__rintrrrr%r Eventr-r1r#r#r#r$r4s"  rc @sJeZdZdZ ddedddejdedBfdd Zd ej j d dfd d Z dS)rz!Application used in the WebserverN webhook_pathbotr update_queue secret_tokencCs6|||d|_|dt|jfg}tjj||dS)N)r<r=r>z/?)shared_objectsTelegramHandlertornadoweb Applicationr%)r"r;r<r=r>handlersr#r#r$r%rs zWebhookAppClass.__init__handlerr'cCsdS)zIOverrides the default implementation since we have our own logging setup.Nr#)r"rEr#r#r$ log_requestszWebhookAppClass.log_requestr2) r4r5r6r7rr Queuer%rArBRequestHandlerrFr#r#r#r$ros rc@s|eZdZdZdZdZdddejdedd fd d Z dd d Z dddZ dddZ de ed Bded Bded Bdd fddZd S)r@z:BaseHandler that processes incoming requests from Telegram)r<r>r=)POSTr<r r=r>r'NcCs(||_||_||_|rtddSdS)zFInitialize for each request - that's the interface provided by tornadozLThe webhook server has a secret token, expecting it in incoming requests nowN)r<r=r>r+r,)r"r<r=r>r#r#r$ initializeszTelegramHandler.initializecCs|dddS)zSets default headers Content-Typez!application/json; charset="utf-8"N) set_headerr0r#r#r$set_default_headerssz#TelegramHandler.set_default_headersc std||jj}t|}|t j td|z t ||j }WntyH}ztjd||dtjjt jdd|d}~ww|ritd|jt|j tr^|j ||j|IdHdSdS) zHandle incoming POST requestzWebhook triggeredzWebhook received data: %szySomething went wrong processing the data received from Telegram. Received data was *not* processed! Received data was: %rexc_infozUpdate could not be processedreasonNz%Received Update with ID %d on Webhook)r+r,_validate_postrequestbodydecodejsonloads set_statusrOKr de_jsonr< ExceptioncriticalrArB HTTPError BAD_REQUEST update_idrr insert_callback_datar=put)r" json_stringdataupdateexcr#r#r$posts>        zTelegramHandler.postcCs|jjdd}|dkrtjtj|jdurC|jjd}|s/t dtjjtjdd||jkrEt d|tjjtjdddSdS) z+Only accept requests with content type JSONrKNzapplication/jsonzX-Telegram-Bot-Api-Secret-Tokenz(Request did not include the secret tokenrPz&Request had the wrong secret token: %sz"Request had the wrong secret token) rSheadersgetrArBr]r FORBIDDENr>r+r,)r" ct_headertokenr#r#r$rRs"    zTelegramHandler._validate_posttypvaluetbcCs0tjd|jjd|r|r|r|||fn|ddS)z@Override the default logging and instead use our custom logging.z%s - %szException in TelegramHandlerrNN)r+r,rS remote_ip)r"rlrmrnr#r#r$ log_exceptions  zTelegramHandler.log_exceptionr3)r4r5r6r7r8SUPPORTED_METHODSr rGrrJrMrfrRtype BaseExceptionrrpr#r#r#r$r@s"  $ r@)$r rVhttprpathlibrrsslrtypesrtypingr tornado.webrAtornado.httpserverrtornado.netutilr r ImportErrortelegramr telegram._utils.loggingr telegram.ext._extbotr r r4r+rrBrCrrHr@r#r#r#r$s2              ;