
    LMh{                         d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl	m
Z 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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)
    )getenv)warn)Session)default_user_agent   )tqdm)TqdmWarning)__version__   )
MonoWorkerzgithub.com/	casperdcl
guigoruiz1)	DiscordIOtqdm_discordtdranger   trangec                   h     e Zd ZdZdZde d e             Z fdZe	d             Z
d Zd Z xZS )	r   z.Non-blocking file-like IO using a Discord Bot.zhttps://discord.com/api/v10ztqdm (https://tqdm.github.io, z) c                     t                                                       || _        || _        t	                      | _        | j        j        | _        | j	         dS )z0Creates a new message in the given `channel_id`.N)
super__init__token
channel_idr   session	__class____name__text
message_id)selfr   r   r   s      T/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/tqdm/contrib/discord.pyr   zDiscordIO.__init__   sI    
$yyN+	    c                    t          | d          r| j        S 	 | j                            | j         d| j         dd| j         | j        ddd| j         di          	                                }|
                    d	          d
k    rt          dt          d           d S |d         | _        | j        S # t          $ r,}t          j        t!          |                     Y d }~d S d }~ww xY w)N_message_id
/channels/z	/messagesBot Authorizationz
User-Agentcontent`headersjson
error_codei  z2Creation rate limit: try increasing `mininterval`.r   )
stacklevelid)hasattrr"   r   postAPIr   r   UAr   r+   getr   r	   	Exception	tqdm_autowritestr)r   reses      r   r   zDiscordIO.message_id(   s3   4'' 	$##	(,##8AAtAAA*=*=*=TWUU!1TY!1!1!12 $ 4 4 59DFF  ww|$$++I Q0 0 0 0 0 0 $'t9 ''  	$ 	$ 	$OCFF#########	$s   AB9 9
C/!C**C/c           	         |sd}|                     dd                                          }|| j        k    rdS | j        }|dS || _        	 |                     | j        j        | j         d| j         d| d| j	         | j
        dd	d
| j         d
i          }|S # t          $ r,}t          j        t          |                     Y d}~dS d}~ww xY w)z/Replaces internal `message_id`'s text with `s`.z... Nr#   
/messages/r$   r%   r'   r(   r)   )replacestripr   r   submitr   patchr1   r   r   r2   r4   r5   r6   r7   )r   sr   futurer9   s        r   r6   zDiscordIO.write;   s    	AIIdB%%''	>>F_
F			[["8NNtNN*NN*=*=*=TWUU!1TY!1!1!12	 ! 4 4F M  	$ 	$ 	$OCFF#########	$s   AB 
C)!CCc                    	 |                      | j        j        | j         d| j         d| j         d| j         | j        d          }|S # t          $ r,}t          j
        t          |                     Y d}~dS d}~ww xY w)zDeletes internal `message_id`.r#   r=   r$   r%   )r*   N)r@   r   deleter1   r   r   r   r2   r4   r5   r6   r7   )r   rC   r9   s      r   rE   zDiscordIO.deleteQ   s    	[[#8SStSS$/SS*=*=*=TWUU ! W WF M  	$ 	$ 	$OCFF#########	$s   A	A 
B!A>>B)r   
__module____qualname____doc__r1   r
   r   r2   r   propertyr   r6   rE   __classcell__r   s   @r   r   r      s        88
'C	O+	O	O9K9K9M9M	O	OB     ( ( X($  ,
 
 
 
 
 
 
r    r   c                   @     e Zd ZdZ f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          sk|                                }t          |                    dt	          d                    |                    dt	          d                              | _         t                      j        |i | dS )a7  
        Parameters
        ----------
        token  : str, required. Discord bot token
            [default: ${TQDM_DISCORD_TOKEN}].
        channel_id  : int, required. Discord channel ID
            [default: ${TQDM_DISCORD_CHANNEL_ID}].

        See `tqdm.auto.tqdm.__init__` for other parameters.
        disabler   TQDM_DISCORD_TOKENr   TQDM_DISCORD_CHANNEL_IDN)r3   copyr   popr   dior   r   r   argskwargsr   s      r   r   ztqdm_discord.__init__l   s     zz)$$ 	M[[]]F 

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

<0I)J)JKKM MDH 	$)&)))))r    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_dictr3   r>   rS   r6   format_meter)r   rV   fmtr   s      r   rZ   ztqdm_discord.display~   s    !!&!!!77<&& 	: #L 1 9 9+!' !''.ww'D'D  !:C(t(//3//00000r    c                      t                      j        |i | | j        s| j                            d           d S d S )Nr<   )r   clearrN   rS   r6   rT   s      r   r_   ztqdm_discord.clear   sL    t&v&&&| 	HNN2	 	r    c                     | j         rd S t                                                       | j        s-| j        | j        dk    s| j                                         d S d S d S )Nr   )rN   r   closeleaveposrS   rE   )r   r   s    r   ra   ztqdm_discord.close   sd    < 	F
 	tz1dh!mmHOO	 	mmr    )	r   rF   rG   rH   r   rZ   r_   ra   rJ   rK   s   @r   r   r   ^   s         * * * * *$1 1 1 1 1    
        r    r   c                  *    t          t          |  fi |S )zAShortcut for `tqdm.contrib.discord.tqdm(range(*args), **kwargs)`.)r   range)rU   rV   s     r   r   r      s    t/////r    N)rH   osr   warningsr   requestsr   requests.utilsr   autor   r5   stdr	   versionr
   utils_workerr   
__author____all__r   r   r   r   rY   r    r   <module>rp      s8  	 	                   - - - - - - $ $ $ $ $ $       ! ! ! ! ! ! $ $ $ $ $ $k<89

D
D
DA A A A A
 A A AH4 4 4 4 49 4 4 4n0 0 0 	r    