U M{j{@sdZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd lmZd d lmZd d dgiZdddddgZGdddeZGddde 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) )getenv)warn)Session)default_user_agent)tqdm) TqdmWarning) __version__) MonoWorkerz github.com/Z casperdclZ guigoruiz1 DiscordIO tqdm_discordtdrangertrangecsReZdZdZdZdedeZfddZe ddZ d d Z d d Z Z S) r z.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)selfrrrS/var/www/html/bwave/python/venv/lib/python3.8/site-packages/tqdm/contrib/discord.pyrs   zDiscordIO.__init__c Cst|dr|jSzH|jj|jd|jdd|j|jddd|jdid }Wn0t k r}zt t |W5d}~XYn0X|d d krtd td d n|d|_|jSdS)N _message_id /channels/z /messagesBot  Authorizationz User-Agentcontent`headersjsonZ error_codeiz2Creation rate limit: try increasing `mininterval`.r) stacklevelid)hasattrrrpostAPIrrUArr& Exception tqdm_autowritestrgetrr)rreserrrr(s"   zDiscordIO.message_idc Cs|sd}|dd}||jkr&dS|j}|dkr8dS||_zL|j|jj|jd|jd|d|j |j dd d |jd id }Wn0t k r}zt t|W5d}~XYnX|SdS) z/Replaces internal `message_id`'s text with `s`.z... Nr /messages/rr r"r#r$)replacestriprrsubmitrpatchr+rrr,r-r.r/r0)rsrfuturer3rrrr/;s&   zDiscordIO.writec Csxz>|j|jj|jd|jd|jd|j|jdd}Wn0tk rn}zt t |W5d}~XYnX|SdS)zDeletes internal `message_id`.rr6rr )r%N) r9rdeleter+rrrr,r-r.r/r0)rr<r3rrrr=Qs  zDiscordIO.delete)r __module__ __qualname____doc__r+r rr,rpropertyrr/r= __classcell__rrrrr s  csDeZdZdZfddZfddZfddZfdd ZZS) 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}'): ... ... csH|ds6|}t|dtd|dtd|_tj||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. disablerZTQDM_DISCORD_TOKENrZTQDM_DISCORD_CHANNEL_IDN)r1copyr poprdiorrrargskwargsrrrrls ztqdm_discord.__init__c s^tjf||j}|ddr>|ddddd|d<nd|d<|j|jf|dS)NZ bar_formatzz {bar:10u}z{bar}z{l_bar}{bar:10u}{r_bar})rdisplayZ format_dictr1r7rFr/Z format_meter)rrIfmtrrrrJ~s  ztqdm_discord.displaycs$tj|||js |jddS)Nr5)rclearrCrFr/rGrrrrLsztqdm_discord.clearcs<|jr dSt|js8|jdkr.|jdks8|jdS)Nr)rCrcloseZleaveposrFr=)rrrrrMs  ztqdm_discord.close) rr>r?r@rrJrLrMrBrrrrr ^s   cOstt|f|S)zAShortcut for `tqdm.contrib.discord.tqdm(range(*args), **kwargs)`.)r range)rHrIrrrrsN)r@osrwarningsrrequestsrZrequests.utilsrautorr.Zstdrversionr Z utils_workerr __author____all__r r rrrrrrs         D7