U )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|||_||_t|_|jj|_ |j dS)z-Creates a new message in the given `chat_id`.N) superr __init__tokenchat_idrsession __class____name__text message_id)selfrrr9/tmp/pip-unpacked-wheel-u8lm3wqg/tqdm/contrib/telegram.pyrs  zTelegramIO.__init__c Cst|dr|jSz8|jj|jd|jd|jd|jddd}Wn0t k rx}zt t |W5d}~XYn4X| ddkrtd td d n|d d |_|jSdS)N _message_idz%s/sendMessage` MarkdownV2)rr parse_modedataZ error_codeiz2Creation rate limit: try increasing `mininterval`.r) stacklevelresultr)hasattrrrpostAPIrrrjson Exception tqdm_autowritestrgetrr)rreserrrr(s$  zTelegramIO.message_idc Cs|sd}|dd}||jkr&dS|j}|dkr8dS||_z8|j|jj|jd|jd|d|j |ddd }Wn0t k r}zt t |W5d}~XYnX|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.writec Cshz.|j|jj|jd|j|j|jdd}Wn0tk r^}zt t |W5d}~XYnX|SdS)zDeletes internal `message_id`.z%s/deleteMessage)rrr!N) r4rr&r'rrrr)r*r+r,)rr6r/rrrdeletePs  zTelegramIO.delete) r __module__ __qualname____doc__r'rpropertyrr+r7 __classcell__rrrrr 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|ds6|}t|dtd|dtd|_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 rrargskwargsrrrrls ztqdm_telegram.__init__c sbtt|jf||j}|ddrB|ddddd|d<nd|d<|j|jf|dS)NZ bar_formatzz {bar:10u}z{bar}z{l_bar}{bar:10u}{r_bar}) rr displayZ format_dictr-r2r@r+Z format_meter)rrCfmtrrrrD~s  ztqdm_telegram.displaycs(tt|j|||js$|jddS)Nr1)rr clearr=r@r+rArrrrFsztqdm_telegram.clearcs@|jr dStt||js<|jdkr2|jdks<|jdS)Nr)r=rr closeZleaveposr@r7)rrrrrGs ztqdm_telegram.close) rr8r9r:rrDrFrGr<rrrrr \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 )rBrCrrrrsN)r: __future__rosrwarningsrrequestsrautorr*Zstdrutilsr Z utils_workerr __author____all__r r rrrrrrs         A9