o
    [’×i•  ã                   @   sº   d Z ddlmZ ddlZddlmZ z
ddlmZ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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 Discord bot.

Usage:
>>> from tqdm.contrib.discord import tqdm, trange
>>> for i in tqdm(iterable, token='{token}', channel_id='{channel_id}'):
...     ...

![screenshot](https://img.tqdm.ml/screenshot-discord.png)
é    )Úabsolute_importN)Úgetenv)ÚClientÚClientConfigzPlease `pip install disco-py`é   )Útqdm)Ú_rangeé   )Ú
MonoWorkerzgithub.com/Ú	casperdcl)Ú	DiscordIOÚtqdm_discordÚtdranger   Ú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 Discord Bot.c              
      sz   t t| ƒ ¡  tƒ }||_t|ƒ}| jj| _z|j	 
|| j¡| _W dS  ty< } zt t|ƒ¡ W Y d}~dS d}~ww )z0Creates a new message in the given `channel_id`.N)Úsuperr   Ú__init__r   Útokenr   Ú	__class__Ú__name__ÚtextÚapiÚchannels_messages_createÚmessageÚ	ExceptionÚ	tqdm_autoÚwriteÚstr)Úselfr   Ú
channel_idÚconfigÚclientÚe©r   © úR/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/tqdm/contrib/discord.pyr      s   
€ÿzDiscordIO.__init__c              
   C   s€   |sd}|  dd¡ ¡ }|| jkrdS || _z|  | jjd| d ¡}W |S  ty? } zt t	|ƒ¡ W Y d}~dS d}~ww )z,Replaces internal `message`'s text with `s`.z...úÚ Nú`)
ÚreplaceÚstripr   Úsubmitr   Úeditr   r   r   r   )r   ÚsÚfuturer!   r#   r#   r$   r   +   s   
ý€ÿzDiscordIO.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 Discord Bot.
    May take a few seconds to create (`__init__`).

    - create a discord bot (not public, no requirement of OAuth2 code
      grant, only send message permissions) & invite it to a channel:
      <https://discordpy.readthedocs.io/en/latest/discord.html>
    - copy the bot `{token}` & `{channel_id}` and paste below

    >>> from tqdm.contrib.discord import tqdm, trange
    >>> for i in tqdm(iterable, token='{token}', channel_id='{channel_id}'):
    ...     ...
    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. Discord token
            [default: ${TQDM_DISCORD_TOKEN}].
        channel_id  : int, required. Discord channel ID
            [default: ${TQDM_DISCORD_CHANNEL_ID}].
        mininterval  : float, optional.
          Minimum of [default: 1.5] to avoid rate limit.

        See `tqdm.auto.tqdm.__init__` for other parameters.
        ÚdisableÚ
HTTPClientr   ÚTQDM_DISCORD_TOKENr   ÚTQDM_DISCORD_CHANNEL_IDg      ø?ÚminintervalN)ÚgetÚcopyÚloggingÚ	getLoggerÚsetLevelÚWARNINGr   Úpopr   ÚdioÚmaxr   r   r   ©r   ÚargsÚkwargsr"   r#   r$   r   I   s   
þztqdm_discord.__init__c                    sj   t t| ƒjdi |¤Ž | j}| dd ¡r#|d  dd¡ dd¡|d< nd|d< | j | jdi |¤Ž¡ d S )NÚ
bar_formatz<bar/>z	{bar:10u}z{bar}z{l_bar}{bar:10u}{r_bar}r#   )	r   r   ÚdisplayÚformat_dictr7   r(   r>   r   Úformat_meter)r   rB   Úfmtr"   r#   r$   rD   _   s   ÿÿztqdm_discord.displayc                    s0   t t| ƒj|i |¤Ž | js| j d¡ d S d S )Nr&   )r   r   Úclearr2   r>   r   r@   r"   r#   r$   rH   i   s   ÿztqdm_discord.clear)r   r.   r/   r0   r   rD   rH   r1   r#   r#   r"   r$   r   ;   s
    
r   c                  O   s   t t| Ž fi |¤ŽS )z„
    A shortcut for `tqdm.contrib.discord.tqdm(xrange(*args), **kwargs)`.
    On Python3+, `range` is used instead of `xrange`.
    )r   r   )rA   rB   r#   r#   r$   r   o   s   r   )r0   Ú
__future__r   r9   Úosr   Údisco.clientr   r   ÚImportErrorÚautor   r   Úutilsr   Úutils_workerr
   Ú
__author__Ú__all__r   r   r   r   r#   r#   r#   r$   Ú<module>   s&    
ÿ
4	