o Fh/@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@sBeZdZdZddeeddedeee fde e ded e d e d dfd d Z e d efddZd e d dfddZd5ddZd5ddZd5ddZd6ddZde eede ede ed efddZedeefdee efdede e d e d e d df dd Zedeefdee efdede e d e d e d df d!d"Zedeefdee efdede e d e d e d df d#d$Zedeefdee efdede e d e d e d df d%d&Zedeefdee efdede e d e d e d df d'd(Zedeefdee efdede e d e d e d df d)d*Z edeefdee efdede e d e d e d df d+d,Z!edeefdee efdede e d e d e d df d-d.Z"edeefdee efdede e d e d e d df d/d0Z#edeefdee efdede e d e d e d df d1d2Z$edeefdee efdede e d e d e d df d3d4Z%dS)7ChatActionSendera 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_sleepbotchat_idrrrrreturncCsF||_||_||_||_||_||_t|_t|_ t|_ d|_ dS)a` :param bot: instance of the bot :param chat_id: target chat id :param message_thread_id: unique identifier for the target message thread; supergroups only :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 L/var/www/html/venv/lib/python3.10/site-packages/aiogram/utils/chat_action.py__init__s zChatActionSender.__init__cCs t|jSN)boolrrr r r!running=s zChatActionSender.runningcsJttjt|j|IdHWddS1swYdSr#)rasyncio TimeoutErrorwait_forrwait)rrr r r!_waitAs "zChatActionSender._waitc s$td|j|j|jjzrd}||jIdH|j slt }td|j|j|jj||jj |j|j|j dIdH|d7}|jt |}||IdH|j rWtd|j|j|jj|jdSWtd|j|j|jj|jdStd|j|j|jj|jw)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 monotonicsend_chat_actionrrrset)rcounterstartrr r r!_workerEsd    zChatActionSender._workerc st|j4IdH%|j|j|jrtdt||_ WdIdHdS1IdHs3wYdS)NzAlready running) rrclearrr& RuntimeErrorr' create_taskr7rr%r r r!_runjs  .zChatActionSender._runc s|j4IdH0|js WdIdHdS|js*|j|jIdHd|_WdIdHdS1IdHs>wYdSr#)rr&rr0r4rr*rr%r r r!_stoprs  .zChatActionSender._stopcs|IdH|Sr#)r;r%r r r! __aenter__{szChatActionSender.__aenter__exc_type exc_value tracebackcs|IdHdSr#)r<)rr>r?r@r r r! __aexit__szChatActionSender.__aexit__cC||||d||dS)z2Create instance of the sender with `typing` actionrrrrrrrr clsrrrrrr r r!r zChatActionSender.typingcCrB)z8Create instance of the sender with `upload_photo` action upload_photorCr rDr r r!rGrFzChatActionSender.upload_photocCrB)z8Create instance of the sender with `record_video` action record_videorCr rDr r r!rHrFzChatActionSender.record_videocCrB)z8Create instance of the sender with `upload_video` action upload_videorCr rDr r r!rIrFzChatActionSender.upload_videocCrB)z8Create instance of the sender with `record_voice` action record_voicerCr rDr r r!rJrFzChatActionSender.record_voicecCrB)z8Create instance of the sender with `upload_voice` action upload_voicerCr rDr r r!rKrFzChatActionSender.upload_voicecCrB)z;Create instance of the sender with `upload_document` actionupload_documentrCr rDr r r!rLrFz ChatActionSender.upload_documentcCrB)z:Create instance of the sender with `choose_sticker` actionchoose_stickerrCr rDr r r!rM rFzChatActionSender.choose_stickercCrB)z9Create instance of the sender with `find_location` action find_locationrCr rDr r r!rNrFzChatActionSender.find_locationcCrB)z=Create instance of the sender with `record_video_note` actionrecord_video_noterCr rDr r r!rO2rFz"ChatActionSender.record_video_notecCrB)z=Create instance of the sender with `upload_video_note` actionupload_video_noterCr rDr r r!rPErFz"ChatActionSender.upload_video_note)rN)rr)&__name__ __module__ __qualname____doc__DEFAULT_INTERVALDEFAULT_INITIAL_SLEEPrr strintr floatr"propertyr$r&r+r7r;r<r=r BaseExceptionrrrA classmethodrrGrHrIrJrKrLrMrNrOrPr r r r!rs    %                         rc@sJeZdZdZdeeeeefge efdedeeefdefddZ dS) ChatActionMiddlewarezP Helps to automatically use chat action sender for all message handlers handlereventdatarc st|ts|||IdHS|d}t|dpd}i}t|trB|d}r+||d<|d}r6||d<|d} rA| |d<nt|trLd|d<n||d<t|tr[|jr[|jnd|d<td ||j j d|4IdH|||IdHWdIdHS1IdHswYdS) Nr chat_actionrrrrr)rrr ) isinstancerrdictgetr$is_topic_messagerrchatr/) rr^r_r`rrakwargsrrrr r r!__call__^s4    "0zChatActionMiddleware.__call__N) rQrRrSrTrrr rWrrrhr r r r!r]Ys r]) r'loggingr1rr contextlibrtypesrrrrrr r r r aiogramr raiogram.dispatcher.flagsr aiogram.typesrr getLoggerrQr-rUrVrr]r r r r!s"  $  I