o _cl@sdZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd lmZd d lmZd d giZgdZGdddeZGddde 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)selfrrr@/usr/local/lib/python3.10/dist-packages/tqdm/contrib/telegram.pyrs   zTelegramIO.__init__c Cst|dr|jSz|jj|jd|jd|jd|jddd}Wnt y>}zt t |WYd}~dSd}~ww| ddkrOtd td d dS|d d |_|jS)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}||jkrdS|j}|durdS||_z|j|jj|jd|jd|d|j |ddd }W|St yV}zt t |WYd}~dSd}~ww) 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 Csfz|j|jj|jd|j|j|jdd}W|Sty2}zt t |WYd}~dSd}~ww)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  r 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}'): ... ... csP|ds|}t|dtd|dtd|_tt|j|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. disablerZTQDM_TELEGRAM_TOKENrZTQDM_TELEGRAM_CHAT_IDN) r-copyr poprtgiorr rrargskwargsrrrrls ztqdm_telegram.__init__c sjtt|jdi||j}|ddr#|ddddd|d<nd|d<|j|jdi|dS)NZ bar_formatzz {bar:10u}z{bar}z{l_bar}{bar:10u}{r_bar}r) rr displayZ format_dictr-r2r@r+Z format_meter)rrCfmtrrrrD~s ztqdm_telegram.displaycs0tt|j|i||js|jddSdS)Nr1)rr clearr=r@r+rArrrrFsztqdm_telegram.clearcsH|jrdStt||js |jdur|jdks"|jdSdSdS)Nr)r=rr closeZleaveposr@r7)rrrrrGs ztqdm_telegram.close) rr8r9r:rrDrFrGr<rrrrr \s    r cOstt|fi|S)z A shortcut for `tqdm.contrib.telegram.tqdm(xrange(*args), **kwargs)`. On Python3+, `range` is used instead of `xrange`. )r r )rBrCrrrrsrN)r: __future__rosrwarningsrrequestsrautorr*Zstdrutilsr Z utils_workerr __author____all__r r rrrrrrs         A9