U md@sdZddlmZddlmZddlmZddlmZ ddl m Z dd l m Z d d giZd d dddgZGdd d e ZGdd d e ZddZeZeZdS)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) )getenv)warn)Session)tqdm) TqdmWarning) 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)selfrrrN/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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)rrrrr)replacestriprrsubmitrr$r%rrr'r(r)r*)rsrfuturer-rrrr)8s*   zTelegramIO.writec Cshz.|j|jj|jd|j|j|jdd}Wn0tk r^}zt t |W5d}~XYnX|SdS)zDeletes internal `message_id`.z%s/deleteMessage)rrrN) r2rr$r%rrrr'r(r)r*)rr4r-rrrdeleteMs  zTelegramIO.delete) r __module__ __qualname____doc__r%rpropertyrr)r5 __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 rrargskwargsrrrris 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+r0r>r)Z format_meter)rrAfmtrrrrB{s  ztqdm_telegram.displaycs(tt|j|||js$|jddS)Nr/)rr clearr;r>r)r?rrrrDsztqdm_telegram.clearcs@|jr dStt||js<|jdkr2|jdks<|jdS)Nr)r;rr closeZleaveposr>r5)rrrrrEs ztqdm_telegram.close) rr6r7r8rrBrDrEr:rrrrr Ys   cOstt|f|S)zBShortcut for `tqdm.contrib.telegram.tqdm(range(*args), **kwargs)`.)r range)r@rArrrr sN)r8osrwarningsrrequestsrautorr(ZstdrZ utils_workerr __author____all__r r r r rrrrs       A9