o j@sdZddlmZddlmZddlmZddlmZ ddl m Z ddl m Z dd lmZd d lmZd d dgiZgdZGdddeZGddde ZddZeZeZdS)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) )warn)Session)default_user_agent)tqdm) TqdmWarning)envwrap) __version__) MonoWorkerz github.com/ casperdcl guigoruiz1) DiscordIO tqdm_discordtdrangertrangecsReZdZdZdZdedeZfddZe ddZ d d Z d d Z Z S) rz.Non-blocking file-like IO using a Discord Bot.zhttps://discord.com/api/v10ztqdm (https://tqdm.github.io, z) cs2t||_||_t|_|jj|_|j dS)z0Creates a new message in the given `channel_id`.N) super__init__token channel_idrsession __class____name__text message_id)selfrrr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/tqdm/contrib/discord.pyrs   zDiscordIO.__init__c Cst|dr|jSz*|jj|jd|jdd|j|jddd|jdid}| }| Wn.t y`}z"|j d krGt d td d ntt|WYd}~dSWYd}~dSd}~ww|d |_|jS)N _message_id /channels/z /messagesBot  Authorizationz User-Agentcontent`headersjsoniz2Creation rate limit: try increasing `mininterval`.r) stacklevelid)hasattrrrpostAPIrrUArr(raise_for_status Exception status_coderr tqdm_autowritestr)rreqreserrrr(s*    zDiscordIO.message_idc Cs|sd}|dd}||jkrdS|j}|durdS||_z'|j|jj|jd|jd|d|j |j dd d |jd id }W|St y`}zt t|WYd}~dSd}~ww) z/Replaces internal `message_id`'s text with `s`.z... Nr /messages/r!r"r$r%r&)replacestriprrsubmitrpatchr-rrr.r0r2r3r4)rsrfuturer7rrrr3=s* zDiscordIO.writec Csvz |j|jj|jd|jd|jd|j|jdd}W|Sty:}zt t |WYd}~dSd}~ww)zDeletes internal `message_id`.r r:r!r")r'N) r=rdeleter-rrrr.r0r2r3r4)rr@r7rrrrASszDiscordIO.delete)r __module__ __qualname____doc__r-r rr.rpropertyrr3rA __classcell__rrrrrs  rcsZeZdZdZedddddddfdd Zfd d Zfd d ZfddZZ S)ra 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}'): ... ... rdiscordT) is_methodN)rrcs4|ds|}t|||_tj|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. disableN)getcopyrdiorr)rrrargskwargsrrrrns  ztqdm_discord.__init__c s\tjdi||j}|ddr |ddddd|d<|j|jdi|dS)N bar_formatzz {bar:10u}z{bar}r)rdisplay format_dictrJr;rLr3 format_meter)rrNfmtrrrrPs ztqdm_discord.displaycs,tj|i||js|jddSdS)Nr9)rclearrIrLr3)rrMrNrrrrTsztqdm_discord.clearcsD|jrdSt|js|jdur|jdks |jdSdSdS)Nr)rIrcloseleaveposrLrA)rrrrrUs  ztqdm_discord.close) rrBrCrDrrrPrTrUrFrrrrr`s   rcOstt|fi|S)zAShortcut for `tqdm.contrib.discord.tqdm(range(*args), **kwargs)`.)rrange)rMrNrrrrsrN)rDwarningsrrequestsrrequests.utilsrautorr2stdrutilsrversionr utils_workerr __author____all__rrrrrrrrs         F4