
    0Rj                         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y)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)
    )warn)Session)default_user_agent   )tqdm)TqdmWarning)envwrap)__version__   )
MonoWorkerzgithub.com/	casperdcl
guigoruiz1)	DiscordIOtqdm_discordtdranger   trangec                   \     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                   y)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      \/var/www/html/meridian/meridian-ai/venv/lib/python3.12/site-packages/tqdm/contrib/discord.pyr   zDiscordIO.__init__   s=    
$yNN++	    c                    t        | d      r| j                  S 	 | j                  j                  | j                   d| j
                   dd| j                   | j                  ddd| j                   di      }|j                         }|j                          |d	   | _        | j                  S # t        $ rO}j                  d
k(  rt        dt        d       n#t        j                   t#        |             Y d }~y Y d }~y d }~ww xY w)N_message_id
/channels/z	/messagesBot Authorizationz
User-Agentcontent`headersjsonidi  z2Creation rate limit: try increasing `mininterval`.r   )
stacklevel)hasattrr"   r   postAPIr   r   UAr   r+   raise_for_status	Exceptionstatus_coder   r   	tqdm_autowritestr)r   reqreses       r   r   zDiscordIO.message_id(   s    4'###	$,,##88*Jt&7yA,0*=TWWU1TYYKq!12 $ 4C ((*C  "  #4yD###  	(#%I Q0 A''0	(s   BB2 2	D
;A DD
c           	         |sd}|j                  dd      j                         }|| j                  k(  ry| j                  }|y|| _        	 | j	                  | j
                  j                  | j                   d| j                   d| d| j                   | j                  dd	d
| j                   d
i      }|S # t        $ r(}t        j                  t        |             Y d}~y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   patchr0   r   r   r1   r3   r5   r6   r7   )r   sr   futurer:   s        r   r6   zDiscordIO.write=   s    AIIdB%%'		>__
			[[""88*Jt&7z*N,0*=TWWU1TYYKq!12	 ! 4F M  	$OOCF##	$s   A/B= =	C.C))C.c                 @   	 | j                  | j                  j                  | j                   d| j                   d| j
                   d| j                   | j                  d      }|S # t        $ r(}t        j                  t        |             Y d}~yd}~ww xY w)zDeletes internal `message_id`.r#   r>   r$   r%   )r*   N)rA   r   deleter0   r   r   r   r1   r3   r5   r6   r7   )r   rD   r:   s      r   rF   zDiscordIO.deleteS   s    	[[##88*Jt&7z$//ARS,0*=TWWU ! WF M  	$OOCF##	$s   A(A, ,	B5BB)r   
__module____qualname____doc__r0   r
   r   r1   r   propertyr   r6   rF   __classcell__r   s   @r   r   r      sE    8
'C)+b9K9M8N	OB $ $(,
r    r   c                   d     e Zd ZdZ eddd      ddd 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}'):
    ...     ...
    r   discordT)	is_methodN)r   r   c                    |j                  d      s!|j                         }t        ||      | _        t	        |   |i | y)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.
        disableN)getcopyr   dior   r   )r   r   r   argskwargsr   s        r   r   ztqdm_discord.__init__n   s<     zz)$[[]F 
3DH$)&)r    c                    t        |   di | | j                  }|j                  dd       r(|d   j	                  dd      j	                  dd      |d<   | j
                  j                   | j                  di |       y )N
bar_formatz<bar/>z	{bar:10u}z{bar} )r   displayformat_dictrR   r?   rT   r6   format_meter)r   rV   fmtr   s      r   rZ   ztqdm_discord.display   sz    !&!77<& #L 1 9 9+!''.ww'D (t((/3/0r    c                 t    t        |   |i | | j                  s| j                  j	                  d       y y )Nr=   )r   clearrQ   rT   r6   )r   rU   rV   r   s      r   r_   ztqdm_discord.clear   s0    t&v&||HHNN2 r    c                     | j                   ry t        | 	          | j                  s7| j                  | j                  dk(  s| j
                  j                          y y y )Nr   )rQ   r   closeleaveposrT   rF   )r   r   s    r   ra   ztqdm_discord.close   sF    <<

tzz1dhh!mHHOO 7D
r    )
r   rG   rH   rI   r	   r   rZ   r_   ra   rK   rL   s   @r   r   r   `   s>     VY$/$(T * 0* 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)rI   warningsr   requestsr   requests.utilsr   autor   r5   stdr   utilsr	   versionr
   utils_workerr   
__author____all__r   r   r   r   rY   r    r   <module>rp      se   	   - $   ! $k<89

DC
 CL19 1h0 	r    