a
    Lªf5  ã                   @   s¦   d Z ddlZddlmZ zddlmZmZ W n eyF   edƒ‚Y n0 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 trange(10, token='{token}', channel_id='{channel_id}'):
...     ...

![screenshot](https://tqdm.github.io/img/screenshot-discord.png)
é    N)Úgetenv)ÚClientÚClientConfigzPlease `pip install disco-py`é   )Útqdmé   )Ú
MonoWorkerzgithub.com/Z	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              
      s|   t ƒ  ¡  tƒ }||_t|ƒ}| jj| _z|j 	|| j¡| _
W n8 tyv } z t t|ƒ¡ d| _
W Y d}~n
d}~0 0 dS )z0Creates a new message in the given `channel_id`.N)ÚsuperÚ__init__r   Útokenr   Ú	__class__Ú__name__ÚtextÚapiZchannels_messages_createÚmessageÚ	ExceptionÚ	tqdm_autoÚwriteÚstr)Úselfr   Ú
channel_idÚconfigÚclientÚe©r   © úM/var/www/ai-form-bot/venv/lib/python3.9/site-packages/tqdm/contrib/discord.pyr      s    

zDiscordIO.__init__c              
   C   s”   |sd}|  dd¡ ¡ }|| jkr&dS | j}|du r8dS || _z|  |jd| d ¡}W n2 tyŠ } zt t	|ƒ¡ W Y d}~nd}~0 0 |S dS )z,Replaces internal `message`'s text with `s`.z...úÚ Nú`)
ÚreplaceÚstripr   r   ZsubmitZeditr   r   r   r   )r   Úsr   Ú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                    st   |  d¡s^| ¡ }t d¡ tj¡ t| dtdƒ¡| dtdƒ¡ƒ| _	t
d|  dd¡ƒ|d< 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.
        ÚdisableZ
HTTPClientr   ZTQDM_DISCORD_TOKENr   ZTQDM_DISCORD_CHANNEL_IDg      ø?ZminintervalN)ÚgetÚcopyÚloggingÚ	getLoggerÚsetLevelÚWARNINGr	   Úpopr   ÚdioÚmaxr   r   ©r   ÚargsÚkwargsr   r   r    r   J   s    
þztqdm_discord.__init__c                    sf   t ƒ jf i |¤Ž | j}| dd ¡rB|d  dd¡ dd¡|d< nd|d< | j | jf i |¤Ž¡ d S )NZ
bar_formatz<bar/>z	{bar:10u}z{bar}z{l_bar}{bar:10u}{r_bar})r   ÚdisplayZformat_dictr-   r$   r4   r   Zformat_meter)r   r8   Úfmtr   r   r    r9   `   s    ÿÿ
ztqdm_discord.displayc                    s(   t ƒ j|i |¤Ž | js$| j d¡ d S )Nr"   )r   Úclearr,   r4   r   r6   r   r   r    r;   j   s    ztqdm_discord.clear)r   r(   r)   r*   r   r9   r;   r+   r   r   r   r    r
   <   s   
r
   c                  O   s   t t| Ž fi |¤ŽS )zAShortcut for `tqdm.contrib.discord.tqdm(range(*args), **kwargs)`.)r
   Úrange)r7   r8   r   r   r    r   p   s    r   )r*   r/   Úosr   Zdisco.clientr   r   ÚImportErrorÚautor   r   Zutils_workerr   Ú
__author__Ú__all__r	   r
   r   r   r   r   r   r    Ú<module>   s   

"4