U
    ·ÜŸ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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)
é    )Úwarn)ÚSessioné   )Útqdm)ÚTqdmWarning)Úenvwrapé   )Ú
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                    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   © ú9/tmp/pip-unpacked-wheel-ktn_y856/tqdm/contrib/telegram.pyr      s    

zTelegramIO.__init__c              
   C   sº   t | dƒr| jS zF| jj| j› | j› d�d| j› d�| jddœd�}| ¡ }| 	¡  W nJ t
k
r  } z,|jdkr‚tdtd	d
� nt t|ƒ¡ W 5 d }~X Y nX |d d | _| jS d S )NÚ_message_idz/sendMessageú`Ú
MarkdownV2)r   r   Ú
parse_mode©Údatai­  z2Creation rate limit: try increasing `mininterval`.r   )Ú
stacklevelÚresultr   )Úhasattrr   r   ÚpostÚAPIr   r   r   ÚjsonÚraise_for_statusÚ	ExceptionÚstatus_coder   r   Ú	tqdm_autoÚwriteÚstr)r   ÚreqÚ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 || _z:| j| jj| j› | j› d�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/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+   :   s*    
  ÿþ
 zTelegramIO.writec              
   C   s\   z"| j | jjd| j| jdœd�}W n0 tk
rR } zt t|ƒ¡ W 5 d}~X Y nX |S dS )zDeletes internal `message_id`.z${self.API}{self.token}/deleteMessage)r   r   r   N)	r4   r   r$   r   r   r(   r*   r+   r,   )r   r6   r/   r   r   r   ÚdeleteO   s     þ
 zTelegramIO.delete)r   Ú
__module__Ú__qualname__Ú__doc__r%   r   Úpropertyr   r+   r7   Ú__classcell__r   r   r   r   r
      s   	
c                       sZ   e Zd ZdZedddd�dddœ‡ 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}'):
    ...     ...
    r   ZtelegramT)Z	is_methodN)r   r   c                   s0   |  d¡s| ¡ }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.
        ÚdisableN)ÚgetÚcopyr
   Útgior   r   )r   r   r   ÚargsÚkwargsr   r   r   r   k   s    
ztqdm_telegram.__init__c                    sT   t ƒ jf |Ž | j}| dd ¡r<|d  dd¡ dd¡|d< | j | jf |Ž¡ d S )NZ
bar_formatz<bar/>z	{bar:10u}z{bar})r   ÚdisplayZformat_dictr>   r2   r@   r+   Zformat_meter)r   rB   Úfmtr   r   r   rC   |   s     ÿ ÿztqdm_telegram.displayc                    s$   t ƒ j||Ž | js | j d¡ d S )Nr1   )r   Úclearr=   r@   r+   )r   rA   rB   r   r   r   rE   „   s    ztqdm_telegram.clearc                    s<   | j r
d S tƒ  ¡  | js8| jd kr.| jdks8| j ¡  d S )Nr   )r=   r   ÚcloseZleaveÚposr@   r7   )r   r   r   r   rF   ‰   s
    
ztqdm_telegram.close)
r   r8   r9   r:   r   r   rC   rE   rF   r<   r   r   r   r   r   [   s   c                  O   s   t t| Ž f|ŽS )zBShortcut for `tqdm.contrib.telegram.tqdm(range(*args), **kwargs)`.)r   Úrange)rA   rB   r   r   r   r   ‘   s    N)r:   Úwarningsr   Úrequestsr   Úautor   r*   Zstdr   Úutilsr   Zutils_workerr	   Ú
__author__Ú__all__r
   r   r   r   r   r   r   r   Ú<module>   s   

C6