o
    Ð­j{  ã                   @   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   © úQ/var/www/html/CropPilot/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