U
    9%e                     @   s   d Z ddlmZ ddlmZ ddlmZ ddlmZ	 ddl
mZ dd	lmZ d
dgiZdddddgZG dd deZG dd de	Zdd ZeZeZdS )z
Sends updates to a Telegram bot.

Usage:
>>> from tqdm.contrib.telegram import tqdm, trange
>>> for i in trange(10, token='{token}', chat_id='{chat_id}'):
...     ...

![screenshot](https://tqdm.github.io/img/screenshot-telegram.gif)
    )getenv)warn)Session   )tqdm)TqdmWarning   )
MonoWorkerzgithub.com/Z	casperdcl
TelegramIOtqdm_telegramttgranger   trangec                       s@   e Zd ZdZdZ fddZedd Zdd Zd	d
 Z	  Z
S )r
   z/Non-blocking file-like IO using a Telegram Bot.zhttps://api.telegram.org/botc                    s6   t t|   || _|| _t | _| jj| _	| j
 dS )z-Creates a new message in the given `chat_id`.N)superr
   __init__tokenchat_idr   session	__class____name__text
message_id)selfr   r   r    T/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/tqdm/contrib/telegram.pyr      s    
zTelegramIO.__init__c              
   C   s   t | dr| jS z8| jj| jd| j  d| j d | jddd }W n0 t	k
rx } zt
t| W 5 d }~X Y n4X |ddkrtd	td
d n|d d | _| jS d S )N_message_idz%s/sendMessage`
MarkdownV2)r   r   
parse_modedataZ
error_codei  z2Creation rate limit: try increasing `mininterval`.r   )
stacklevelresultr   )hasattrr   r   postAPIr   r   r   json	Exception	tqdm_autowritestrgetr   r   )r   reser   r   r   r   %   s$    
  zTelegramIO.message_idc              
   C   s   |sd}| dd }|| jkr&dS | j}|dkr8dS || _z8| j| jj| jd| j  d| d | j	|ddd	}W n0 t
k
r } ztt| W 5 d}~X Y nX |S dS )
z/Replaces internal `message_id`'s text with `s`.z... Nz%s/editMessageTextr   r   )r   r   r   r   r   )replacestripr   r   submitr   r$   r%   r   r   r'   r(   r)   r*   )r   sr   futurer-   r   r   r   r)   8   s*    
  
 zTelegramIO.writec              
   C   sh   z.| j | jj| jd| j  | j| jdd}W n0 tk
r^ } zt	t
| W 5 d}~X Y nX |S dS )zDeletes internal `message_id`.z%s/deleteMessage)r   r   r   N)r2   r   r$   r%   r   r   r   r'   r(   r)   r*   )r   r4   r-   r   r   r   deleteM   s     
 zTelegramIO.delete)r   
__module____qualname____doc__r%   r   propertyr   r)   r5   __classcell__r   r   r   r   r
      s   	
c                       sD   e Zd ZdZ fddZ fddZ fddZ fdd	Z  ZS )
r   a;  
    Standard `tqdm.auto.tqdm` but also sends updates to a Telegram Bot.
    May take a few seconds to create (`__init__`).

    - create a bot <https://core.telegram.org/bots#6-botfather>
    - copy its `{token}`
    - add the bot to a chat and send it a message such as `/start`
    - go to <https://api.telegram.org/bot`{token}`/getUpdates> to find out
      the `{chat_id}`
    - paste the `{token}` & `{chat_id}` below

    >>> from tqdm.contrib.telegram import tqdm, trange
    >>> for i in tqdm(iterable, token='{token}', chat_id='{chat_id}'):
    ...     ...
    c                    sL   | ds6| }t|dtd|dtd| _tt| j|| dS )a.  
        Parameters
        ----------
        token  : str, required. Telegram token
            [default: ${TQDM_TELEGRAM_TOKEN}].
        chat_id  : str, required. Telegram chat ID
            [default: ${TQDM_TELEGRAM_CHAT_ID}].

        See `tqdm.auto.tqdm.__init__` for other parameters.
        disabler   ZTQDM_TELEGRAM_TOKENr   ZTQDM_TELEGRAM_CHAT_IDN)	r+   copyr
   popr   tgior   r   r   r   argskwargsr   r   r   r   i   s    
ztqdm_telegram.__init__c                    sb   t t| jf | | j}|dd rB|d dddd|d< nd|d< | j| jf | d S )NZ
bar_formatz<bar/>z	{bar:10u}z{bar}z{l_bar}{bar:10u}{r_bar})	r   r   displayZformat_dictr+   r0   r>   r)   Zformat_meter)r   rA   fmtr   r   r   rB   {   s      
ztqdm_telegram.displayc                    s(   t t| j|| | js$| jd d S )Nr/   )r   r   clearr;   r>   r)   r?   r   r   r   rD      s    ztqdm_telegram.clearc                    s@   | j r
d S tt|   | js<| jd kr2| jdks<| j  d S )Nr   )r;   r   r   closeZleaveposr>   r5   )r   r   r   r   rE      s
    ztqdm_telegram.close)	r   r6   r7   r8   r   rB   rD   rE   r:   r   r   r   r   r   Y   s
   
c                  O   s   t t|  f|S )zBShortcut for `tqdm.contrib.telegram.tqdm(range(*args), **kwargs)`.)r   range)r@   rA   r   r   r   r      s    N)r8   osr   warningsr   requestsr   autor   r(   Zstdr   Zutils_workerr	   
__author____all__r
   r   r   r   r   r   r   r   <module>   s   

A9