File: //home/arjun/.local/lib/python3.10/site-packages/tqdm/contrib/__pycache__/discord.cpython-310.pyc
o
���g{ � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z ddl m
Z
dd lmZ d
d lmZ d d
dgiZg d�ZG dd� de�ZG dd� de �Zdd� ZeZ
eZdS )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}'):
... ...
� )�getenv)�warn)�Session)�default_user_agent� )�tqdm)� TqdmWarning)� __version__� )�
MonoWorkerz github.com/� casperdcl�
guigoruiz1)� DiscordIO� tqdm_discord�tdranger �trangec sR e Zd ZdZdZde� de� � �Z� fdd�Ze dd� �Z
d d
� Z d d � Z � Z
S )
r z.Non-blocking file-like IO using a Discord Bot.zhttps://discord.com/api/v10z tqdm (https://tqdm.github.io, z) c s2 t � �� || _|| _t� | _| jj| _| j dS )z0Creates a new message in the given `channel_id`.N)
�super�__init__�token�
channel_idr �session� __class__�__name__�text�
message_id)�selfr r �r � �G/home/arjun/.local/lib/python3.10/site-packages/tqdm/contrib/discord.pyr s
zDiscordIO.__init__c
C s� t | d�r| jS z$| jj| j� d| j� d�d| j� �| jd�dd| j� d�id�� � }W n t
yF } zt � t
|�� W Y d }~d S d }~ww |�d �d
krWtd td d
� d S |d | _| jS )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(
�� ��
�
zDiscordIO.message_idc
C s� |sd}|� dd��� }|| jkrdS | j}|du r dS || _z'| j| jj| j� d| j� d|� �d| j � �| j
d�d d
| j� d
�id �}W |S t y` } zt �
t|�� W Y d}~dS d}~ww ) 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*
�� ��zDiscordIO.writec
C sv z | j | jj| j� d| j� d| j� �d| j� �| jd�d�}W |S ty: } zt �
t |�� W Y d}~dS d}~ww )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 �� ��zDiscordIO.delete)r �
__module__� __qualname__�__doc__r. r r r/ r �propertyr r2 r@ �
__classcell__r r r r r s
r c sD e Zd ZdZ� fdd�Z� fdd�Z� fdd�Z� fdd �Z� ZS )
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|�dtd��|�dtd���| _t� j|i |�� dS )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
�ztqdm_discord.__init__c sf t � jdi |�� | j}|�dd �r!|d �dd��dd�|d< nd|d< | j�| jdi |��� 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 �� ztqdm_discord.displayc s, t � j|i |�� | js| j�d� d S d S )Nr8 )r �clearrF rK r2 rL r r r rT � s �ztqdm_discord.clearc sD | j rd S t� �� | js | jd u r| jdks | j�� d S d S d S )Nr )rF r �close�leave�posrK r@ )r r r r rU � s
�ztqdm_discord.close) r rA rB rC r rP rT rU rE r r r r r ^ s
r c O s t t| � fi |��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
D7