o
    {j6h{                     @   s   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 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)
    )getenv)warn)Session)default_user_agent   )tqdm)TqdmWarning)__version__   )
MonoWorkerzgithub.com/	casperdcl
guigoruiz1)	DiscordIOtqdm_discordtdranger   trangec                       sR   e Zd ZdZdZde de  Z fddZe	dd Z
d	d
 Zdd Z  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                    s2   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    c/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/site-packages/tqdm/contrib/discord.pyr      s   


zDiscordIO.__init__c              
   C   s   t | dr| jS z$| jj| j d| j dd| j | jddd| j did	 }W n t
yF } ztt| W Y d }~d S d }~ww |d	d
krWtdtdd d S |d | _| jS )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(   	Exception	tqdm_autowritestrgetr   r   )r   reser   r   r   r   (   s(   


zDiscordIO.message_idc              
   C   s   |sd}| dd }|| jkrdS | j}|du rdS || _z'| j| jj| j d| j d| d| j	 | j
dd	d
| j d
id}W |S  ty` } ztt| W Y d}~dS d}~ww )z/Replaces internal `message_id`'s text with `s`.z... Nr    
/messages/r!   r"   r$   r%   r&   )replacestripr   r   submitr   patchr.   r   r   r/   r0   r1   r2   r3   )r   sr   futurer6   r   r   r   r2   ;   s*   
zDiscordIO.writec              
   C   sv   z | j | jj| j d| j d| j d| j | jdd}W |S  ty: } zt	
t| W Y d}~dS d}~ww )zDeletes internal `message_id`.r    r9   r!   r"   )r'   N)r<   r   deleter.   r   r   r   r/   r0   r1   r2   r3   )r   r?   r6   r   r   r   r@   Q   s   zDiscordIO.delete)r   
__module____qualname____doc__r.   r	   r   r/   r   propertyr   r2   r@   __classcell__r   r   r   r   r      s    	
r   c                       sD   e Zd ZdZ fd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                    sL   | ds| }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)r4   copyr   popr   dior   r   r   argskwargsr   r   r   r   l   s   
ztqdm_discord.__init__c                    sf   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   displayformat_dictr4   r:   rK   r2   format_meter)r   rN   fmtr   r   r   rP   ~   s   ztqdm_discord.displayc                    s,   t  j|i | | js| jd d S d S )Nr8   )r   clearrF   rK   r2   rL   r   r   r   rT      s   ztqdm_discord.clearc                    sD   | j rd S t   | js| jd u r| jdks | j  d S d S d S )Nr   )rF   r   closeleaveposrK   r@   )r   r   r   r   rU      s   
ztqdm_discord.close)	r   rA   rB   rC   r   rP   rT   rU   rE   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)rM   rN   r   r   r   r      s   r   N)rC   osr   warningsr   requestsr   requests.utilsr   autor   r1   stdr   versionr	   utils_workerr   
__author____all__r   r   r   r   r   r   r   r   <module>   s     
D7