o
r¨ÊhÁ ã @ s¾ d Z ddlmZ ddlmZ ddlmZ z
ddlmZm Z W n e
y) e
dƒ‚w ddl m Z
ddlmZ d d
lmZ d d giZg d
¢ZG dd„ deƒZG dd„ de
ƒZdd„ ZeZ eZdS )z!
Sends updates to a Discord bot.
é )Úabsolute_import)Údeepcopy)Úgetenv)ÚClientÚ ClientConfigz Please `pip install disco-py`)Útqdm)Ú_rangeé )Ú
MonoWorkerz github.com/Ú casperdcl)Ú DiscordIOÚ tqdm_discordÚtdranger Útrangec s( e Zd ZdZ‡ fdd„Zdd„ Z‡ ZS )r z.Non-blocking file-like IO using a Discord Bot.c
sz t t| ƒ ¡ tƒ }||_t|ƒ}| jj| _z |j
|| j¡| _ W dS t y< } zt
t|ƒ¡ W Y d}~dS d}~ww )z0Creates a new message in the given `channel_id`.N)Úsuperr Ú__init__r Útokenr Ú __class__Ú__name__ÚtextÚapiÚchannels_messages_createÚmessageÚ ExceptionÚ tqdm_autoÚwriteÚstr)Úselfr Ú
channel_idÚconfigÚclientÚe©r © úF/var/www/html/env/lib/python3.10/site-packages/tqdm/contrib/discord.pyr s
ÿ €ÿzDiscordIO.__init__c
C s€ |sdS | dd¡ ¡ }|| jkrdS || _z| | jjd| d ¡}W |S ty? } zt t |ƒ¡ W Y d}~dS d}~ww )z,Replaces internal `message`'s text with `s`.Nú
Ú ú`)
ÚreplaceÚstripr Úsubmitr Úeditr r r r )r ÚsÚfuturer! r# r# r$ r # s
ý €ÿzDiscordIO.write)r Ú
__module__Ú __qualname__Ú__doc__r r Ú
__classcell__r# r# r" r$ r s
r c s8 e Zd ZdZ‡ 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 s\ t |ƒ}t| dtdƒ¡| dtdƒ¡ƒ| _td| dd¡ƒ|d< tt| ƒj |i |¤Ž dS )a”
Parameters
----------
token : str, required. Discord token
[default: ${TQDM_DISCORD_TOKEN}].
channel_id : int, required. Discord channel ID
[default: ${TQDM_DISCORD_CHANNEL_ID}].
mininterval : float, optional.
Minimum of [default: 1.5] to avoid rate limit.
See `tqdm.auto.tqdm.__init__` for other parameters.
r ÚTQDM_DISCORD_TOKENr ÚTQDM_DISCORD_CHANNEL_IDg ø?Ú minintervalN)
r r Úpopr ÚdioÚmaxÚgetr r
r )r ÚargsÚkwargsr" r# r$ r A s
þztqdm_discord.__init__c sz t t| ƒjdi |¤Ž | j}d|v r!|d r!|d dd¡|d< nd|d< |d dd¡|d< | j | jdi |¤Ž¡ d S )NÚ
bar_formatzz{bar}z{l_bar}{bar}{r_bar}z {bar:10u}r# )r r
ÚdisplayÚ format_dictr( r6 r Ú format_meter)r r: Úfmtr" r# r$ r< V s ztqdm_discord.displayc s | ¡ ztj| _W n ty Y nw W d ƒ n1 s w Y tt| ƒj| g|¢R i |¤Ž}| ¡ z | j tj¡ W n tyJ Y nw | jt_W d ƒ |S 1 sZw Y |S )z‰
Workaround for mixed-class same-stream nested progressbars.
See [#509](https://github.com/tqdm/tqdm/issues/509)
N)Úget_lockr Ú
_instancesÚAttributeErrorr r
Ú__new__Úupdate)Úclsr9 r: Úinstancer" r# r$ rC ` s(
ÿ€ ý
ÿ
úùztqdm_discord.__new__)r r. r/ r0 r r< rC r1 r# r# r" r$ r
3 s
r
c O s t t| Ž fi |¤ŽS )z„
A shortcut for `tqdm.contrib.discord.tqdm(xrange(*args), **kwargs)`.
On Python3+, `range` is used instead of `xrange`.
)r
r )r9 r: r# r# r$ r u s r N)r0 Ú
__future__r Úcopyr Úosr Ú disco.clientr r Ú ImportErrorÚ tqdm.autor r Ú
tqdm.utilsr Ú utils_workerr
Ú
__author__Ú__all__r r
r r r# r# r# r$ Ú s& ÿ
B