o /hc@sddlZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z m Z mZmZddlZddlmZz dd lmZd ZWn eyQd ZYnwdd lmZdd lmZddlmZe rlddlmZeeddZ GdddZ!Gdddej"j#Z$Gdddej"j%Z&dS)N) HTTPStatus)Path)socket) SSLContext) TracebackType) TYPE_CHECKINGOptionalTypeUnion) HTTPServer)bind_unix_socketTF)Update) get_logger)ExtBot)BotUpdater) class_namec@sleZdZdZdZ ddededddeed ee ee e ff d d Z dd ee jd dfddZdddZdS) 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 RuntimeErrorr rrrrr isinstancerr strasyncioLockrr)selfrrrrrr&[/var/www/html/govbot/env/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__r"intrrr rrr(r#Eventr0r4r&r&r&r'r4s"  rc @sJeZdZdZ ddedddejdeefdd Zd e j 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@rAz/?)shared_objectsTelegramHandlertornadoweb Applicationr()r%r>r?r@rAhandlersr&r&r'r(rs zWebhookAppClass.__init__handlerr*cCsdS)zIOverrides the default implementation since we have our own logging setup.Nr&)r%rHr&r&r' log_requestszWebhookAppClass.log_requestr5) r7r8r9r:r"r#Queuerr(rDrERequestHandlerrIr&r&r&r'ros rc@s|eZdZdZdZdZdddejdedd fd d Z dd d Z dddZ dddZ de eede ede edd fddZd S)rCz:BaseHandler that processes incoming requests from Telegram)r?rAr@)POSTr?rr@rAr*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@rAr.r/)r%r?r@rAr&r&r' initializeszTelegramHandler.initializecCs|dddS)zSets default headers Content-Typez!application/json; charset="utf-8"N) set_headerr3r&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? ExceptioncriticalrDrE HTTPError BAD_REQUEST update_idr!rinsert_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 JSONrNNzapplication/jsonzX-Telegram-Bot-Api-Secret-Tokenz(Request did not include the secret tokenrSz&Request had the wrong secret token: %sz"Request had the wrong secret token) rVheadersgetrDrEr`r FORBIDDENrAr.r/)r% ct_headertokenr&r&r'rUs"    zTelegramHandler._validate_posttypvaluetbcCs0tjd|jjd|r|r|r|||fn|ddS)z@Override the default logging and instead use our custom logging.z%s - %szException in TelegramHandlerrQN)r.r/rV remote_ip)r%rorprqr&r&r' log_exceptions  zTelegramHandler.log_exceptionr6)r7r8r9r:r;SUPPORTED_METHODSr#rJr"rMrPrirUrr BaseExceptionrrsr&r&r&r'rCs"  $ rC)'r#rYhttprpathlibrrsslrtypesrtypingrrr r tornado.webrDtornado.httpserverr tornado.netutilr r ImportErrortelegramr telegram._utils.loggingrtelegram.ext._extbotrrr7r.rrErFrrKrCr&r&r&r's2             ;