o j@sdZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z d d giZgd ZGd dde ZGdddeZddZeZeZdS)z Sends updates to a Telegram bot. Usage: >>> from tqdm.contrib.telegram import tqdm, trange >>> for i in trange(10, token='{token}', chat_id='{chat_id}'): ... ... ![screenshot](https://tqdm.github.io/img/screenshot-telegram.gif) )warn)Session)tqdm) TqdmWarning)envwrap) MonoWorkerz github.com/ casperdcl) TelegramIO tqdm_telegramttgrangertrangecs@eZdZdZdZfddZeddZddZd d Z Z S) r z/Non-blocking file-like IO using a Telegram Bot.zhttps://api.telegram.org/botcs2t||_||_t|_|jj|_|j dS)z-Creates a new message in the given `chat_id`.N) super__init__tokenchat_idrsession __class____name__text message_id)selfrrr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/tqdm/contrib/telegram.pyrs   zTelegramIO.__init__c Cst|dr|jSz#|jj|j|jdd|jd|jddd}|}| Wn.t yY}z"|j dkr@t dt d d ntt|WYd}~dSWYd}~dSd}~ww|d d |_|jS) N _message_idz /sendMessage` MarkdownV2)rr parse_modedataiz2Creation rate limit: try increasing `mininterval`.r) stacklevelresultr)hasattrrrpostAPIrrrjsonraise_for_status Exception status_coderr tqdm_autowritestr)rreqreserrrr%s,   zTelegramIO.message_idc Cs|sd}|dd}||jkrdS|j}|durdS||_z|j|jj|j|jdd|d|j |ddd }W|St yW}zt t |WYd}~dSd}~ww) z/Replaces internal `message_id`'s text with `s`.z... Nz/editMessageTextrr)rrrrr )replacestriprrsubmitrr%r&rrr)r+r,r-)rsrfuturer0rrrr,:s* zTelegramIO.writec CsZz|j|jjd|j|jdd}W|Sty,}ztt|WYd}~dSd}~ww)zDeletes internal `message_id`.z${self.API}{self.token}/deleteMessage)rrr N) r5rr%rrr)r+r,r-)rr7r0rrrdeleteOs zTelegramIO.delete) r __module__ __qualname____doc__r&rpropertyrr,r8 __classcell__rrrrr s  r csZeZdZdZedddddddfdd Zfd d Zfd d ZfddZZ S)r a; Standard `tqdm.auto.tqdm` but also sends updates to a Telegram Bot. May take a few seconds to create (`__init__`). - create a bot - copy its `{token}` - add the bot to a chat and send it a message such as `/start` - go to to find out the `{chat_id}` - paste the `{token}` & `{chat_id}` below >>> from tqdm.contrib.telegram import tqdm, trange >>> for i in tqdm(iterable, token='{token}', chat_id='{chat_id}'): ... ... rtelegramT) is_methodN)rrcs4|ds|}t|||_tj|i|dS)a. Parameters ---------- token : str, required. Telegram token [default: ${TQDM_TELEGRAM_TOKEN}]. chat_id : str, required. Telegram chat ID [default: ${TQDM_TELEGRAM_CHAT_ID}]. See `tqdm.auto.tqdm.__init__` for other parameters. disableN)getcopyr tgiorr)rrrargskwargsrrrrks  ztqdm_telegram.__init__c s\tjdi||j}|ddr |ddddd|d<|j|jdi|dS)N bar_formatzz {bar:10u}z{bar}r)rdisplay format_dictrAr3rCr, format_meter)rrEfmtrrrrG|s ztqdm_telegram.displaycs,tj|i||js|jddSdS)Nr2)rclearr@rCr,)rrDrErrrrKsztqdm_telegram.clearcsD|jrdSt|js|jdur|jdks |jdSdSdS)Nr)r@rcloseleaveposrCr8)rrrrrLs  ztqdm_telegram.close) rr9r:r;rrrGrKrLr=rrrrr [s   r cOstt|fi|S)zBShortcut for `tqdm.contrib.telegram.tqdm(range(*args), **kwargs)`.)r range)rDrErrrr sr N)r;warningsrrequestsrautorr+stdrutilsr utils_workerr __author____all__r r r rrrrrs      C6