o
    Ð­j�  ã                   @   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g d¢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/Ú	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                    s2   t ƒ  ¡  || _|| _tƒ | _| jj| _| j	 dS )z-Creates a new message in the given `chat_id`.N)
ÚsuperÚ__init__ÚtokenÚchat_idr   ÚsessionÚ	__class__Ú__name__ÚtextÚ
message_id)Úselfr   r   ©r   © úR/var/www/html/CropPilot/venv/lib/python3.10/site-packages/tqdm/contrib/telegram.pyr      s   


zTelegramIO.__init__c              
   C   s²   t | dƒr| jS z| jj| jd| j  d| j d | jddœd� ¡ }W n t	y> } zt
 t|ƒ¡ W Y d }~d S d }~ww | d¡dkrOtd	td
d� d S |d d | _| jS )NÚ_message_idz%s/sendMessageú`Ú
MarkdownV2)r   r   Ú
parse_mode©ÚdataÚ
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u rdS || _z| j| jj| jd| j  d| d | j	|ddœd	�}W |S  t
yV } zt t|ƒ¡ W Y d}~dS d}~ww )
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   sf   z| j | jj| jd| j  | j| jdœd�}W |S  ty2 } zt 	t
|ƒ¡ W Y d}~dS d}~ww )zDeletes internal `message_id`.z%s/deleteMessage)r   r   r    N)r4   r   r&   r'   r   r   r   r)   r*   r+   r,   )r   r6   r/   r   r   r   ÚdeleteM   s   þý€ÿzTelegramIO.delete)r   Ú
__module__Ú__qualname__Ú__doc__r'   r   Úpropertyr   r+   r7   Ú__classcell__r   r   r   r   r      s    	
r   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¡s| ¡ }t| dtdƒ¡| dtdƒ¡ƒ| _tƒ j|i |¤Ž 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   ÚTQDM_TELEGRAM_TOKENr   ÚTQDM_TELEGRAM_CHAT_IDN)r-   Úcopyr   Úpopr   Útgior   r   ©r   ÚargsÚkwargsr   r   r   r   i   s   
þztqdm_telegram.__init__c                    sf   t ƒ jdi |¤Ž | j}| dd ¡r!|d  dd¡ dd¡|d< nd|d< | j | jdi |¤Ž¡ d S )NÚ
bar_formatz<bar/>z	{bar:10u}z{bar}z{l_bar}{bar:10u}{r_bar}r   )r   ÚdisplayÚformat_dictr-   r2   rB   r+   Úformat_meter)r   rE   Úfmtr   r   r   rG   {   s   ÿÿztqdm_telegram.displayc                    s,   t ƒ j|i |¤Ž | js| j d¡ d S d S )Nr1   )r   Úclearr=   rB   r+   rC   r   r   r   rK   …   s   ÿztqdm_telegram.clearc                    sD   | j rd S tƒ  ¡  | js| jd u r| jdks | j ¡  d S d S d S )Nr   )r=   r   ÚcloseÚleaveÚposrB   r7   )r   r   r   r   rL   Š   s   
ÿztqdm_telegram.close)	r   r8   r9   r:   r   rG   rK   rL   r<   r   r   r   r   r   Y   s    
r   c                  O   s   t t| Ž fi |¤ŽS )zBShortcut for `tqdm.contrib.telegram.tqdm(range(*args), **kwargs)`.)r   Úrange)rD   rE   r   r   r   r   ’   s   r   N)r:   Úosr   Úwarningsr   Úrequestsr   Úautor   r*   Ústdr   Úutils_workerr	   Ú
__author__Ú__all__r   r   r   r   r   r   r   r   Ú<module>   s    

A9