o Ciz@sdZddlZddlmZz ddlmZmZWn ey!edwddlm Z dd l m Z d d giZ gd ZGd dde ZGddde ZddZeZ 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}'): ... ... ![screenshot](https://tqdm.github.io/img/screenshot-discord.png) N)getenv)Client ClientConfigzPlease `pip install disco-py`)tqdm) MonoWorkerz github.com/ casperdcl) DiscordIO tqdm_discordtdrangertrangecs(eZdZdZfddZddZZS)r z.Non-blocking file-like IO using a Discord Bot.c stt|t}||_t|}|jj|_z |j ||j|_ WdSt y?}zt t|d|_ WYd}~dSd}~ww)z0Creates a new message in the given `channel_id`.N)superr __init__rtokenr __class____name__textapichannels_messages_createmessage Exception tqdm_autowritestr)selfr channel_idconfigclienterl/var/www/html/fyndo/python/python_agents/rag_suite/venv/lib/python3.10/site-packages/tqdm/contrib/discord.pyrs zDiscordIO.__init__c Cs|sd}|dd}||jkrdS|j}|durdS||_z||jd|d}W|StyG}ztt |WYd}~dSd}~ww)z,Replaces internal `message`'s text with `s`.z... N`) replacestriprrsubmiteditrrrr)rsrfuturerr!r!r"r)s  zDiscordIO.write)r __module__ __qualname____doc__rr __classcell__r!r!r r"r s  r cs8eZdZdZfddZfddZfddZZS)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: - 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}'): ... ... csx|ds/|}tdtjt|dtd|dtd|_ t d|dd|d<t t |j |i|d S) 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. disable HTTPClientrTQDM_DISCORD_TOKENrTQDM_DISCORD_CHANNEL_IDg? minintervalN)getcopylogging getLoggersetLevelWARNINGr poprdiomaxrr rrargskwargsr r!r"rJs ztqdm_discord.__init__c sjtt|jdi||j}|ddr#|ddddd|d<nd|d<|j|jdi|dS)N bar_formatzz {bar:10u}z{bar}z{l_bar}{bar:10u}{r_bar}r!) rr display format_dictr5r&r<r format_meter)rr@fmtr r!r"rB`s ztqdm_discord.displaycs0tt|j|i||js|jddSdS)Nr$)rr clearr0r<rr>r r!r"rFjsztqdm_discord.clear)rr,r-r.rrBrFr/r!r!r r"r <s   r cOstt|fi|S)zAShortcut for `tqdm.contrib.discord.tqdm(range(*args), **kwargs)`.)r range)r?r@r!r!r"r psr )r.r7osr disco.clientrr ImportErrorautorr utils_workerr __author____all__r r r r r!r!r!r"s"     "4