
    g5                         d Z ddlZddlmZ 	 ddlmZmZ n# e$ r  ed          w xY wddlm	Z
 dd	lmZ d
dgiZg dZ G d de          Z G d de
          Z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/	casperdcl)	DiscordIOtqdm_discordtdranger   trangec                   (     e Zd ZdZ fdZd Z xZS )r   z.Non-blocking file-like IO using a Discord Bot.c                 ~   t                                                       t                      }||_        t	          |          }| j        j        | _        	 |j        	                    || j                  | _
        dS # t          $ r3}t          j        t          |                     d| _
        Y d}~dS d}~ww xY w)z0Creates a new message in the given `channel_id`.N)super__init__r   tokenr   	__class____name__textapichannels_messages_createmessage	Exception	tqdm_autowritestr)selfr   
channel_idconfigclienter   s         P/var/www/html/ai-engine/env/lib/python3.11/site-packages/tqdm/contrib/discord.pyr   zDiscordIO.__init__   s    N+		 !:>>z49UUDLLL 	  	  	 OCFF###DLLLLLLL	 s   %A? ?
B<	(B77B<c                 R   |sd}|                     dd                                          }|| j        k    rdS | j        }|dS || _        	 |                     |j        d|z   dz             }|S # t          $ r,}t          j        t          |                     Y d}~dS d}~ww xY w)z,Replaces internal `message`'s text with `s`.z... N`)
replacestripr   r   submiteditr   r   r   r   )r   sr   futurer"   s        r#   r   zDiscordIO.write)   s     	AIIdB%%''	>>F,?F		[[sQw}==F M  	$ 	$ 	$OCFF#########	$s   !A0 0
B&:!B!!B&)r   
__module____qualname____doc__r   r   __classcell__r   s   @r#   r   r      sM        88                   r   c                   6     e Zd ZdZ fdZ fdZ fdZ x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           	         |                     d          s|                                }t          j        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.
        disable
HTTPClientr   TQDM_DISCORD_TOKENr   TQDM_DISCORD_CHANNEL_IDg      ?minintervalN)getcopylogging	getLoggersetLevelWARNINGr   popr   diomaxr   r   r   argskwargsr   s      r#   r   ztqdm_discord.__init__J   s     zz)$$ 	M[[]]Fl++44W_EEE 

7F+?$@$@AA

<0I)J)JKKM MDH %(VZZs-K-K$L$LF=!$)&)))))r3   c                 0    t                      j        di | | j        }|                    dd           r4|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   displayformat_dictr;   r(   rB   r   format_meter)r   rF   fmtr   s      r#   rJ   ztqdm_discord.display`   s    !!&!!!77<&& 	: #L 1 9 9+!' !''.ww'D'D  !:C(t(//3//00000r3   c                      t                      j        |i | | j        s| j                            d           d S d S )Nr&   )r   clearr6   rB   r   rD   s      r#   rO   ztqdm_discord.clearj   sL    t&v&&&| 	HNN2	 	r3   )r   r.   r/   r0   r   rJ   rO   r1   r2   s   @r#   r   r   <   st         * * * * *,1 1 1 1 1        r3   r   c                  *    t          t          |  fi |S )zAShortcut for `tqdm.contrib.discord.tqdm(range(*args), **kwargs)`.)r   range)rE   rF   s     r#   r   r   p   s    t/////r3   )r0   r=   osr   disco.clientr   r   ImportErrorautor   r   utils_workerr	   
__author____all__r   r   r   r   rI   r3   r#   <module>rY      s*  	 	       7111111111 7 7 7
+5
6
667 % $ $ $ $ $ $ $ $ $ $ $k]+

D
D
D    
   D1 1 1 1 19 1 1 1h0 0 0 	s    (