3 dl@sdZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd lmZd d lmZd d giZdddddgZGdddeZGddde ZddZeZ eZdS)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://img.tqdm.ml/screenshot-telegram.gif) )absolute_import)getenv)warn)Session)tqdm) TqdmWarning)_range) MonoWorkerz github.com/Z casperdcl TelegramIO tqdm_telegramttgrangertrangecs@eZdZdZdZfddZeddZddZd d Z Z S) r z/Non-blocking file-like IO using a Telegram Bot.zhttps://api.telegram.org/botcs6tt|j||_||_t|_|jj|_ |j dS)z-Creates a new message in the given `chat_id`.N) superr __init__tokenchat_idrsession __class____name__text message_id)selfrr)r5/tmp/pip-build-v9q4h5k9/tqdm/tqdm/contrib/telegram.pyrs  zTelegramIO.__init__cCst|dr|jSy8|jj|jd|jd|jd|jdddj}Wn0t k rx}zt j t |WYdd}~Xn4X|j ddkrtd td d n|d d |_|jSdS)N _message_idz%s/sendMessage` MarkdownV2)rr parse_mode)dataZ error_codeiz2Creation rate limit: try increasing `mininterval`.r) stacklevelresultr)hasattrrrpostAPIrrrjson Exception tqdm_autowritestrgetrr)rreserrrr(s   zTelegramIO.message_idcCs|sd}|jddj}||jkr&dS|j}|dkr8dS||_y8|j|jj|jd|jd|d|j |ddd }Wn0t k r}zt j t |WYdd}~XnX|SdS) z/Replaces internal `message_id`'s text with `s`.z... Nz%s/editMessageTextrr)rrrr)r )replacestriprrsubmitrr$r%rrr'r(r)r*)rsrfuturer-rrrr);s"  zTelegramIO.writecCshy.|j|jj|jd|j|j|jdd}Wn0tk r^}ztj t |WYdd}~XnX|SdS)zDeletes internal `message_id`.z%s/deleteMessage)rr)r N) r2rr$r%rrrr'r(r)r*)rr4r-rrrdeletePs zTelegramIO.delete) r __module__ __qualname____doc__r%rpropertyrr)r5 __classcell__rr)rrr s  csDeZdZdZfddZfddZfddZfdd ZZS) 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}'): ... ... csL|jds6|j}t|jdtd|jdtd|_tt|j||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. disablerZTQDM_TELEGRAM_TOKENrZTQDM_TELEGRAM_CHAT_IDN) r+copyr poprtgiorr r)rargskwargs)rrrrls ztqdm_telegram.__init__c sbtt|jf||j}|jddrB|djddjdd|d<nd|d<|jj|jf|dS)NZ bar_formatzz {bar:10u}z{bar}z{l_bar}{bar:10u}{r_bar}) rr displayZ format_dictr+r0r>r)Z format_meter)rr@fmt)rrrrA~s ztqdm_telegram.displaycs(tt|j|||js$|jjddS)Nr/)rr clearr;r>r))rr?r@)rrrrCsztqdm_telegram.clearcs@|jr dStt|j|jp0|jdko0|jdks<|jjdS)Nr)r;rr closeZleaveposr>r5)r)rrrrDs ztqdm_telegram.close) rr6r7r8rrArCrDr:rr)rrr \s   cOstt|f|S)z A shortcut for `tqdm.contrib.telegram.tqdm(xrange(*args), **kwargs)`. On Python3+, `range` is used instead of `xrange`. )r r )r?r@rrrrsN)r8 __future__rosrwarningsrrequestsrautorr(Zstdrutilsr Z utils_workerr __author____all__r r rrrrrr s         A9