3 dd @sdZddlmZddlZddlmZyddlmZWnek rPedYnXddl m Z dd l m Z d d lmZd d dgiZdddddgZGdddeZGddde ZddZeZ eZdS)z Sends updates to a Slack app. Usage: >>> from tqdm.contrib.slack import tqdm, trange >>> for i in trange(10, token='{token}', channel='{channel}'): ... ... ![screenshot](https://img.tqdm.ml/screenshot-slack.png) )absolute_importN)getenv) WebClientzPlease `pip install slack-sdk`)tqdm)_range) MonoWorkerz github.com/Z 0x2b3bfa0Z casperdclSlackIO tqdm_slacktsrangertrangecs(eZdZdZfddZddZZS)r z,Non-blocking file-like IO using a Slack app.csxtt|jt|d|_|jj|_y|jj||jd|_ Wn6t k rr}zt j t |d|_ WYdd}~XnXdS)z-Creates a new message in the given `channel`.)token)channeltextN)superr __init__rclient __class____name__rZchat_postMessagemessage Exception tqdm_autowritestr)selfrre)r2/tmp/pip-build-v9q4h5k9/tqdm/tqdm/contrib/slack.pyrs  zSlackIO.__init__cCs|sd}|jddj}||jkr&dS|j}|dkr8dS||_y*|j|jj|d|dd|dd}Wn0tk r}ztj t |WYdd}~XnX|SdS) z,Replaces internal `message`'s text with `s`.z... Nrts`)rr!r) replacestriprrZsubmitrZ chat_updaterrrr)rsrfuturerrrrr*s  z SlackIO.write)r __module__ __qualname____doc__rr __classcell__rr)rrr s cs8eZdZdZfddZfddZfddZZS)r a Standard `tqdm.auto.tqdm` but also sends updates to a Slack app. May take a few seconds to create (`__init__`). - create a Slack app with the `chat:write` scope & invite it to a channel: - copy the bot `{token}` & `{channel}` and paste below >>> from tqdm.contrib.slack import tqdm, trange >>> for i in tqdm(iterable, token='{token}', channel='{channel}'): ... ... cst|jds^|j}tjdjtjt|jdtd|jdtd|_ t d|jdd|d<t t |j ||d S) a Parameters ---------- token : str, required. Slack token [default: ${TQDM_SLACK_TOKEN}]. channel : int, required. Slack channel [default: ${TQDM_SLACK_CHANNEL}]. mininterval : float, optional. Minimum of [default: 1.5] to avoid rate limit. See `tqdm.auto.tqdm.__init__` for other parameters. disableZ HTTPClientrZTQDM_SLACK_TOKENrZTQDM_SLACK_CHANNELg?Z minintervalN)getcopylogging getLoggersetLevelWARNINGr poprsiomaxrr r)rargskwargs)rrrrJs ztqdm_slack.__init__c stt|jf||j}|jddrB|djddjdd|d<nd|d<|ddkrnd d d d g|d<d |d<|jj|jf|dS)NZ bar_formatzz `{bar:10}`z{bar}z `{bar:10u}`z{l_bar}`{bar:10}`{r_bar}asciiFz:black_square:z:small_blue_diamond:z:large_blue_diamond:z:large_blue_square:iPZncols) rr displayZ format_dictr,r#r3rZ format_meter)rr6fmt)rrrr8`s   ztqdm_slack.displaycs(tt|j|||js$|jjddS)Nr )rr clearr+r3r)rr5r6)rrrr:nsztqdm_slack.clear)rr'r(r)rr8r:r*rr)rrr >s   cOstt|f|S)z A shortcut for `tqdm.contrib.slack.tqdm(xrange(*args), **kwargs)`. On Python3+, `range` is used instead of `xrange`. )r r)r5r6rrrr ts)r) __future__rr.osrZ slack_sdkr ImportErrorautorrutilsrZ utils_workerr __author____all__r r r r rrrr s"      !6