o
    Zcd                     @   s   d Z ddlmZ ddlZddlmZ zddlmZ W n ey%   edw ddl	m
Z dd	lmZ d
dlmZ dddgiZg dZG dd deZG dd deZdd ZeZ
eZdS )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)	WebClientzPlease `pip install slack-sdk`   )tqdm)_range   )
MonoWorkerzgithub.com/	0x2b3bfa0	casperdcl)SlackIO
tqdm_slacktsranger   trangec                       s(   e Zd ZdZ fddZdd Z  ZS )r   z,Non-blocking file-like IO using a Slack app.c              
      sz   t t|   t|d| _| jj| _z| jj|| jd| _	W dS  t
y< } ztt| d| _	W Y d}~dS d}~ww )z-Creates a new message in the given `channel`.)token)channeltextN)superr   __init__r   client	__class____name__r   chat_postMessagemessage	Exception	tqdm_autowritestr)selfr   r   er    H/var/www/chikooza/env/lib/python3.10/site-packages/tqdm/contrib/slack.pyr      s   
zSlackIO.__init__c              
   C   s   |sd}| dd }|| jkrdS | j}|du rdS || _z| j| jj|d |d d| d d}W |S  tyO } zt	t
| W Y d}~dS d}~ww )	z,Replaces internal `message`'s text with `s`.z... Nr   ts`)r   r%   r   )replacestripr   r   submitr   chat_updater   r   r   r   )r   sr   futurer   r!   r!   r"   r   *   s$   
zSlackIO.write)r   
__module____qualname____doc__r   r   __classcell__r!   r!   r    r"   r      s    r   c                       s8   e Zd ZdZ fddZ fddZ fddZ  ZS )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: <https://api.slack.com/authentication/basics>
    - copy the bot `{token}` & `{channel}` and paste below
    >>> from tqdm.contrib.slack import tqdm, trange
    >>> for i in tqdm(iterable, token='{token}', channel='{channel}'):
    ...     ...
    c                    sx   | ds/| }tdtj t|dtd|dtd| _	t
d| dd|d< tt| j|i | 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.
        disable
HTTPClientr   TQDM_SLACK_TOKENr   TQDM_SLACK_CHANNELg      ?minintervalN)getcopylogging	getLoggersetLevelWARNINGr   popr   siomaxr   r   r   r   argskwargsr    r!   r"   r   J   s   
ztqdm_slack.__init__c                    s   t t| jdi | | j}|dd r#|d dddd|d< nd|d< |d du r7g d	|d< d
|d< | j| jdi | d S )N
bar_formatz<bar/>z
`{bar:10}`z{bar}z`{bar:10u}`z{l_bar}`{bar:10}`{r_bar}asciiF)z:black_square:z:small_blue_diamond:z:large_blue_diamond:z:large_blue_square:iP  ncolsr!   )	r   r   displayformat_dictr6   r'   r=   r   format_meter)r   rA   fmtr    r!   r"   rE   `   s   ztqdm_slack.displayc                    s0   t t| j|i | | js| jd d S d S )Nr$   )r   r   clearr1   r=   r   r?   r    r!   r"   rI   n   s   ztqdm_slack.clear)r   r-   r.   r/   r   rE   rI   r0   r!   r!   r    r"   r   >   s
    r   c                  O   s   t t|  fi |S )z
    A shortcut for `tqdm.contrib.slack.tqdm(xrange(*args), **kwargs)`.
    On Python3+, `range` is used instead of `xrange`.
    )r   r   )r@   rA   r!   r!   r"   r   t   s   r   )r/   
__future__r   r8   osr   	slack_sdkr   ImportErrorautor   r   utilsr   utils_workerr	   
__author____all__r   r   r   r   r!   r!   r!   r"   <module>   s&    
!6	