a y{Jh+@sddlZddlZddlZddlmZmZddlmZddlmZddl m Z m Z m Z m Z mZmZmZddlmZmZddlmZddlmZmZeeZd Zd ZGd d d ZGd ddeZdS)N)EventLock)suppress) TracebackType)Any AwaitableCallableDictOptionalTypeUnion)BaseMiddlewareBot)get_flag)MessageTelegramObjectg@gc @sxeZdZdZddeedeeee fe e ee e ddddZ e edd d Ze dd d d ZddddZddddZddddZddddZe eee ee eedddZeeefee efee e ddddZeeefee efee e ddddZeeefee efee e ddddZeeefee efee e ddd d!Zeeefee efee e ddd"d#Zeeefee efee e ddd$d%Z eeefee efee e ddd&d'Z!eeefee efee e ddd(d)Z"eeefee efee e ddd*d+Z#eeefee efee e ddd,d-Z$eeefee efee e ddd.d/Z%dS)0ChatActionSendera This utility helps to automatically send chat action until long actions is done to take acknowledge bot users the bot is doing something and not crashed. Provides simply to use context manager. Technically sender start background task with infinity loop which works until action will be finished and sends the `chat action `_ every 5 seconds. Ntyping)message_thread_idactioninterval initial_sleep)botchat_idrrrrreturncCsF||_||_||_||_||_||_t|_t|_ t|_ d|_ dS)z :param bot: instance of the bot :param chat_id: target chat id :param action: chat action type :param interval: interval between iterations :param initial_sleep: sleep before first sending of the action N) rrrrrrr_lockr _close_event _closed_event_task)selfrrrrrrr K/var/www/shaz/venv/lib/python3.9/site-packages/aiogram/utils/chat_action.py__init__szChatActionSender.__init__)rcCs t|jSN)boolrrr r r!running<szChatActionSender.running)rrcsFttj(t|j|IdHWdn1s80YdSr#)rasyncio TimeoutErrorwait_forrwait)rrr r r!_wait@s zChatActionSender._waitc std|j|j|jjzd}||jIdH|j st }td|j|j|jj||jj |j|j|j dIdH|d7}|jt |}||IdHq0Wtd|j|j|jj|jn$td|j|j|jj|j0dS)Nz9Started chat action %r sender in chat_id=%s via bot id=%drzFSent chat action %r to chat_id=%s via bot %d (already sent actions %d))rrrz:Finished chat action %r sender in chat_id=%s via bot id=%d)loggerdebugrrridr+rris_settime monotonicZsend_chat_actionrrrset)rcounterstartrr r r!_workerDsP   zChatActionSender._workerc sp|j4IdHH|j|j|jr2tdt||_ WdIdHql1IdHsb0YdS)NzAlready running) rrclearrr& RuntimeErrorr'Z create_taskr6rr%r r r!_runis   zChatActionSender._runc s|j4IdH\|js,WdIdHdS|jsP|j|jIdHd|_WdIdHq1IdHsv0YdSr#)rr&rr0r3rr*rr%r r r!_stopqs  zChatActionSender._stopcs|IdH|Sr#)r9r%r r r! __aenter__zszChatActionSender.__aenter__)exc_type exc_value tracebackrcs|IdHdSr#)r:)rr<r=r>r r r! __aexit__~szChatActionSender.__aexit__)rrrrrcCs|||d||dS)z2Create instance of the sender with `typing` actionrrrrrrr clsrrrrr r r!rs zChatActionSender.typingcCs|||d||dS)z8Create instance of the sender with `upload_photo` action upload_photor@r rAr r r!rCs zChatActionSender.upload_photocCs|||d||dS)z8Create instance of the sender with `record_video` action record_videor@r rAr r r!rDs zChatActionSender.record_videocCs|||d||dS)z8Create instance of the sender with `upload_video` action upload_videor@r rAr r r!rEs zChatActionSender.upload_videocCs|||d||dS)z8Create instance of the sender with `record_voice` action record_voicer@r rAr r r!rFs zChatActionSender.record_voicecCs|||d||dS)z8Create instance of the sender with `upload_voice` action upload_voicer@r rAr r r!rGs zChatActionSender.upload_voicecCs|||d||dS)z;Create instance of the sender with `upload_document` actionupload_documentr@r rAr r r!rHs z ChatActionSender.upload_documentcCs|||d||dS)z:Create instance of the sender with `choose_sticker` actionchoose_stickerr@r rAr r r!rIs zChatActionSender.choose_stickercCs|||d||dS)z9Create instance of the sender with `find_location` action find_locationr@r rAr r r!rJs zChatActionSender.find_locationcCs|||d||dS)z=Create instance of the sender with `record_video_note` actionrecord_video_noter@r rAr r r!rKs z"ChatActionSender.record_video_notecCs|||d||dS)z=Create instance of the sender with `upload_video_note` actionupload_video_noter@r rAr r r!rL0s z"ChatActionSender.upload_video_note)&__name__ __module__ __qualname____doc__DEFAULT_INTERVALDEFAULT_INITIAL_SLEEPrr strintr floatr"propertyr$r&r+r6r9r:r;r BaseExceptionrrr? classmethodrrCrDrErFrGrHrIrJrKrLr r r r!rs  %              rc@sDeZdZdZeeeeefge efeeeefedddZ dS)ChatActionMiddlewarezP Helps to automatically use chat action sender for all message handlers )handlereventdatarc st|ts|||IdHS|d}t|dp.d}i}t|tr|d}rT||d<|d}rj||d<|d} r| |d<nt|trd|d<n||d<t|tr|jr|jnd|d<tf||j j d|4IdH&|||IdHWdIdHS1IdHs0YdS) Nr chat_actionrrrrr)rr) isinstancerrdictgetr$Zis_topic_messagerrZchatr/) rrZr[r\rr]kwargsrrrr r r!__call__Gs.     "zChatActionMiddleware.__call__N) rMrNrOrPrrr rSrrrbr r r r!rYBs  rY) r'loggingr1rr contextlibrtypesrrrrrr r r r Zaiogramr rZaiogram.dispatcher.flagsrZ aiogram.typesrr getLoggerrMr-rQrRrrYr r r r!s   $  2